mirror of
https://github.com/gofiber/storage.git
synced 2025-09-27 12:52:25 +08:00
131 lines
2.9 KiB
Go
131 lines
2.9 KiB
Go
package couchbase
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gofiber/utils"
|
|
)
|
|
|
|
func TestSetCouchbase_ShouldReturnNoError(t *testing.T) {
|
|
testStorage := New(Config{
|
|
Username: "admin",
|
|
Password: "123456",
|
|
Host: "127.0.0.1:8091",
|
|
Bucket: "fiber_storage",
|
|
})
|
|
|
|
err := testStorage.Set("test", []byte("test"), 0)
|
|
|
|
utils.AssertEqual(t, nil, err)
|
|
}
|
|
|
|
func TestGetCouchbase_ShouldReturnNil_WhenDocumentNotFound(t *testing.T) {
|
|
testStorage := New(Config{
|
|
Username: "admin",
|
|
Password: "123456",
|
|
Host: "127.0.0.1:8091",
|
|
Bucket: "fiber_storage",
|
|
})
|
|
|
|
val, err := testStorage.Get("not_found_key")
|
|
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, 0, len(val))
|
|
}
|
|
|
|
func TestSetAndGet_GetShouldReturn_SettedValueWithoutError(t *testing.T) {
|
|
testStorage := New(Config{
|
|
Username: "admin",
|
|
Password: "123456",
|
|
Host: "127.0.0.1:8091",
|
|
Bucket: "fiber_storage",
|
|
})
|
|
|
|
err := testStorage.Set("test", []byte("fiber_test_value"), 0)
|
|
utils.AssertEqual(t, nil, err)
|
|
|
|
val, err := testStorage.Get("test")
|
|
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, val, []byte("fiber_test_value"))
|
|
}
|
|
|
|
func TestSetAndGet_GetShouldReturnNil_WhenTTLExpired(t *testing.T) {
|
|
testStorage := New(Config{
|
|
Username: "admin",
|
|
Password: "123456",
|
|
Host: "127.0.0.1:8091",
|
|
Bucket: "fiber_storage",
|
|
})
|
|
|
|
err := testStorage.Set("test", []byte("fiber_test_value"), 3*time.Second)
|
|
utils.AssertEqual(t, nil, err)
|
|
|
|
time.Sleep(6 * time.Second)
|
|
|
|
val, err := testStorage.Get("test")
|
|
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, 0, len(val))
|
|
}
|
|
|
|
func TestSetAndDelete_DeleteShouldReturn_NoError(t *testing.T) {
|
|
testStorage := New(Config{
|
|
Username: "admin",
|
|
Password: "123456",
|
|
Host: "127.0.0.1:8091",
|
|
Bucket: "fiber_storage",
|
|
})
|
|
|
|
err := testStorage.Set("test", []byte("fiber_test_value"), 0)
|
|
utils.AssertEqual(t, nil, err)
|
|
|
|
err = testStorage.Delete("test")
|
|
utils.AssertEqual(t, nil, err)
|
|
|
|
_, err = testStorage.Get("test")
|
|
utils.AssertEqual(t, nil, err)
|
|
}
|
|
|
|
func TestSetAndReset_ResetShouldReturn_NoError(t *testing.T) {
|
|
testStorage := New(Config{
|
|
Username: "admin",
|
|
Password: "123456",
|
|
Host: "127.0.0.1:8091",
|
|
Bucket: "fiber_storage",
|
|
})
|
|
|
|
err := testStorage.Set("test", []byte("fiber_test_value"), 0)
|
|
utils.AssertEqual(t, nil, err)
|
|
|
|
err = testStorage.Reset()
|
|
utils.AssertEqual(t, nil, err)
|
|
|
|
_, err = testStorage.Get("test")
|
|
utils.AssertEqual(t, nil, err)
|
|
}
|
|
|
|
func TestClose_CloseShouldReturn_NoError(t *testing.T) {
|
|
testStorage := New(Config{
|
|
Username: "admin",
|
|
Password: "123456",
|
|
Host: "127.0.0.1:8091",
|
|
Bucket: "fiber_storage",
|
|
})
|
|
|
|
err := testStorage.Close()
|
|
utils.AssertEqual(t, nil, err)
|
|
}
|
|
|
|
func TestGetConn_ReturnsNotNill(t *testing.T) {
|
|
testStorage := New(Config{
|
|
Username: "admin",
|
|
Password: "123456",
|
|
Host: "127.0.0.1:8091",
|
|
Bucket: "fiber_storage",
|
|
})
|
|
conn := testStorage.Conn()
|
|
utils.AssertEqual(t, true, conn != nil)
|
|
}
|