diff options
Diffstat (limited to 'backend/internal')
| -rw-r--r-- | backend/internal/redis/taskqueue/queue_test.go | 18 |
1 files changed, 15 insertions, 3 deletions
diff --git a/backend/internal/redis/taskqueue/queue_test.go b/backend/internal/redis/taskqueue/queue_test.go index b54d22a..aa817c5 100644 --- a/backend/internal/redis/taskqueue/queue_test.go +++ b/backend/internal/redis/taskqueue/queue_test.go @@ -3,6 +3,7 @@ package taskqueue import ( "context" "errors" + "os" "testing" "time" @@ -11,12 +12,23 @@ import ( "github.com/stretchr/testify/require" ) +func getRedisClient() *redis.Client { + addr := os.Getenv("REDIS_ADDR") + if addr == "" { + addr = "localhost:6379" + } + + return redis.NewClient(&redis.Options{ + Addr: addr, + }) +} + func TestTaskQueue(t *testing.T) { if testing.Short() { t.Skip() } - client := redis.NewClient(new(redis.Options)) + client := getRedisClient() defer func(client *redis.Client) { _ = client.Close() }(client) @@ -202,7 +214,7 @@ func TestTaskQueue_List(t *testing.T) { t.Skip() } - client := redis.NewClient(new(redis.Options)) + client := getRedisClient() defer func(client *redis.Client) { _ = client.Close() }(client) @@ -371,7 +383,7 @@ func TestTaskQueue_Return(t *testing.T) { t.Skip() } - client := redis.NewClient(new(redis.Options)) + client := getRedisClient() defer func(client *redis.Client) { _ = client.Close() }(client) |