mirror of
https://github.com/woodchen-ink/webp_server_go.git
synced 2025-07-18 05:32:02 +08:00
43c275e3ec
* WIP JXL * Fix test * Tries to fix autobuild * Tries to fix autobuild * Add setup go in codeql * Bump actions version * Do not print curl output in CI * Do not print curl output in CI * Remove Metadata on RAW image * Update sample config * better loop * Prefetch should also respect AllowedType * Better Export params and UA handle * Only do conversion on supported formats * CONVERT_TYPES default to webp only * CONVERT_TYPES default to webp only * Add GIF to AllowedTypes * Update README
111 lines
2.5 KiB
Go
111 lines
2.5 KiB
Go
package helper
|
|
|
|
import (
|
|
"testing"
|
|
"webp_server_go/config"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
config.ConfigPath = "../config.json"
|
|
config.LoadConfig()
|
|
m.Run()
|
|
config.ConfigPath = "config.json"
|
|
}
|
|
|
|
func TestFileCount(t *testing.T) {
|
|
// test helper dir
|
|
count := FileCount("./")
|
|
assert.Equal(t, int64(4), count)
|
|
}
|
|
|
|
func TestImageExists(t *testing.T) {
|
|
t.Run("file not exists", func(t *testing.T) {
|
|
assert.False(t, ImageExists("dgyuaikdsa"))
|
|
})
|
|
|
|
// TODO: how to test lock?
|
|
|
|
t.Run("test dir", func(t *testing.T) {
|
|
assert.False(t, ImageExists("/tmp"))
|
|
})
|
|
|
|
t.Run("test file", func(t *testing.T) {
|
|
assert.True(t, ImageExists("./helper_test.go"))
|
|
})
|
|
}
|
|
|
|
func TestCheckAllowedType(t *testing.T) {
|
|
t.Run("not allowed type", func(t *testing.T) {
|
|
assert.False(t, CheckAllowedType("./helper_test.go"))
|
|
})
|
|
|
|
t.Run("allowed type", func(t *testing.T) {
|
|
assert.True(t, CheckAllowedType("test.jpg"))
|
|
})
|
|
}
|
|
|
|
func TestGuessSupportedFormat(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
userAgent string
|
|
accept string
|
|
expected map[string]bool
|
|
}{
|
|
{
|
|
name: "WebP/AVIF/JXL Supported",
|
|
userAgent: "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/17.4 Safari/605.1.15", // iPad
|
|
accept: "image/webp, image/avif",
|
|
expected: map[string]bool{
|
|
"raw": true,
|
|
"webp": true,
|
|
"avif": true,
|
|
"jxl": true,
|
|
},
|
|
},
|
|
{
|
|
name: "WebP/AVIF Supported",
|
|
userAgent: "iPhone OS 16",
|
|
accept: "image/webp, image/png",
|
|
expected: map[string]bool{
|
|
"raw": true,
|
|
"webp": true,
|
|
"avif": true,
|
|
"jxl": false,
|
|
},
|
|
},
|
|
{
|
|
name: "Both Supported",
|
|
userAgent: "iPhone OS 16",
|
|
accept: "image/webp, image/avif",
|
|
expected: map[string]bool{"raw": true, "webp": true, "avif": true, "jxl": false},
|
|
},
|
|
{
|
|
name: "No Supported Formats",
|
|
userAgent: "Unknown OS",
|
|
accept: "image/jpeg, image/gif",
|
|
expected: map[string]bool{"raw": true, "webp": false, "avif": false, "jxl": false},
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
header := &fasthttp.RequestHeader{}
|
|
header.Set("user-agent", test.userAgent)
|
|
header.Set("accept", test.accept)
|
|
|
|
result := GuessSupportedFormat(header)
|
|
|
|
if len(result) != len(test.expected) {
|
|
t.Errorf("Expected %v, but got %v", test.expected, result)
|
|
}
|
|
|
|
for k, v := range test.expected {
|
|
assert.Equal(t, v, result[k])
|
|
}
|
|
})
|
|
}
|
|
}
|