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
sqlite3/sqlite3_test.go
View File

@@ -5,8 +5,8 @@ import (
"testing"
"time"
"github.com/gofiber/utils"
_ "github.com/mattn/go-sqlite3"
"github.com/stretchr/testify/require"
)
var testStore = New(Config{
@@ -20,7 +20,7 @@ func Test_SQLite3_Set(t *testing.T) {
)
err := testStore.Set(key, val, 0)
utils.AssertEqual(t, nil, err)
require.Nil(t, err)
}
func Test_SQLite3_Set_Override(t *testing.T) {
@@ -30,10 +30,10 @@ func Test_SQLite3_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_SQLite3_Get(t *testing.T) {
@@ -43,11 +43,11 @@ func Test_SQLite3_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_SQLite3_Set_Expiration(t *testing.T) {
@@ -58,26 +58,23 @@ func Test_SQLite3_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_SQLite3_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_SQLite3_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_SQLite3_Delete(t *testing.T) {
@@ -87,79 +84,74 @@ func Test_SQLite3_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_SQLite3_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_SQLite3_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(testStore.sqlSelect, "john")
err = row.Scan(nil, nil)
utils.AssertEqual(t, sql.ErrNoRows, err)
require.Equal(t, sql.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_SQLite3_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_SQLite3_Close(t *testing.T) {
utils.AssertEqual(t, nil, testStore.Close())
require.Nil(t, testStore.Close())
}
func Test_SQLite3_Conn(t *testing.T) {
utils.AssertEqual(t, true, testStore.Conn() != nil)
}
require.True(t, testStore.Conn() != nil)
}