diff options
Diffstat (limited to 'model')
-rw-r--r-- | model/file_info.go | 7 | ||||
-rw-r--r-- | model/file_info_test.go | 22 |
2 files changed, 24 insertions, 5 deletions
diff --git a/model/file_info.go b/model/file_info.go index 131baff6d..f785042b3 100644 --- a/model/file_info.go +++ b/model/file_info.go @@ -32,6 +32,11 @@ func GetInfoForBytes(filename string, data []byte) (*FileInfo, *AppError) { mimeType = mime.TypeByExtension(extension) } + if extension != "" && extension[0] == '.' { + // the client expects a file extension without the leading period + extension = extension[1:] + } + hasPreviewImage := isImage if mimeType == "image/gif" { // just show the gif itself instead of a preview image for animated gifs @@ -45,7 +50,7 @@ func GetInfoForBytes(filename string, data []byte) (*FileInfo, *AppError) { return &FileInfo{ Filename: filename, Size: size, - Extension: extension[1:], + Extension: extension, MimeType: mimeType, HasPreviewImage: hasPreviewImage, }, nil diff --git a/model/file_info_test.go b/model/file_info_test.go index ecf0d509c..e89681626 100644 --- a/model/file_info_test.go +++ b/model/file_info_test.go @@ -19,7 +19,7 @@ func TestGetInfoForBytes(t *testing.T) { } else if info.Size != 1000 { t.Fatalf("Got incorrect size: %v", info.Size) } else if info.Extension != "txt" { - t.Fatalf("Git incorrect file extension: %v", info.Extension) + t.Fatalf("Got incorrect file extension: %v", info.Extension) } else if info.MimeType != "text/plain; charset=utf-8" { t.Fatalf("Got incorrect mime type: %v", info.MimeType) } else if info.HasPreviewImage { @@ -33,7 +33,7 @@ func TestGetInfoForBytes(t *testing.T) { } else if info.Size != 1000 { t.Fatalf("Got incorrect size: %v", info.Size) } else if info.Extension != "png" { - t.Fatalf("Git incorrect file extension: %v", info.Extension) + t.Fatalf("Got incorrect file extension: %v", info.Extension) } else if info.MimeType != "image/png" { t.Fatalf("Got incorrect mime type: %v", info.MimeType) } else if !info.HasPreviewImage { @@ -49,7 +49,7 @@ func TestGetInfoForBytes(t *testing.T) { } else if info.Size != 35 { t.Fatalf("Got incorrect size: %v", info.Size) } else if info.Extension != "gif" { - t.Fatalf("Git incorrect file extension: %v", info.Extension) + t.Fatalf("Got incorrect file extension: %v", info.Extension) } else if info.MimeType != "image/gif" { t.Fatalf("Got incorrect mime type: %v", info.MimeType) } else if !info.HasPreviewImage { @@ -67,10 +67,24 @@ func TestGetInfoForBytes(t *testing.T) { } else if info.Size != 38689 { t.Fatalf("Got incorrect size: %v", info.Size) } else if info.Extension != "gif" { - t.Fatalf("Git incorrect file extension: %v", info.Extension) + t.Fatalf("Got incorrect file extension: %v", info.Extension) } else if info.MimeType != "image/gif" { t.Fatalf("Got incorrect mime type: %v", info.MimeType) } else if info.HasPreviewImage { t.Fatalf("Got HasPreviewImage = true for animated gif") } + + if info, err := GetInfoForBytes("filewithoutextension", fakeFile); err != nil { + t.Fatal(err) + } else if info.Filename != "filewithoutextension" { + t.Fatalf("Got incorrect filename: %v", info.Filename) + } else if info.Size != 1000 { + t.Fatalf("Got incorrect size: %v", info.Size) + } else if info.Extension != "" { + t.Fatalf("Got incorrect file extension: %v", info.Extension) + } else if info.MimeType != "" { + t.Fatalf("Got incorrect mime type: %v", info.MimeType) + } else if info.HasPreviewImage { + t.Fatalf("Got HasPreviewImage = true for non-image file") + } } |