mirror of
https://github.com/fleetdm/fleet
synced 2026-05-15 13:08:42 +00:00
b2efc9f51c
* Make redis conn timeout and keep alive configurable * Document new configs * Correct config name
45 lines
953 B
Go
45 lines
953 B
Go
package pubsub
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/fleetdm/fleet/v4/server/datastore/redis"
|
|
redigo "github.com/gomodule/redigo/redis"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func SetupRedisForTest(t *testing.T, cluster bool) (store *redisQueryResults, teardown func()) {
|
|
var (
|
|
addr = "127.0.0.1:"
|
|
password = ""
|
|
database = 0
|
|
useTLS = false
|
|
dupResults = false
|
|
port = "6379"
|
|
)
|
|
if cluster {
|
|
port = "7001"
|
|
}
|
|
addr += port
|
|
|
|
pool, err := redis.NewRedisPool(addr, password, database, useTLS, 5*time.Second, 10*time.Second)
|
|
require.NoError(t, err)
|
|
store = NewRedisQueryResults(pool, dupResults)
|
|
|
|
conn := store.pool.Get()
|
|
defer conn.Close()
|
|
_, err = conn.Do("PING")
|
|
require.Nil(t, err)
|
|
|
|
teardown = func() {
|
|
err := redis.EachRedisNode(store.pool, func(conn redigo.Conn) error {
|
|
_, err := conn.Do("FLUSHDB")
|
|
return err
|
|
})
|
|
require.NoError(t, err)
|
|
store.pool.Close()
|
|
}
|
|
|
|
return store, teardown
|
|
}
|