mirror of
https://github.com/gofiber/storage.git
synced 2025-10-04 16:22:52 +08:00
Use testify.Require for all Storage Drivers
This commit is contained in:
@@ -4,7 +4,7 @@ import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/gofiber/utils"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestSetCouchbase_ShouldReturnNoError(t *testing.T) {
|
||||
@@ -17,7 +17,7 @@ func TestSetCouchbase_ShouldReturnNoError(t *testing.T) {
|
||||
|
||||
err := testStorage.Set("test", []byte("test"), 0)
|
||||
|
||||
utils.AssertEqual(t, nil, err)
|
||||
require.Nil(t, err)
|
||||
}
|
||||
|
||||
func TestGetCouchbase_ShouldReturnNil_WhenDocumentNotFound(t *testing.T) {
|
||||
@@ -30,8 +30,8 @@ func TestGetCouchbase_ShouldReturnNil_WhenDocumentNotFound(t *testing.T) {
|
||||
|
||||
val, err := testStorage.Get("not_found_key")
|
||||
|
||||
utils.AssertEqual(t, nil, err)
|
||||
utils.AssertEqual(t, 0, len(val))
|
||||
require.Nil(t, err)
|
||||
require.Zero(len(val))
|
||||
}
|
||||
|
||||
func TestSetAndGet_GetShouldReturn_SettedValueWithoutError(t *testing.T) {
|
||||
@@ -43,12 +43,12 @@ func TestSetAndGet_GetShouldReturn_SettedValueWithoutError(t *testing.T) {
|
||||
})
|
||||
|
||||
err := testStorage.Set("test", []byte("fiber_test_value"), 0)
|
||||
utils.AssertEqual(t, nil, err)
|
||||
require.Nil(t, err)
|
||||
|
||||
val, err := testStorage.Get("test")
|
||||
|
||||
utils.AssertEqual(t, nil, err)
|
||||
utils.AssertEqual(t, val, []byte("fiber_test_value"))
|
||||
require.Nil(t, err)
|
||||
require.Equal(t, val, []byte("fiber_test_value"))
|
||||
}
|
||||
|
||||
func TestSetAndGet_GetShouldReturnNil_WhenTTLExpired(t *testing.T) {
|
||||
@@ -60,14 +60,14 @@ func TestSetAndGet_GetShouldReturnNil_WhenTTLExpired(t *testing.T) {
|
||||
})
|
||||
|
||||
err := testStorage.Set("test", []byte("fiber_test_value"), 3*time.Second)
|
||||
utils.AssertEqual(t, nil, err)
|
||||
require.Nil(t, err)
|
||||
|
||||
time.Sleep(6 * time.Second)
|
||||
|
||||
val, err := testStorage.Get("test")
|
||||
|
||||
utils.AssertEqual(t, nil, err)
|
||||
utils.AssertEqual(t, 0, len(val))
|
||||
require.Nil(t, err)
|
||||
require.Zero(len(val))
|
||||
}
|
||||
|
||||
func TestSetAndDelete_DeleteShouldReturn_NoError(t *testing.T) {
|
||||
@@ -79,13 +79,13 @@ func TestSetAndDelete_DeleteShouldReturn_NoError(t *testing.T) {
|
||||
})
|
||||
|
||||
err := testStorage.Set("test", []byte("fiber_test_value"), 0)
|
||||
utils.AssertEqual(t, nil, err)
|
||||
require.Nil(t, err)
|
||||
|
||||
err = testStorage.Delete("test")
|
||||
utils.AssertEqual(t, nil, err)
|
||||
require.Nil(t, err)
|
||||
|
||||
_, err = testStorage.Get("test")
|
||||
utils.AssertEqual(t, nil, err)
|
||||
require.Nil(t, err)
|
||||
}
|
||||
|
||||
func TestSetAndReset_ResetShouldReturn_NoError(t *testing.T) {
|
||||
@@ -97,13 +97,13 @@ func TestSetAndReset_ResetShouldReturn_NoError(t *testing.T) {
|
||||
})
|
||||
|
||||
err := testStorage.Set("test", []byte("fiber_test_value"), 0)
|
||||
utils.AssertEqual(t, nil, err)
|
||||
require.Nil(t, err)
|
||||
|
||||
err = testStorage.Reset()
|
||||
utils.AssertEqual(t, nil, err)
|
||||
require.Nil(t, err)
|
||||
|
||||
_, err = testStorage.Get("test")
|
||||
utils.AssertEqual(t, nil, err)
|
||||
require.Nil(t, err)
|
||||
}
|
||||
|
||||
func TestClose_CloseShouldReturn_NoError(t *testing.T) {
|
||||
@@ -115,7 +115,7 @@ func TestClose_CloseShouldReturn_NoError(t *testing.T) {
|
||||
})
|
||||
|
||||
err := testStorage.Close()
|
||||
utils.AssertEqual(t, nil, err)
|
||||
require.Nil(t, err)
|
||||
}
|
||||
|
||||
func TestGetConn_ReturnsNotNill(t *testing.T) {
|
||||
@@ -125,6 +125,5 @@ func TestGetConn_ReturnsNotNill(t *testing.T) {
|
||||
Host: "127.0.0.1:8091",
|
||||
Bucket: "fiber_storage",
|
||||
})
|
||||
conn := testStorage.Conn()
|
||||
utils.AssertEqual(t, true, conn != nil)
|
||||
require.True(t, testStorage.Conn() != nil)
|
||||
}
|
||||
|
Reference in New Issue
Block a user