You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.4 KiB
67 lines
1.4 KiB
package redis
|
|
|
|
import (
|
|
"flag"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/bilibili/kratos/pkg/container/pool"
|
|
"github.com/bilibili/kratos/pkg/testing/lich"
|
|
xtime "github.com/bilibili/kratos/pkg/time"
|
|
)
|
|
|
|
var (
|
|
testRedisAddr string
|
|
testPool *Pool
|
|
testConfig *Config
|
|
)
|
|
|
|
func setupTestConfig(addr string) {
|
|
c := getTestConfig(addr)
|
|
c.Config = &pool.Config{
|
|
Active: 20,
|
|
Idle: 2,
|
|
IdleTimeout: xtime.Duration(90 * time.Second),
|
|
}
|
|
testConfig = c
|
|
}
|
|
|
|
func getTestConfig(addr string) *Config {
|
|
return &Config{
|
|
Name: "test",
|
|
Proto: "tcp",
|
|
Addr: addr,
|
|
DialTimeout: xtime.Duration(time.Second),
|
|
ReadTimeout: xtime.Duration(time.Second),
|
|
WriteTimeout: xtime.Duration(time.Second),
|
|
}
|
|
}
|
|
|
|
func setupTestPool() {
|
|
testPool = NewPool(testConfig)
|
|
}
|
|
|
|
// DialDefaultServer starts the test server if not already started and dials a
|
|
// connection to the server.
|
|
func DialDefaultServer() (Conn, error) {
|
|
c, err := Dial("tcp", testRedisAddr, DialReadTimeout(1*time.Second), DialWriteTimeout(1*time.Second))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
c.Do("FLUSHDB")
|
|
return c, nil
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
flag.Set("f", "./test/docker-compose.yaml")
|
|
if err := lich.Setup(); err != nil {
|
|
panic(err)
|
|
}
|
|
defer lich.Teardown()
|
|
testRedisAddr = "localhost:6379"
|
|
setupTestConfig(testRedisAddr)
|
|
setupTestPool()
|
|
ret := m.Run()
|
|
os.Exit(ret)
|
|
}
|
|
|