mirror of
https://github.com/woodchen-ink/webp_server_go.git
synced 2025-07-18 13:42:02 +08:00
35 lines
681 B
Go
35 lines
681 B
Go
// webp_server_go - prefetch_test.go
|
|
// 2020-11-10 09:27
|
|
// Benny <benny.think@gmail.com>
|
|
|
|
package main
|
|
|
|
import (
|
|
"github.com/stretchr/testify/assert"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestPrefetchImages(t *testing.T) {
|
|
// single thread
|
|
fp := "./prefetch"
|
|
_ = os.Mkdir(fp, 0755)
|
|
prefetchImages("./pics", "./prefetch", "80")
|
|
count := fileCount("./prefetch")
|
|
assert.Equal(t, 8, count)
|
|
_ = os.RemoveAll(fp)
|
|
|
|
// concurrency
|
|
jobs = 2
|
|
_ = os.Mkdir(fp, 0755)
|
|
prefetchImages("./pics", "./prefetch", "80")
|
|
count = fileCount("./prefetch")
|
|
assert.Equal(t, 6, count)
|
|
_ = os.RemoveAll(fp)
|
|
}
|
|
|
|
func TestBadPrefetch(t *testing.T) {
|
|
jobs = 1
|
|
prefetchImages("./pics2", "./prefetch", "80")
|
|
}
|