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

chore(mysql): use require in tests

Browse Source
This commit is contained in:
Manuel de la Peña
2025-04-15 16:26:36 +02:00
parent 61d00d996b
commit 0e4dfc0d07
1 changed files with 32 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

80
mysql/mysql_test.go
View File

@@ -22,7 +22,7 @@ const (
mysqlDatabase string = "fiber" mysqlDatabase string = "fiber"
) )
func newTestStore(t testing.TB) (*Storage, error) { func newTestStore(t testing.TB) *Storage {
t.Helper() t.Helper()
ctx := context.Background() ctx := context.Background()
@@ -30,14 +30,12 @@ func newTestStore(t testing.TB) (*Storage, error) {
c := mustStartMySQL(t) c := mustStartMySQL(t)
conn, err := c.ConnectionString(ctx) conn, err := c.ConnectionString(ctx)
if err != nil { require.NoError(t, err)
return nil, err
}
return New(Config{ return New(Config{
ConnectionURI: conn, ConnectionURI: conn,
Reset: true, Reset: true,
}), nil })
} }
func mustStartMySQL(t testing.TB) *mysql.MySQLContainer { func mustStartMySQL(t testing.TB) *mysql.MySQLContainer {
@@ -64,8 +62,8 @@ func mustStartMySQL(t testing.TB) *mysql.MySQLContainer {
} }
func Test_MYSQL_New(t *testing.T) { func Test_MYSQL_New(t *testing.T) {
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err) defer testStore.Close()
require.True(t, testStore.db != nil) require.True(t, testStore.db != nil)
require.NoError(t, testStore.Close()) require.NoError(t, testStore.Close())
@@ -91,11 +89,10 @@ func Test_MYSQL_Set(t *testing.T) {
val = []byte("doe") val = []byte("doe")
) )
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err)
defer testStore.Close() defer testStore.Close()
err = testStore.Set(key, val, 0) err := testStore.Set(key, val, 0)
require.NoError(t, err) require.NoError(t, err)
} }
@@ -105,11 +102,10 @@ func Test_MYSQL_Set_Override(t *testing.T) {
val = []byte("doe") val = []byte("doe")
) )
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err)
defer testStore.Close() defer testStore.Close()
err = testStore.Set(key, val, 0) err := testStore.Set(key, val, 0)
require.NoError(t, err) require.NoError(t, err)
err = testStore.Set(key, val, 0) err = testStore.Set(key, val, 0)
@@ -122,11 +118,10 @@ func Test_MYSQL_Get(t *testing.T) {
val = []byte("doe") val = []byte("doe")
) )
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err)
defer testStore.Close() defer testStore.Close()
err = testStore.Set(key, val, 0) err := testStore.Set(key, val, 0)
require.NoError(t, err) require.NoError(t, err)
result, err := testStore.Get(key) result, err := testStore.Get(key)
@@ -141,11 +136,10 @@ func Test_MYSQL_Set_Expiration(t *testing.T) {
exp = 1 * time.Second exp = 1 * time.Second
) )
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err)
defer testStore.Close() defer testStore.Close()
err = testStore.Set(key, val, exp) err := testStore.Set(key, val, exp)
require.NoError(t, err) require.NoError(t, err)
time.Sleep(1100 * time.Millisecond) time.Sleep(1100 * time.Millisecond)
@@ -158,8 +152,7 @@ func Test_MYSQL_Set_Expiration(t *testing.T) {
func Test_MYSQL_Get_Expired(t *testing.T) { func Test_MYSQL_Get_Expired(t *testing.T) {
key := "john" key := "john"
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err)
defer testStore.Close() defer testStore.Close()
result, err := testStore.Get(key) result, err := testStore.Get(key)
@@ -168,8 +161,7 @@ func Test_MYSQL_Get_Expired(t *testing.T) {
} }
func Test_MYSQL_Get_NotExist(t *testing.T) { func Test_MYSQL_Get_NotExist(t *testing.T) {
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err)
defer testStore.Close() defer testStore.Close()
result, err := testStore.Get("notexist") result, err := testStore.Get("notexist")
@@ -183,11 +175,10 @@ func Test_MYSQL_Delete(t *testing.T) {
val = []byte("doe") val = []byte("doe")
) )
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err)
defer testStore.Close() defer testStore.Close()
err = testStore.Set(key, val, 0) err := testStore.Set(key, val, 0)
require.NoError(t, err) require.NoError(t, err)
err = testStore.Delete(key) err = testStore.Delete(key)
@@ -201,11 +192,10 @@ func Test_MYSQL_Delete(t *testing.T) {
func Test_MYSQL_Reset(t *testing.T) { func Test_MYSQL_Reset(t *testing.T) {
val := []byte("doe") val := []byte("doe")
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err)
defer testStore.Close() defer testStore.Close()
err = testStore.Set("john1", val, 0) err := testStore.Set("john1", val, 0)
require.NoError(t, err) require.NoError(t, err)
err = testStore.Set("john2", val, 0) err = testStore.Set("john2", val, 0)
@@ -227,11 +217,10 @@ func Test_MYSQL_GC(t *testing.T) {
testVal := []byte("doe") testVal := []byte("doe")
// This key should expire // This key should expire
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err)
defer testStore.Close() defer testStore.Close()
err = testStore.Set("john", testVal, time.Nanosecond) err := testStore.Set("john", testVal, time.Nanosecond)
require.NoError(t, err) require.NoError(t, err)
testStore.gc(time.Now()) testStore.gc(time.Now())
@@ -252,11 +241,10 @@ func Test_MYSQL_GC(t *testing.T) {
func Test_MYSQL_Non_UTF8(t *testing.T) { func Test_MYSQL_Non_UTF8(t *testing.T) {
val := []byte("0xF5") val := []byte("0xF5")
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err)
defer testStore.Close() defer testStore.Close()
err = testStore.Set("0xF6", val, 0) err := testStore.Set("0xF6", val, 0)
require.NoError(t, err) require.NoError(t, err)
result, err := testStore.Get("0xF6") result, err := testStore.Get("0xF6")
@@ -265,28 +253,25 @@ func Test_MYSQL_Non_UTF8(t *testing.T) {
} }
func Test_MYSQL_Close(t *testing.T) { func Test_MYSQL_Close(t *testing.T) {
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err) require.NoError(t, testStore.Close())
require.Nil(t, testStore.Close())
} }
func Test_MYSQL_Conn(t *testing.T) { func Test_MYSQL_Conn(t *testing.T) {
testStore, err := newTestStore(t) testStore := newTestStore(t)
require.NoError(t, err)
defer testStore.Close() defer testStore.Close()
require.True(t, testStore.Conn() != nil) require.True(t, testStore.Conn() != nil)
} }
func Benchmark_MYSQL_Set(b *testing.B) { func Benchmark_MYSQL_Set(b *testing.B) {
testStore, err := newTestStore(b) testStore := newTestStore(b)
require.NoError(b, err)
defer testStore.Close() defer testStore.Close()
b.ReportAllocs() b.ReportAllocs()
b.ResetTimer() b.ResetTimer()
var err error
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)
} }
@@ -295,11 +280,10 @@ func Benchmark_MYSQL_Set(b *testing.B) {
} }
func Benchmark_MYSQL_Get(b *testing.B) { func Benchmark_MYSQL_Get(b *testing.B) {
testStore, err := newTestStore(b) testStore := newTestStore(b)
require.NoError(b, err)
defer testStore.Close() 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)
b.ReportAllocs() b.ReportAllocs()
@@ -313,13 +297,13 @@ func Benchmark_MYSQL_Get(b *testing.B) {
} }
func Benchmark_MYSQL_SetAndDelete(b *testing.B) { func Benchmark_MYSQL_SetAndDelete(b *testing.B) {
testStore, err := newTestStore(b) testStore := newTestStore(b)
require.NoError(b, err)
defer testStore.Close() defer testStore.Close()
b.ReportAllocs() b.ReportAllocs()
b.ResetTimer() b.ResetTimer()
var err error
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
_ = testStore.Set("john", []byte("doe"), 0) _ = testStore.Set("john", []byte("doe"), 0)
err = testStore.Delete("john") err = testStore.Delete("john")