You've already forked img-proxy-url-generator
46 lines
759 B
Go
46 lines
759 B
Go
package generator
|
|
|
|
import "testing"
|
|
|
|
func TestGenerator_StringToFormat(t *testing.T) {
|
|
g, err := NewGenerator(Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
formats := []string{"jpg", "jpeg", "png", "gif", "bmp", "webp", "ico", ""}
|
|
|
|
for _, format := range formats {
|
|
format, err := g.StringToFormat(format)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !isFormat(format) {
|
|
t.Error("format not match")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGenerator_StringToFormatError(t *testing.T) {
|
|
g, err := NewGenerator(Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
_, err = g.StringToFormat("notaformat")
|
|
if err == nil {
|
|
t.Error("format did not return error")
|
|
}
|
|
}
|
|
|
|
func isFormat(t interface{}) bool {
|
|
switch t.(type) {
|
|
case Format:
|
|
return true
|
|
default:
|
|
return false
|
|
|
|
}
|
|
}
|