mirror of
https://github.com/gofiber/storage.git
synced 2025-09-28 05:12:19 +08:00
148 lines
2.4 KiB
Go
148 lines
2.4 KiB
Go
package memory
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gofiber/fiber/v2/utils"
|
|
)
|
|
|
|
func Test_Set(t *testing.T) {
|
|
|
|
store := New()
|
|
|
|
id := "hello"
|
|
value := []byte("Hi there!")
|
|
|
|
store.Set(id, value, 0)
|
|
|
|
utils.AssertEqual(t, entry{value, 0}, store.db[id])
|
|
|
|
}
|
|
|
|
func Test_SetExpiry(t *testing.T) {
|
|
|
|
store := New()
|
|
|
|
id := "hello"
|
|
value := []byte("Hi there!")
|
|
expiry := time.Second * 10
|
|
|
|
store.Set(id, value, expiry)
|
|
|
|
now := time.Now().Unix()
|
|
fromStore, found := store.db[id]
|
|
utils.AssertEqual(t, true, found)
|
|
|
|
delta := fromStore.expiry - now
|
|
upperBound := int64(expiry.Seconds())
|
|
lowerBound := upperBound - 2
|
|
|
|
if !(delta <= upperBound && delta > lowerBound) {
|
|
t.Fatalf("Test_SetExpiry: expiry delta out of bounds (is %d, must be %d<x<=%d)", delta, lowerBound, upperBound)
|
|
}
|
|
|
|
}
|
|
|
|
func Test_GC(t *testing.T) {
|
|
|
|
// New() isn't being used here so the gcInterval can be set low
|
|
store := &Storage{
|
|
db: make(map[string]entry),
|
|
gcInterval: time.Second * 1,
|
|
}
|
|
go store.gc()
|
|
|
|
id := "hello"
|
|
value := []byte("Hi there!")
|
|
|
|
expireAt := time.Now().Add(time.Second * 2).Unix()
|
|
|
|
store.db[id] = entry{value, expireAt}
|
|
|
|
time.Sleep(time.Second * 4) // The purpose of the long delay is to ensure the GC has time to run and delete the value
|
|
|
|
_, found := store.db[id]
|
|
utils.AssertEqual(t, false, found)
|
|
|
|
}
|
|
|
|
func Test_Get(t *testing.T) {
|
|
|
|
store := New()
|
|
|
|
id := "hello"
|
|
value := []byte("Hi there!")
|
|
|
|
store.db[id] = entry{value, 0}
|
|
|
|
returnedValue, err := store.Get(id)
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, value, returnedValue)
|
|
|
|
}
|
|
|
|
func Test_Delete(t *testing.T) {
|
|
|
|
store := New()
|
|
|
|
id := "hello"
|
|
value := []byte("Hi there!")
|
|
|
|
store.db[id] = entry{value, 0}
|
|
|
|
err := store.Delete(id)
|
|
utils.AssertEqual(t, nil, err)
|
|
|
|
_, found := store.db[id]
|
|
utils.AssertEqual(t, false, found)
|
|
|
|
}
|
|
|
|
func Test_Clear(t *testing.T) {
|
|
|
|
store := New()
|
|
|
|
id := "hello"
|
|
value := []byte("Hi there!")
|
|
|
|
store.db[id] = entry{value, 0}
|
|
|
|
err := store.Clear()
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, make(map[string]entry), store.db)
|
|
|
|
}
|
|
|
|
func Benchmark_Set(b *testing.B) {
|
|
|
|
store := New()
|
|
|
|
id := "hello"
|
|
value := []byte("Hi there!")
|
|
expiry := time.Duration(0)
|
|
|
|
b.ResetTimer()
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
store.Set(id, value, expiry)
|
|
}
|
|
|
|
}
|
|
|
|
func Benchmark_Get(b *testing.B) {
|
|
|
|
store := New()
|
|
|
|
id := "hello"
|
|
value := []byte("Hi there!")
|
|
|
|
store.db[id] = entry{value, 0}
|
|
|
|
b.ResetTimer()
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
store.Get(id)
|
|
}
|
|
}
|