93 lines
2.5 KiB
Go
93 lines
2.5 KiB
Go
|
package redis
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/go-redis/redismock/v8"
|
||
|
|
"go.signoz.io/signoz/pkg/query-service/cache/status"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestStore(t *testing.T) {
|
||
|
|
db, mock := redismock.NewClientMock()
|
||
|
|
c := WithClient(db)
|
||
|
|
|
||
|
|
mock.ExpectSet("key", []byte("value"), 10*time.Second).RedisNil()
|
||
|
|
c.Store("key", []byte("value"), 10*time.Second)
|
||
|
|
|
||
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
||
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRetrieve(t *testing.T) {
|
||
|
|
db, mock := redismock.NewClientMock()
|
||
|
|
c := WithClient(db)
|
||
|
|
mock.ExpectSet("key", []byte("value"), 10*time.Second).RedisNil()
|
||
|
|
c.Store("key", []byte("value"), 10*time.Second)
|
||
|
|
|
||
|
|
mock.ExpectGet("key").SetVal("value")
|
||
|
|
data, retrieveStatus, err := c.Retrieve("key", false)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("unexpected error: %s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if retrieveStatus != status.RetrieveStatusHit {
|
||
|
|
t.Errorf("expected status %d, got %d", status.RetrieveStatusHit, retrieveStatus)
|
||
|
|
}
|
||
|
|
|
||
|
|
if string(data) != "value" {
|
||
|
|
t.Errorf("expected value %s, got %s", "value", string(data))
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
||
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSetTTL(t *testing.T) {
|
||
|
|
db, mock := redismock.NewClientMock()
|
||
|
|
c := WithClient(db)
|
||
|
|
mock.ExpectSet("key", []byte("value"), 10*time.Second).RedisNil()
|
||
|
|
c.Store("key", []byte("value"), 10*time.Second)
|
||
|
|
|
||
|
|
mock.ExpectExpire("key", 4*time.Second).RedisNil()
|
||
|
|
c.SetTTL("key", 4*time.Second)
|
||
|
|
|
||
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
||
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRemove(t *testing.T) {
|
||
|
|
db, mock := redismock.NewClientMock()
|
||
|
|
c := WithClient(db)
|
||
|
|
mock.ExpectSet("key", []byte("value"), 10*time.Second).RedisNil()
|
||
|
|
c.Store("key", []byte("value"), 10*time.Second)
|
||
|
|
|
||
|
|
mock.ExpectDel("key").RedisNil()
|
||
|
|
c.Remove("key")
|
||
|
|
|
||
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
||
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBulkRemove(t *testing.T) {
|
||
|
|
db, mock := redismock.NewClientMock()
|
||
|
|
c := WithClient(db)
|
||
|
|
mock.ExpectSet("key", []byte("value"), 10*time.Second).RedisNil()
|
||
|
|
c.Store("key", []byte("value"), 10*time.Second)
|
||
|
|
|
||
|
|
mock.ExpectSet("key2", []byte("value2"), 10*time.Second).RedisNil()
|
||
|
|
c.Store("key2", []byte("value2"), 10*time.Second)
|
||
|
|
|
||
|
|
mock.ExpectDel("key").RedisNil()
|
||
|
|
mock.ExpectDel("key2").RedisNil()
|
||
|
|
c.BulkRemove([]string{"key", "key2"})
|
||
|
|
|
||
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
||
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
||
|
|
}
|
||
|
|
}
|