mirror of
https://github.com/crawlab-team/crawlab.git
synced 2026-01-22 17:31:03 +01:00
92 lines
1.5 KiB
Go
92 lines
1.5 KiB
Go
package test
|
|
|
|
import (
|
|
"github.com/crawlab-team/crawlab/db"
|
|
"github.com/crawlab-team/crawlab/db/redis"
|
|
"testing"
|
|
)
|
|
|
|
func init() {
|
|
var err error
|
|
T, err = NewTest()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
type Test struct {
|
|
client db.RedisClient
|
|
TestCollection string
|
|
TestMessage string
|
|
TestMessages []string
|
|
TestMessagesMap map[string]string
|
|
TestKeysAlpha []string
|
|
TestKeysBeta []string
|
|
TestLockKey string
|
|
}
|
|
|
|
func (t *Test) Setup(t2 *testing.T) {
|
|
t2.Cleanup(t.Cleanup)
|
|
}
|
|
|
|
func (t *Test) Cleanup() {
|
|
keys, _ := t.client.AllKeys()
|
|
for _, key := range keys {
|
|
_ = t.client.Del(key)
|
|
}
|
|
}
|
|
|
|
var T *Test
|
|
|
|
func NewTest() (t *Test, err error) {
|
|
// test
|
|
t = &Test{}
|
|
|
|
// client
|
|
t.client, err = redis.GetRedisClient()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// test collection
|
|
t.TestCollection = "test_collection"
|
|
|
|
// test message
|
|
t.TestMessage = "this is a test message"
|
|
|
|
// test messages
|
|
t.TestMessages = []string{
|
|
"test message 1",
|
|
"test message 2",
|
|
"test message 3",
|
|
}
|
|
|
|
// test messages map
|
|
t.TestMessagesMap = map[string]string{
|
|
"test key 1": "test value 1",
|
|
"test key 2": "test value 2",
|
|
"test key 3": "test value 3",
|
|
}
|
|
|
|
// test keys alpha
|
|
t.TestKeysAlpha = []string{
|
|
"test key alpha 1",
|
|
"test key alpha 2",
|
|
"test key alpha 3",
|
|
}
|
|
|
|
// test keys beta
|
|
t.TestKeysBeta = []string{
|
|
"test key beta 1",
|
|
"test key beta 2",
|
|
"test key beta 3",
|
|
"test key beta 4",
|
|
"test key beta 5",
|
|
}
|
|
|
|
// test lock key
|
|
t.TestLockKey = "test lock key"
|
|
|
|
return t, nil
|
|
}
|