mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-07-31 10:11:59 +08:00
5b237ee628
* feat(cache): remove the references of old cache * feat(cache): add orgID in query range modules pt1 * feat(cache): add orgID in query range modules pt2 * feat(cache): add orgID in query range modules pt3 * feat(cache): preload metrics for all orgs * feat(cache): fix ruler * feat(cache): fix go build * feat(cache): add orgID to rule * feat(cache): fix tests * feat(cache): address review comments * feat(cache): use correct errors * feat(cache): fix tests * feat(cache): add the cache test package
120 lines
3.7 KiB
Go
120 lines
3.7 KiB
Go
package rediscache
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/SigNoz/signoz/pkg/valuer"
|
|
"github.com/go-redis/redismock/v8"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type CacheableEntity struct {
|
|
Key string
|
|
Value int
|
|
Expiry time.Duration
|
|
}
|
|
|
|
func (ce *CacheableEntity) MarshalBinary() ([]byte, error) {
|
|
return json.Marshal(ce)
|
|
}
|
|
|
|
func (ce *CacheableEntity) UnmarshalBinary(data []byte) error {
|
|
return json.Unmarshal(data, ce)
|
|
}
|
|
|
|
func TestSet(t *testing.T) {
|
|
db, mock := redismock.NewClientMock()
|
|
cache := WithClient(db)
|
|
storeCacheableEntity := &CacheableEntity{
|
|
Key: "some-random-key",
|
|
Value: 1,
|
|
Expiry: time.Microsecond,
|
|
}
|
|
|
|
orgID := valuer.GenerateUUID()
|
|
mock.ExpectSet(strings.Join([]string{orgID.StringValue(), "key"}, "::"), storeCacheableEntity, 10*time.Second).RedisNil()
|
|
_ = cache.Set(context.Background(), orgID, "key", storeCacheableEntity, 10*time.Second)
|
|
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
|
}
|
|
}
|
|
|
|
func TestGet(t *testing.T) {
|
|
db, mock := redismock.NewClientMock()
|
|
cache := WithClient(db)
|
|
storeCacheableEntity := &CacheableEntity{
|
|
Key: "some-random-key",
|
|
Value: 1,
|
|
Expiry: time.Microsecond,
|
|
}
|
|
retrieveCacheableEntity := new(CacheableEntity)
|
|
|
|
orgID := valuer.GenerateUUID()
|
|
mock.ExpectSet(strings.Join([]string{orgID.StringValue(), "key"}, "::"), storeCacheableEntity, 10*time.Second).RedisNil()
|
|
_ = cache.Set(context.Background(), orgID, "key", storeCacheableEntity, 10*time.Second)
|
|
|
|
data, err := storeCacheableEntity.MarshalBinary()
|
|
assert.NoError(t, err)
|
|
|
|
mock.ExpectGet(strings.Join([]string{orgID.StringValue(), "key"}, "::")).SetVal(string(data))
|
|
err = cache.Get(context.Background(), orgID, "key", retrieveCacheableEntity, false)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %s", err)
|
|
}
|
|
|
|
assert.Equal(t, storeCacheableEntity, retrieveCacheableEntity)
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
|
}
|
|
}
|
|
|
|
func TestDelete(t *testing.T) {
|
|
db, mock := redismock.NewClientMock()
|
|
c := WithClient(db)
|
|
storeCacheableEntity := &CacheableEntity{
|
|
Key: "some-random-key",
|
|
Value: 1,
|
|
Expiry: time.Microsecond,
|
|
}
|
|
orgID := valuer.GenerateUUID()
|
|
|
|
mock.ExpectSet(strings.Join([]string{orgID.StringValue(), "key"}, "::"), storeCacheableEntity, 10*time.Second).RedisNil()
|
|
_ = c.Set(context.Background(), orgID, "key", storeCacheableEntity, 10*time.Second)
|
|
|
|
mock.ExpectDel(strings.Join([]string{orgID.StringValue(), "key"}, "::")).RedisNil()
|
|
c.Delete(context.Background(), orgID, "key")
|
|
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
|
}
|
|
}
|
|
|
|
func TestDeleteMany(t *testing.T) {
|
|
db, mock := redismock.NewClientMock()
|
|
c := WithClient(db)
|
|
storeCacheableEntity := &CacheableEntity{
|
|
Key: "some-random-key",
|
|
Value: 1,
|
|
Expiry: time.Microsecond,
|
|
}
|
|
orgID := valuer.GenerateUUID()
|
|
|
|
mock.ExpectSet(strings.Join([]string{orgID.StringValue(), "key"}, "::"), storeCacheableEntity, 10*time.Second).RedisNil()
|
|
_ = c.Set(context.Background(), orgID, "key", storeCacheableEntity, 10*time.Second)
|
|
|
|
mock.ExpectSet(strings.Join([]string{orgID.StringValue(), "key2"}, "::"), storeCacheableEntity, 10*time.Second).RedisNil()
|
|
_ = c.Set(context.Background(), orgID, "key2", storeCacheableEntity, 10*time.Second)
|
|
|
|
mock.ExpectDel(strings.Join([]string{orgID.StringValue(), "key"}, "::"), strings.Join([]string{orgID.StringValue(), "key2"}, "::")).RedisNil()
|
|
c.DeleteMany(context.Background(), orgID, []string{"key", "key2"})
|
|
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("there were unfulfilled expectations: %s", err)
|
|
}
|
|
}
|