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

@@ -63,6 +63,7 @@ func Test_Postgres_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)
@@ -76,6 +77,7 @@ func Test_Postgres_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)
@@ -92,6 +94,7 @@ func Test_Postgres_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)
@@ -110,6 +113,7 @@ func Test_Postgres_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)
@@ -122,6 +126,7 @@ func Test_Postgres_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)
@@ -131,6 +136,7 @@ func Test_Postgres_Get_Expired(t *testing.T) {
func Test_Postgres_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)
@@ -145,6 +151,7 @@ func Test_Postgres_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)
@@ -162,6 +169,7 @@ func Test_Postgres_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)
@@ -186,6 +194,7 @@ func Test_Postgres_GC(t *testing.T) {
testStore, err := newTestStore(t)
require.NoError(t, err)
defer testStore.Close()
// This key should expire
err = testStore.Set("john", testVal, time.Nanosecond)
@@ -211,6 +220,7 @@ func Test_Postgres_Non_UTF8(t *testing.T) {
testStore, err := newTestStore(t)
require.NoError(t, err)
defer testStore.Close()
err = testStore.Set("0xF6", val, 0)
require.NoError(t, err)
@@ -234,6 +244,7 @@ func Test_SslRequiredMode(t *testing.T) {
func Test_Postgres_Conn(t *testing.T) {
testStore, err := newTestStore(t)
require.NoError(t, err)
defer testStore.Close()
require.True(t, testStore.Conn() != nil)
}
@@ -241,6 +252,7 @@ func Test_Postgres_Conn(t *testing.T) {
func Test_Postgres_Close(t *testing.T) {
testStore, err := newTestStore(t)
require.NoError(t, err)
defer testStore.Close()
require.Nil(t, testStore.Close())
}
@@ -248,6 +260,7 @@ func Test_Postgres_Close(t *testing.T) {
func Benchmark_Postgres_Set(b *testing.B) {
testStore, err := newTestStore(b)
require.NoError(b, err)
defer testStore.Close()
b.ReportAllocs()
b.ResetTimer()
@@ -262,6 +275,7 @@ func Benchmark_Postgres_Set(b *testing.B) {
func Benchmark_Postgres_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)
@@ -279,6 +293,7 @@ func Benchmark_Postgres_Get(b *testing.B) {
func Benchmark_Postgres_SetAndDelete(b *testing.B) {
testStore, err := newTestStore(b)
require.NoError(b, err)
defer testStore.Close()
b.ReportAllocs()
b.ResetTimer()