mirror of
https://github.com/woodchen-ink/webp_server_go.git
synced 2025-07-18 13:42:02 +08:00
67d40b4397
* use GitHub Actions as CI * bump versions * Makefile and Dockerfile improvements * upgrade logrus
67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/stretchr/testify/assert"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func walker() []string {
|
|
var list []string
|
|
_ = filepath.Walk("./pics", func(p string, info os.FileInfo, err error) error {
|
|
if !info.IsDir() && !strings.HasPrefix(path.Base(p), ".") {
|
|
list = append(list, p)
|
|
}
|
|
return nil
|
|
})
|
|
return list
|
|
}
|
|
|
|
func TestWebPEncoder(t *testing.T) {
|
|
// Go through every files
|
|
var dest = "/tmp/test-result"
|
|
var target = walker()
|
|
for _, f := range target {
|
|
runEncoder(t, f, dest)
|
|
}
|
|
_ = os.Remove(dest)
|
|
}
|
|
|
|
func TestAvifEncoder(t *testing.T) {
|
|
// Only one file: img_over_16383px.jpg might cause memory issues on CI environment
|
|
var dest = "/tmp/test-result"
|
|
avifEncoder("./pics/big.jpg", dest, 80)
|
|
assertType(t, dest, "image/avif")
|
|
}
|
|
|
|
func TestNonExistImage(t *testing.T) {
|
|
var dest = "/tmp/test-result"
|
|
webpEncoder("./pics/empty.jpg", dest, 80)
|
|
avifEncoder("./pics/empty.jpg", dest, 80)
|
|
}
|
|
|
|
func TestConvertFail(t *testing.T) {
|
|
var dest = "/tmp/test-result"
|
|
webpEncoder("./pics/webp_server.jpg", dest, -1)
|
|
avifEncoder("./pics/webp_server.jpg", dest, -1)
|
|
}
|
|
|
|
func runEncoder(t *testing.T, file string, dest string) {
|
|
if file == "pics/empty.jpg" {
|
|
t.Log("Empty file, that's okay.")
|
|
}
|
|
webpEncoder(file, dest, 80)
|
|
assertType(t, dest, "image/webp")
|
|
|
|
}
|
|
|
|
func assertType(t *testing.T, dest, mime string) {
|
|
data, _ := ioutil.ReadFile(dest)
|
|
types := getFileContentType(data[:512])
|
|
assert.Equalf(t, mime, types, "File %s should be %s", dest, mime)
|
|
}
|