Create a testStore per test

This commit is contained in:
Juan Calderon-Perez
2023-10-29 20:24:11 -04:00
parent 2143a67c8d
commit 0b9f4aebc8
2 changed files with 48 additions and 16 deletions

View File

@@ -121,12 +121,12 @@ func (s *Storage) Conn() redis.UniversalClient {
func (s *Storage) Keys() ([][]byte, error) { func (s *Storage) Keys() ([][]byte, error) {
var keys [][]byte var keys [][]byte
var cursor uint64 var cursor uint64
var err error
for { for {
var batch []string var batch []string
var err error
batch, cursor, err = s.db.Scan(context.Background(), cursor, "*", 10).Result() if batch, cursor, err = s.db.Scan(context.Background(), cursor, "*", 10).Result(); err != nil {
if err != nil {
return nil, err return nil, err
} }

View File

@@ -9,12 +9,11 @@ import (
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
) )
var testStore = New(Config{
Reset: true,
})
func Test_Redis_Set(t *testing.T) { func Test_Redis_Set(t *testing.T) {
var ( var (
testStore = New(Config{
Reset: true,
})
key = "john" key = "john"
val = []byte("doe") val = []byte("doe")
) )
@@ -25,6 +24,9 @@ func Test_Redis_Set(t *testing.T) {
func Test_Redis_Set_Override(t *testing.T) { func Test_Redis_Set_Override(t *testing.T) {
var ( var (
testStore = New(Config{
Reset: true,
})
key = "john" key = "john"
val = []byte("doe") val = []byte("doe")
) )
@@ -42,6 +44,9 @@ func Test_Redis_Set_Override(t *testing.T) {
func Test_Redis_Get(t *testing.T) { func Test_Redis_Get(t *testing.T) {
var ( var (
testStore = New(Config{
Reset: true,
})
key = "john" key = "john"
val = []byte("doe") val = []byte("doe")
) )
@@ -60,6 +65,9 @@ func Test_Redis_Get(t *testing.T) {
func Test_Redis_Expiration(t *testing.T) { func Test_Redis_Expiration(t *testing.T) {
var ( var (
testStore = New(Config{
Reset: true,
})
key = "john" key = "john"
val = []byte("doe") val = []byte("doe")
exp = 1 * time.Second exp = 1 * time.Second
@@ -80,6 +88,9 @@ func Test_Redis_Expiration(t *testing.T) {
} }
func Test_Redis_Get_NotExist(t *testing.T) { func Test_Redis_Get_NotExist(t *testing.T) {
testStore := New(Config{
Reset: true,
})
result, err := testStore.Get("notexist") result, err := testStore.Get("notexist")
require.NoError(t, err) require.NoError(t, err)
require.Zero(t, len(result)) require.Zero(t, len(result))
@@ -87,6 +98,9 @@ func Test_Redis_Get_NotExist(t *testing.T) {
func Test_Redis_Delete(t *testing.T) { func Test_Redis_Delete(t *testing.T) {
var ( var (
testStore = New(Config{
Reset: true,
})
key = "john" key = "john"
val = []byte("doe") val = []byte("doe")
) )
@@ -107,6 +121,9 @@ func Test_Redis_Delete(t *testing.T) {
} }
func Test_Redis_Reset(t *testing.T) { func Test_Redis_Reset(t *testing.T) {
testStore := New(Config{
Reset: true,
})
val := []byte("doe") val := []byte("doe")
err := testStore.Set("john1", val, 0) err := testStore.Set("john1", val, 0)
@@ -136,10 +153,16 @@ func Test_Redis_Reset(t *testing.T) {
} }
func Test_Redis_Close(t *testing.T) { func Test_Redis_Close(t *testing.T) {
testStore := New(Config{
Reset: true,
})
require.Nil(t, testStore.Close()) require.Nil(t, testStore.Close())
} }
func Test_Redis_Conn(t *testing.T) { func Test_Redis_Conn(t *testing.T) {
testStore := New(Config{
Reset: true,
})
require.True(t, testStore.Conn() != nil) require.True(t, testStore.Conn() != nil)
} }
@@ -204,7 +227,7 @@ func Test_Redis_Initalize_WithURL_TLS(t *testing.T) {
err = testStoreUrl.Delete(key) err = testStoreUrl.Delete(key)
require.NoError(t, err) require.NoError(t, err)
keys, err := testStore.Keys() keys, err := testStoreUrl.Keys()
require.NoError(t, err) require.NoError(t, err)
require.Zero(t, keys) require.Zero(t, keys)
@@ -250,7 +273,7 @@ func Test_Redis_Initalize_WithURL_TLS_Verify(t *testing.T) {
err = testStoreUrl.Delete(key) err = testStoreUrl.Delete(key)
require.NoError(t, err) require.NoError(t, err)
keys, err := testStore.Keys() keys, err := testStoreUrl.Keys()
require.NoError(t, err) require.NoError(t, err)
require.Zero(t, keys) require.Zero(t, keys)
@@ -277,7 +300,7 @@ func Test_Redis_Initalize_With_Secure_URL(t *testing.T) {
err = testStoreUrl.Delete(key) err = testStoreUrl.Delete(key)
require.NoError(t, err) require.NoError(t, err)
keys, err := testStore.Keys() keys, err := testStoreUrl.Keys()
require.NoError(t, err) require.NoError(t, err)
require.Zero(t, keys) require.Zero(t, keys)
@@ -305,7 +328,7 @@ func Test_Redis_Universal_Addrs(t *testing.T) {
err = testStoreUniversal.Delete(key) err = testStoreUniversal.Delete(key)
require.NoError(t, err) require.NoError(t, err)
keys, err := testStore.Keys() keys, err := testStoreUniversal.Keys()
require.NoError(t, err) require.NoError(t, err)
require.Zero(t, keys) require.Zero(t, keys)
@@ -335,7 +358,7 @@ func Test_Redis_Universal_With_URL_Undefined(t *testing.T) {
err = testStoreUniversal.Delete(key) err = testStoreUniversal.Delete(key)
require.NoError(t, err) require.NoError(t, err)
keys, err := testStore.Keys() keys, err := testStoreUniversal.Keys()
require.NoError(t, err) require.NoError(t, err)
require.Zero(t, keys) require.Zero(t, keys)
@@ -365,7 +388,7 @@ func Test_Redis_Universal_With_URL_Defined(t *testing.T) {
err = testStoreUniversal.Delete(key) err = testStoreUniversal.Delete(key)
require.NoError(t, err) require.NoError(t, err)
keys, err := testStore.Keys() keys, err := testStoreUniversal.Keys()
require.NoError(t, err) require.NoError(t, err)
require.Zero(t, keys) require.Zero(t, keys)
@@ -396,7 +419,7 @@ func Test_Redis_Universal_With_HostPort(t *testing.T) {
err = testStoreUniversal.Delete(key) err = testStoreUniversal.Delete(key)
require.NoError(t, err) require.NoError(t, err)
keys, err := testStore.Keys() keys, err := testStoreUniversal.Keys()
require.NoError(t, err) require.NoError(t, err)
require.Zero(t, keys) require.Zero(t, keys)
@@ -428,7 +451,7 @@ func Test_Redis_Universal_With_HostPort_And_URL(t *testing.T) {
err = testStoreUniversal.Delete(key) err = testStoreUniversal.Delete(key)
require.NoError(t, err) require.NoError(t, err)
keys, err := testStore.Keys() keys, err := testStoreUniversal.Keys()
require.NoError(t, err) require.NoError(t, err)
require.Zero(t, keys) require.Zero(t, keys)
@@ -462,7 +485,7 @@ func Test_Redis_Cluster(t *testing.T) {
err = testStoreUniversal.Delete(key) err = testStoreUniversal.Delete(key)
require.NoError(t, err) require.NoError(t, err)
keys, err := testStore.Keys() keys, err := testStoreUniversal.Keys()
require.NoError(t, err) require.NoError(t, err)
require.Zero(t, keys) require.Zero(t, keys)
@@ -470,6 +493,9 @@ func Test_Redis_Cluster(t *testing.T) {
} }
func Benchmark_Redis_Set(b *testing.B) { func Benchmark_Redis_Set(b *testing.B) {
testStore := New(Config{
Reset: true,
})
b.ReportAllocs() b.ReportAllocs()
b.ResetTimer() b.ResetTimer()
@@ -482,6 +508,9 @@ func Benchmark_Redis_Set(b *testing.B) {
} }
func Benchmark_Redis_Get(b *testing.B) { func Benchmark_Redis_Get(b *testing.B) {
testStore := New(Config{
Reset: true,
})
err := testStore.Set("john", []byte("doe"), 0) err := testStore.Set("john", []byte("doe"), 0)
require.NoError(b, err) require.NoError(b, err)
@@ -496,6 +525,9 @@ func Benchmark_Redis_Get(b *testing.B) {
} }
func Benchmark_Redis_SetAndDelete(b *testing.B) { func Benchmark_Redis_SetAndDelete(b *testing.B) {
testStore := New(Config{
Reset: true,
})
b.ReportAllocs() b.ReportAllocs()
b.ResetTimer() b.ResetTimer()