chore: close store

This commit is contained in:
Manuel de la Peña
2024-09-18 16:59:58 +02:00
parent 75c8525a46
commit 1bb4d922e9
4 changed files with 59 additions and 0 deletions

View File

@@ -52,6 +52,7 @@ func Test_MongoDB_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)
@@ -65,6 +66,7 @@ func Test_MongoDB_Set_Override(t *testing.T) {
testStore, err := newTestStore(t)
require.NoError(t, err)
defer testStore.Close()
err = testStore.Set(key, val, 0)
require.NoError(t, err)
@@ -81,6 +83,7 @@ func Test_MongoDB_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)
@@ -99,6 +102,7 @@ func Test_MongoDB_Set_Expiration(t *testing.T) {
testStore, err := newTestStore(t)
require.NoError(t, err)
defer testStore.Close()
err = testStore.Set(key, val, exp)
require.NoError(t, err)
@@ -111,6 +115,7 @@ func Test_MongoDB_Get_Expired(t *testing.T) {
testStore, err := newTestStore(t)
require.NoError(t, err)
defer testStore.Close()
result, err := testStore.Get(key)
require.NoError(t, err)
@@ -120,6 +125,7 @@ func Test_MongoDB_Get_Expired(t *testing.T) {
func Test_MongoDB_Get_NotExist(t *testing.T) {
testStore, err := newTestStore(t)
require.NoError(t, err)
defer testStore.Close()
result, err := testStore.Get("notexist")
require.NoError(t, err)
@@ -134,6 +140,7 @@ func Test_MongoDB_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)
@@ -151,6 +158,7 @@ func Test_MongoDB_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)
@@ -173,6 +181,7 @@ func Test_MongoDB_Reset(t *testing.T) {
func Test_MongoDB_Close(t *testing.T) {
testStore, err := newTestStore(t)
require.NoError(t, err)
defer testStore.Close()
require.Nil(t, testStore.Close())
}
@@ -180,6 +189,7 @@ func Test_MongoDB_Close(t *testing.T) {
func Test_MongoDB_Conn(t *testing.T) {
testStore, err := newTestStore(t)
require.NoError(t, err)
defer testStore.Close()
require.True(t, testStore.Conn() != nil)
}
@@ -187,6 +197,7 @@ func Test_MongoDB_Conn(t *testing.T) {
func Benchmark_MongoDB_Set(b *testing.B) {
testStore, err := newTestStore(b)
require.NoError(b, err)
defer testStore.Close()
b.ReportAllocs()
b.ResetTimer()
@@ -201,6 +212,7 @@ func Benchmark_MongoDB_Set(b *testing.B) {
func Benchmark_MongoDB_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)
@@ -218,6 +230,7 @@ func Benchmark_MongoDB_Get(b *testing.B) {
func Benchmark_MongoDB_SetAndDelete(b *testing.B) {
testStore, err := newTestStore(b)
require.NoError(b, err)
defer testStore.Close()
b.ReportAllocs()
b.ResetTimer()