diff --git a/clickhouse/clickhouse_test.go b/clickhouse/clickhouse_test.go index d9e09903..7a63975c 100644 --- a/clickhouse/clickhouse_test.go +++ b/clickhouse/clickhouse_test.go @@ -67,11 +67,12 @@ func getTestConnection(t testing.TB, cfg Config) (*Storage, error) { } func Test_Connection(t *testing.T) { - _, err := getTestConnection(t, Config{ + client, err := getTestConnection(t, Config{ Engine: Memory, Table: "test_table", Clean: true, }) + defer client.Close() require.NoError(t, err) } diff --git a/couchbase/couchbase_test.go b/couchbase/couchbase_test.go index f22b71ab..7b3e85bd 100644 --- a/couchbase/couchbase_test.go +++ b/couchbase/couchbase_test.go @@ -56,6 +56,7 @@ func newTestStore(t testing.TB) (*Storage, error) { func TestSetCouchbase_ShouldReturnNoError(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() err = testStore.Set("test", []byte("test"), 0) @@ -65,6 +66,7 @@ func TestSetCouchbase_ShouldReturnNoError(t *testing.T) { func TestGetCouchbase_ShouldReturnNil_WhenDocumentNotFound(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() val, err := testStore.Get("not_found_key") @@ -75,6 +77,7 @@ func TestGetCouchbase_ShouldReturnNil_WhenDocumentNotFound(t *testing.T) { func TestSetAndGet_GetShouldReturn_SetValueWithoutError(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() err = testStore.Set("test", []byte("fiber_test_value"), 0) require.NoError(t, err) @@ -88,6 +91,7 @@ func TestSetAndGet_GetShouldReturn_SetValueWithoutError(t *testing.T) { func TestSetAndGet_GetShouldReturnNil_WhenTTLExpired(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() err = testStore.Set("test", []byte("fiber_test_value"), 3*time.Second) require.NoError(t, err) @@ -103,6 +107,7 @@ func TestSetAndGet_GetShouldReturnNil_WhenTTLExpired(t *testing.T) { func TestSetAndDelete_DeleteShouldReturn_NoError(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() err = testStore.Set("test", []byte("fiber_test_value"), 0) require.NoError(t, err) @@ -117,6 +122,7 @@ func TestSetAndDelete_DeleteShouldReturn_NoError(t *testing.T) { func TestSetAndReset_ResetShouldReturn_NoError(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() err = testStore.Set("test", []byte("fiber_test_value"), 0) require.NoError(t, err) @@ -131,6 +137,7 @@ func TestSetAndReset_ResetShouldReturn_NoError(t *testing.T) { func TestClose_CloseShouldReturn_NoError(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() err = testStore.Close() require.NoError(t, err) @@ -139,6 +146,7 @@ func TestClose_CloseShouldReturn_NoError(t *testing.T) { func TestGetConn_ReturnsNotNil(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() require.True(t, testStore.Conn() != nil) } @@ -146,6 +154,7 @@ func TestGetConn_ReturnsNotNil(t *testing.T) { func Benchmark_Couchbase_Set(b *testing.B) { testStore, err := newTestStore(b) require.NoError(b, err) + defer testStore.Close() b.ReportAllocs() b.ResetTimer() @@ -160,6 +169,7 @@ func Benchmark_Couchbase_Set(b *testing.B) { func Benchmark_Couchbase_Get(b *testing.B) { testStore, err := newTestStore(b) require.NoError(b, err) + defer testStore.Close() err = testStore.Set("john", []byte("doe"), 0) require.NoError(b, err) @@ -177,6 +187,7 @@ func Benchmark_Couchbase_Get(b *testing.B) { func Benchmark_Couchbase_SetAndDelete(b *testing.B) { testStore, err := newTestStore(b) require.NoError(b, err) + defer testStore.Close() b.ReportAllocs() b.ResetTimer() diff --git a/minio/minio_test.go b/minio/minio_test.go index c2661647..a2ab8605 100644 --- a/minio/minio_test.go +++ b/minio/minio_test.go @@ -64,6 +64,7 @@ func Test_Get(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() err = testStore.Set(key, val, 0) require.NoError(t, err) @@ -84,6 +85,7 @@ func Test_Get_Empty_Key(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() _, err = testStore.Get(key) require.Error(t, err) @@ -97,6 +99,7 @@ func Test_Get_Not_Exists_Key(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() _, err = testStore.Get(key) require.Error(t, err) @@ -110,6 +113,7 @@ func Test_Get_Not_Exists_Bucket(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() // random bucket name testStore.cfg.Bucket = strconv.FormatInt(time.Now().UnixMicro(), 10) @@ -128,6 +132,7 @@ func Test_Set(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() err = testStore.Set(key, val, 0) require.NoError(t, err) @@ -141,6 +146,7 @@ func Test_Set_Empty_Key(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() err = testStore.Set(key, val, 0) @@ -157,6 +163,7 @@ func Test_Set_Not_Exists_Bucket(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() // random bucket name testStore.cfg.Bucket = strconv.FormatInt(time.Now().UnixMicro(), 10) @@ -174,6 +181,7 @@ func Test_Delete(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() err = testStore.Set(key, val, 0) require.NoError(t, err) @@ -190,6 +198,7 @@ func Test_Delete_Empty_Key(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() err = testStore.Set(key, val, 0) require.Error(t, err) @@ -203,6 +212,7 @@ func Test_Delete_Not_Exists_Bucket(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() // random bucket name testStore.cfg.Bucket = strconv.FormatInt(time.Now().UnixMicro(), 10) @@ -220,6 +230,7 @@ func Test_Reset(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() err = testStore.Set("john1", val, 0) require.NoError(t, err) @@ -238,6 +249,7 @@ func Test_Reset(t *testing.T) { func Test_Close(t *testing.T) { testStore, err := newTestStore(t) require.NoError(t, err) + defer testStore.Close() require.NoError(t, testStore.Close()) } @@ -248,6 +260,7 @@ func Benchmark_Minio_Set(b *testing.B) { testStore, err := newTestStore(b) require.NoError(b, err) + defer testStore.Close() for i := 0; i < b.N; i++ { err = testStore.Set("john", []byte("doe"), 0) @@ -259,6 +272,7 @@ func Benchmark_Minio_Set(b *testing.B) { func Benchmark_Minio_Get(b *testing.B) { testStore, err := newTestStore(b) require.NoError(b, err) + defer testStore.Close() err = testStore.Set("john", []byte("doe"), 0) require.NoError(b, err) @@ -276,6 +290,7 @@ func Benchmark_Minio_Get(b *testing.B) { func Benchmark_Minio_SetAndDelete(b *testing.B) { testStore, err := newTestStore(b) require.NoError(b, err) + defer testStore.Close() b.ReportAllocs() b.ResetTimer()