small refactorings of test code and some methods

This commit is contained in:
David Christofas
2020-12-18 14:12:10 +01:00
parent 738a32f9b1
commit cf69537d1b
5 changed files with 127 additions and 97 deletions

View File

@@ -60,9 +60,7 @@ func EncoderForType(fileType string) Encoder {
switch strings.ToLower(fileType) {
case "png":
return PngEncoder{}
case "jpg":
fallthrough
case "jpeg":
case "jpg", "jpeg":
return JpegEncoder{}
default:
return nil

View File

@@ -0,0 +1,22 @@
package thumbnail
import "testing"
func TestEncoderForType(t *testing.T) {
table := map[string]Encoder{
"jpg": JpegEncoder{},
"JPG": JpegEncoder{},
"jpeg": JpegEncoder{},
"JPEG": JpegEncoder{},
"png": PngEncoder{},
"PNG": PngEncoder{},
"invalid": nil,
}
for k, v := range table {
e := EncoderForType(k)
if e != v {
t.Fail()
}
}
}