You've already forked img-proxy-url-generator
61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
package generator
|
|
|
|
import (
|
|
"github.com/siteworxpro/img-proxy-url-generator/config"
|
|
"testing"
|
|
)
|
|
|
|
func TestGenerator_StringToFormat(t *testing.T) {
|
|
g, err := NewGenerator(&config.Config{
|
|
Generator: &config.GeneratorConfig{
|
|
Key: []byte("2c90317177aa7a3c44fa6804bf9bf466930f36ac9262bfdae972e836a9f83d239fd6bcee0c91a29ada58cc7329c787f35d2309f0984f2fd315e2c27bac8ac247"),
|
|
Salt: []byte("2777def3372a385f4aa7e62b2b431927"),
|
|
},
|
|
})
|
|
|
|
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.Config{
|
|
Generator: &config.GeneratorConfig{
|
|
Key: []byte("f2d1f37016b7d12ab27b25377f39fd84e2d3368472ff096261ce7ac3e8490af429d43803836ad6a42a3bd9fb859a38137173619cb00bcb6fe3870e3feab2b764"),
|
|
Salt: []byte("919ed8813f76abfd42968b10e05258db"),
|
|
},
|
|
})
|
|
|
|
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
|
|
|
|
}
|
|
}
|