package/storage
1
0
Fork 0
mirror of https://github.com/gofiber/storage.git synced 2025-10-05 08:37:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use testify.Require for all Storage Drivers

Browse Source
This commit is contained in:
Juan Calderon-Perez
2023-08-19 22:36:33 -04:00
parent 2779bfdf01
commit 02d809987a
66 changed files with 811 additions and 691 deletions
Download Patch File Download Diff File Expand all files Collapse all files

82
postgres/postgres_test.go
View File

@@ -6,8 +6,8 @@ import (
"testing"
"time"
"github.com/gofiber/utils"
"github.com/jackc/pgx/v5"
"github.com/stretchr/testify/require"
)
var testStore = New(Config{
@@ -24,7 +24,7 @@ func Test_Postgres_Set(t *testing.T) {
)
err := testStore.Set(key, val, 0)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
}
func Test_Postgres_Set_Override(t *testing.T) {
@@ -34,10 +34,10 @@ func Test_Postgres_Set_Override(t *testing.T) {
)
err := testStore.Set(key, val, 0)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
err = testStore.Set(key, val, 0)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
}
func Test_Postgres_Get(t *testing.T) {
@@ -47,11 +47,11 @@ func Test_Postgres_Get(t *testing.T) {
)
err := testStore.Set(key, val, 0)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
result, err := testStore.Get(key)
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, val, result)
require.Nil(t, err)
require.Equal(t, val, result)
}
func Test_Postgres_Set_Expiration(t *testing.T) {
@@ -62,26 +62,23 @@ func Test_Postgres_Set_Expiration(t *testing.T) {
)
err := testStore.Set(key, val, exp)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
time.Sleep(1100 * time.Millisecond)
}
func Test_Postgres_Get_Expired(t *testing.T) {
var (
key = "john"
)
key := "john"
result, err := testStore.Get(key)
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, true, len(result) == 0)
require.Nil(t, err)
require.Zero(t, len(result))
}
func Test_Postgres_Get_NotExist(t *testing.T) {
result, err := testStore.Get("notexist")
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, true, len(result) == 0)
require.Nil(t, err)
require.Zero(t, len(result))
}
func Test_Postgres_Delete(t *testing.T) {
@@ -91,79 +88,74 @@ func Test_Postgres_Delete(t *testing.T) {
)
err := testStore.Set(key, val, 0)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
err = testStore.Delete(key)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
result, err := testStore.Get(key)
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, true, len(result) == 0)
require.Nil(t, err)
require.Zero(t, len(result))
}
func Test_Postgres_Reset(t *testing.T) {
var (
val = []byte("doe")
)
val := []byte("doe")
err := testStore.Set("john1", val, 0)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
err = testStore.Set("john2", val, 0)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
err = testStore.Reset()
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
result, err := testStore.Get("john1")
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, true, len(result) == 0)
require.Nil(t, err)
require.Zero(t, len(result))
result, err = testStore.Get("john2")
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, true, len(result) == 0)
require.Nil(t, err)
require.Zero(t, len(result))
}
func Test_Postgres_GC(t *testing.T) {
var (
testVal = []byte("doe")
)
testVal := []byte("doe")
// This key should expire
err := testStore.Set("john", testVal, time.Nanosecond)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
testStore.gc(time.Now())
row := testStore.db.QueryRow(context.Background(), testStore.sqlSelect, "john")
err = row.Scan(nil, nil)
utils.AssertEqual(t, pgx.ErrNoRows, err)
require.Equal(t, pgx.ErrNoRows, err)
// This key should not expire
err = testStore.Set("john", testVal, 0)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
testStore.gc(time.Now())
val, err := testStore.Get("john")
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, testVal, val)
require.Nil(t, err)
require.Equal(t, testVal, val)
}
func Test_Postgres_Non_UTF8(t *testing.T) {
val := []byte("0xF5")
err := testStore.Set("0xF6", val, 0)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
result, err := testStore.Get("0xF6")
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, val, result)
require.Nil(t, err)
require.Equal(t, val, result)
}
func Test_SslRequiredMode(t *testing.T) {
defer func() {
if recover() == nil {
utils.AssertEqual(t, true, nil, "Connection was established with a `require`")
require.Equalf(t, true, nil, "Connection was established with a `require`")
}
}()
_ = New(Config{
@@ -172,9 +164,9 @@ func Test_SslRequiredMode(t *testing.T) {
}
func Test_Postgres_Conn(t *testing.T) {
utils.AssertEqual(t, true, testStore.Conn() != nil)
require.True(t, testStore.Conn() != nil)
}
func Test_Postgres_Close(t *testing.T) {
utils.AssertEqual(t, nil, testStore.Close())
require.Nil(t, testStore.Close())
}