mirror of
https://github.com/woodchen-ink/webp_server_go.git
synced 2025-07-18 05:32:02 +08:00
67d40b4397
* use GitHub Actions as CI * bump versions * Makefile and Dockerfile improvements * upgrade logrus
76 lines
1.8 KiB
Go
76 lines
1.8 KiB
Go
// webp_server_go - webp-server_test
|
|
// 2020-11-10 09:41
|
|
// Benny <benny.think@gmail.com>
|
|
|
|
package main
|
|
|
|
import (
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/stretchr/testify/assert"
|
|
"net"
|
|
"os"
|
|
"runtime"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// due to test limit, we can't test for cli param part.
|
|
|
|
func TestLoadConfig(t *testing.T) {
|
|
c := loadConfig("./config.json")
|
|
assert.Equal(t, "./exhaust", c.ExhaustPath)
|
|
assert.Equal(t, "127.0.0.1", c.Host)
|
|
assert.Equal(t, "3333", c.Port)
|
|
assert.Equal(t, float32(80), c.Quality)
|
|
assert.Equal(t, "./pics", c.ImgPath)
|
|
assert.Equal(t, []string{"jpg", "png", "jpeg", "bmp"}, c.AllowedTypes)
|
|
}
|
|
|
|
func TestDeferInit(t *testing.T) {
|
|
// test initial value
|
|
assert.Equal(t, "", configPath)
|
|
assert.False(t, prefetch)
|
|
assert.Equal(t, false, dumpSystemd)
|
|
assert.Equal(t, false, dumpConfig)
|
|
assert.False(t, verboseMode)
|
|
}
|
|
|
|
func TestMainFunction(t *testing.T) {
|
|
// first test verbose mode
|
|
assert.False(t, verboseMode)
|
|
assert.Equal(t, log.GetLevel(), log.InfoLevel)
|
|
os.Args = append(os.Args, "-v", "-prefetch")
|
|
|
|
// run main function
|
|
go main()
|
|
time.Sleep(time.Second * 5)
|
|
// verbose, prefetch
|
|
assert.Equal(t, log.GetLevel(), log.DebugLevel)
|
|
assert.True(t, verboseMode)
|
|
assert.True(t, prefetch)
|
|
|
|
// test read config value
|
|
assert.Equal(t, "config.json", configPath)
|
|
assert.Equal(t, runtime.NumCPU(), jobs)
|
|
assert.Equal(t, false, dumpSystemd)
|
|
assert.Equal(t, false, dumpConfig)
|
|
|
|
// test port
|
|
conn, err := net.DialTimeout("tcp", net.JoinHostPort("127.0.0.1", "3333"), time.Second*2)
|
|
assert.Nil(t, err)
|
|
assert.NotNil(t, conn)
|
|
}
|
|
|
|
func TestProxySwitch(t *testing.T) {
|
|
// real proxy mode
|
|
assert.False(t, proxyMode)
|
|
config.ImgPath = "https://z.cn"
|
|
switchProxyMode()
|
|
assert.True(t, proxyMode)
|
|
|
|
// normal
|
|
config.ImgPath = os.TempDir()
|
|
switchProxyMode()
|
|
assert.False(t, proxyMode)
|
|
}
|