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.
85 lines
1.9 KiB
85 lines
1.9 KiB
package memcache
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/go-kratos/kratos/pkg/container/pool"
|
|
xtime "github.com/go-kratos/kratos/pkg/time"
|
|
)
|
|
|
|
var testConnASCII Conn
|
|
var testMemcache *Memcache
|
|
var testPool *Pool
|
|
var testMemcacheAddr string
|
|
|
|
func setupTestConnASCII(addr string) {
|
|
var err error
|
|
cnop := DialConnectTimeout(time.Duration(2 * time.Second))
|
|
rdop := DialReadTimeout(time.Duration(2 * time.Second))
|
|
wrop := DialWriteTimeout(time.Duration(2 * time.Second))
|
|
testConnASCII, err = Dial("tcp", addr, cnop, rdop, wrop)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
testConnASCII.Delete("test")
|
|
testConnASCII.Delete("test1")
|
|
testConnASCII.Delete("test2")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func setupTestMemcache(addr string) {
|
|
testConfig := &Config{
|
|
Config: &pool.Config{
|
|
Active: 10,
|
|
Idle: 10,
|
|
IdleTimeout: xtime.Duration(time.Second),
|
|
WaitTimeout: xtime.Duration(time.Second),
|
|
Wait: false,
|
|
},
|
|
Addr: addr,
|
|
Proto: "tcp",
|
|
DialTimeout: xtime.Duration(time.Second),
|
|
ReadTimeout: xtime.Duration(time.Second),
|
|
WriteTimeout: xtime.Duration(time.Second),
|
|
}
|
|
testMemcache = New(testConfig)
|
|
}
|
|
|
|
func setupTestPool(addr string) {
|
|
config := &Config{
|
|
Name: "test",
|
|
Proto: "tcp",
|
|
Addr: addr,
|
|
DialTimeout: xtime.Duration(time.Second),
|
|
ReadTimeout: xtime.Duration(time.Second),
|
|
WriteTimeout: xtime.Duration(time.Second),
|
|
}
|
|
config.Config = &pool.Config{
|
|
Active: 10,
|
|
Idle: 5,
|
|
IdleTimeout: xtime.Duration(90 * time.Second),
|
|
}
|
|
testPool = NewPool(config)
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
testMemcacheAddr = os.Getenv("TEST_MEMCACHE_ADDR")
|
|
if testExampleAddr == "" {
|
|
log.Print("TEST_MEMCACHE_ADDR not provide skip test.")
|
|
// ignored test.
|
|
os.Exit(0)
|
|
}
|
|
setupTestConnASCII(testMemcacheAddr)
|
|
setupTestMemcache(testMemcacheAddr)
|
|
setupTestPool(testMemcacheAddr)
|
|
// TODO: add setupexample?
|
|
testExampleAddr = testMemcacheAddr
|
|
|
|
ret := m.Run()
|
|
os.Exit(ret)
|
|
}
|
|
|