chore: close store for previous modules

This commit is contained in:
Manuel de la Peña
2024-09-18 16:56:00 +02:00
parent 4d7a172400
commit 75c8525a46
3 changed files with 28 additions and 1 deletions

View File

@@ -67,11 +67,12 @@ func getTestConnection(t testing.TB, cfg Config) (*Storage, error) {
} }
func Test_Connection(t *testing.T) { func Test_Connection(t *testing.T) {
_, err := getTestConnection(t, Config{ client, err := getTestConnection(t, Config{
Engine: Memory, Engine: Memory,
Table: "test_table", Table: "test_table",
Clean: true, Clean: true,
}) })
defer client.Close()
require.NoError(t, err) require.NoError(t, err)
} }

View File

@@ -56,6 +56,7 @@ func newTestStore(t testing.TB) (*Storage, error) {
func TestSetCouchbase_ShouldReturnNoError(t *testing.T) { func TestSetCouchbase_ShouldReturnNoError(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
err = testStore.Set("test", []byte("test"), 0) err = testStore.Set("test", []byte("test"), 0)
@@ -65,6 +66,7 @@ func TestSetCouchbase_ShouldReturnNoError(t *testing.T) {
func TestGetCouchbase_ShouldReturnNil_WhenDocumentNotFound(t *testing.T) { func TestGetCouchbase_ShouldReturnNil_WhenDocumentNotFound(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
val, err := testStore.Get("not_found_key") 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) { func TestSetAndGet_GetShouldReturn_SetValueWithoutError(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
err = testStore.Set("test", []byte("fiber_test_value"), 0) err = testStore.Set("test", []byte("fiber_test_value"), 0)
require.NoError(t, err) require.NoError(t, err)
@@ -88,6 +91,7 @@ func TestSetAndGet_GetShouldReturn_SetValueWithoutError(t *testing.T) {
func TestSetAndGet_GetShouldReturnNil_WhenTTLExpired(t *testing.T) { func TestSetAndGet_GetShouldReturnNil_WhenTTLExpired(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
err = testStore.Set("test", []byte("fiber_test_value"), 3*time.Second) err = testStore.Set("test", []byte("fiber_test_value"), 3*time.Second)
require.NoError(t, err) require.NoError(t, err)
@@ -103,6 +107,7 @@ func TestSetAndGet_GetShouldReturnNil_WhenTTLExpired(t *testing.T) {
func TestSetAndDelete_DeleteShouldReturn_NoError(t *testing.T) { func TestSetAndDelete_DeleteShouldReturn_NoError(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
err = testStore.Set("test", []byte("fiber_test_value"), 0) err = testStore.Set("test", []byte("fiber_test_value"), 0)
require.NoError(t, err) require.NoError(t, err)
@@ -117,6 +122,7 @@ func TestSetAndDelete_DeleteShouldReturn_NoError(t *testing.T) {
func TestSetAndReset_ResetShouldReturn_NoError(t *testing.T) { func TestSetAndReset_ResetShouldReturn_NoError(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
err = testStore.Set("test", []byte("fiber_test_value"), 0) err = testStore.Set("test", []byte("fiber_test_value"), 0)
require.NoError(t, err) require.NoError(t, err)
@@ -131,6 +137,7 @@ func TestSetAndReset_ResetShouldReturn_NoError(t *testing.T) {
func TestClose_CloseShouldReturn_NoError(t *testing.T) { func TestClose_CloseShouldReturn_NoError(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
err = testStore.Close() err = testStore.Close()
require.NoError(t, err) require.NoError(t, err)
@@ -139,6 +146,7 @@ func TestClose_CloseShouldReturn_NoError(t *testing.T) {
func TestGetConn_ReturnsNotNil(t *testing.T) { func TestGetConn_ReturnsNotNil(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
require.True(t, testStore.Conn() != nil) require.True(t, testStore.Conn() != nil)
} }
@@ -146,6 +154,7 @@ func TestGetConn_ReturnsNotNil(t *testing.T) {
func Benchmark_Couchbase_Set(b *testing.B) { func Benchmark_Couchbase_Set(b *testing.B) {
testStore, err := newTestStore(b) testStore, err := newTestStore(b)
require.NoError(b, err) require.NoError(b, err)
defer testStore.Close()
b.ReportAllocs() b.ReportAllocs()
b.ResetTimer() b.ResetTimer()
@@ -160,6 +169,7 @@ func Benchmark_Couchbase_Set(b *testing.B) {
func Benchmark_Couchbase_Get(b *testing.B) { func Benchmark_Couchbase_Get(b *testing.B) {
testStore, err := newTestStore(b) testStore, err := newTestStore(b)
require.NoError(b, err) require.NoError(b, err)
defer testStore.Close()
err = testStore.Set("john", []byte("doe"), 0) err = testStore.Set("john", []byte("doe"), 0)
require.NoError(b, err) require.NoError(b, err)
@@ -177,6 +187,7 @@ func Benchmark_Couchbase_Get(b *testing.B) {
func Benchmark_Couchbase_SetAndDelete(b *testing.B) { func Benchmark_Couchbase_SetAndDelete(b *testing.B) {
testStore, err := newTestStore(b) testStore, err := newTestStore(b)
require.NoError(b, err) require.NoError(b, err)
defer testStore.Close()
b.ReportAllocs() b.ReportAllocs()
b.ResetTimer() b.ResetTimer()

View File

@@ -64,6 +64,7 @@ func Test_Get(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
err = testStore.Set(key, val, 0) err = testStore.Set(key, val, 0)
require.NoError(t, err) require.NoError(t, err)
@@ -84,6 +85,7 @@ func Test_Get_Empty_Key(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
_, err = testStore.Get(key) _, err = testStore.Get(key)
require.Error(t, err) require.Error(t, err)
@@ -97,6 +99,7 @@ func Test_Get_Not_Exists_Key(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
_, err = testStore.Get(key) _, err = testStore.Get(key)
require.Error(t, err) require.Error(t, err)
@@ -110,6 +113,7 @@ func Test_Get_Not_Exists_Bucket(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
// random bucket name // random bucket name
testStore.cfg.Bucket = strconv.FormatInt(time.Now().UnixMicro(), 10) testStore.cfg.Bucket = strconv.FormatInt(time.Now().UnixMicro(), 10)
@@ -128,6 +132,7 @@ func Test_Set(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
err = testStore.Set(key, val, 0) err = testStore.Set(key, val, 0)
require.NoError(t, err) require.NoError(t, err)
@@ -141,6 +146,7 @@ func Test_Set_Empty_Key(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
err = testStore.Set(key, val, 0) err = testStore.Set(key, val, 0)
@@ -157,6 +163,7 @@ func Test_Set_Not_Exists_Bucket(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
// random bucket name // random bucket name
testStore.cfg.Bucket = strconv.FormatInt(time.Now().UnixMicro(), 10) testStore.cfg.Bucket = strconv.FormatInt(time.Now().UnixMicro(), 10)
@@ -174,6 +181,7 @@ func Test_Delete(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
err = testStore.Set(key, val, 0) err = testStore.Set(key, val, 0)
require.NoError(t, err) require.NoError(t, err)
@@ -190,6 +198,7 @@ func Test_Delete_Empty_Key(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
err = testStore.Set(key, val, 0) err = testStore.Set(key, val, 0)
require.Error(t, err) require.Error(t, err)
@@ -203,6 +212,7 @@ func Test_Delete_Not_Exists_Bucket(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
// random bucket name // random bucket name
testStore.cfg.Bucket = strconv.FormatInt(time.Now().UnixMicro(), 10) testStore.cfg.Bucket = strconv.FormatInt(time.Now().UnixMicro(), 10)
@@ -220,6 +230,7 @@ func Test_Reset(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
err = testStore.Set("john1", val, 0) err = testStore.Set("john1", val, 0)
require.NoError(t, err) require.NoError(t, err)
@@ -238,6 +249,7 @@ func Test_Reset(t *testing.T) {
func Test_Close(t *testing.T) { func Test_Close(t *testing.T) {
testStore, err := newTestStore(t) testStore, err := newTestStore(t)
require.NoError(t, err) require.NoError(t, err)
defer testStore.Close()
require.NoError(t, testStore.Close()) require.NoError(t, testStore.Close())
} }
@@ -248,6 +260,7 @@ func Benchmark_Minio_Set(b *testing.B) {
testStore, err := newTestStore(b) testStore, err := newTestStore(b)
require.NoError(b, err) require.NoError(b, err)
defer testStore.Close()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
err = testStore.Set("john", []byte("doe"), 0) 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) { func Benchmark_Minio_Get(b *testing.B) {
testStore, err := newTestStore(b) testStore, err := newTestStore(b)
require.NoError(b, err) require.NoError(b, err)
defer testStore.Close()
err = testStore.Set("john", []byte("doe"), 0) err = testStore.Set("john", []byte("doe"), 0)
require.NoError(b, err) require.NoError(b, err)
@@ -276,6 +290,7 @@ func Benchmark_Minio_Get(b *testing.B) {
func Benchmark_Minio_SetAndDelete(b *testing.B) { func Benchmark_Minio_SetAndDelete(b *testing.B) {
testStore, err := newTestStore(b) testStore, err := newTestStore(b)
require.NoError(b, err) require.NoError(b, err)
defer testStore.Close()
b.ReportAllocs() b.ReportAllocs()
b.ResetTimer() b.ResetTimer()