aboutsummaryrefslogtreecommitdiffstats
path: root/conn_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'conn_test.go')
-rw-r--r--conn_test.go120
1 files changed, 99 insertions, 21 deletions
diff --git a/conn_test.go b/conn_test.go
index 47bdde4..cfcad13 100644
--- a/conn_test.go
+++ b/conn_test.go
@@ -7,7 +7,7 @@ import (
"time"
)
-func setupTestDB[K any, V any](tb testing.TB) *Cache[K, V] {
+func setupTestCache[K, V any](tb testing.TB) *Cache[K, V] {
tb.Helper()
db, err := OpenMem[K, V]()
@@ -24,13 +24,91 @@ func setupTestDB[K any, V any](tb testing.TB) *Cache[K, V] {
return &db
}
-func TestDBGetSet(t *testing.T) {
+func TestCacheSetConfig(t *testing.T) {
+ tests := []struct {
+ name string
+ options []Option
+ wantErr bool
+ expectedPolicy EvictionPolicyType
+ expectedMaxCost uint64
+ snapshotTime time.Duration
+ cleanupTime time.Duration
+ }{
+ {
+ name: "Set all valid options",
+ options: []Option{
+ WithPolicy(PolicyLRU),
+ WithMaxCost(10000),
+ SetSnapshotTime(2 * time.Minute),
+ SetCleanupTime(30 * time.Second),
+ },
+ wantErr: false,
+ expectedPolicy: PolicyLRU,
+ expectedMaxCost: 10000,
+ snapshotTime: 2 * time.Minute,
+ cleanupTime: 30 * time.Second,
+ },
+ {
+ name: "Invalid policy returns error",
+ options: []Option{
+ WithPolicy(-1),
+ },
+ wantErr: true,
+ },
+ {
+ name: "Set only max cost",
+ options: []Option{
+ WithMaxCost(2048),
+ },
+ wantErr: false,
+ expectedMaxCost: 2048,
+ },
+ {
+ name: "Set only snapshot and cleanup",
+ options: []Option{
+ SetSnapshotTime(15 * time.Second),
+ SetCleanupTime(1 * time.Minute),
+ },
+ wantErr: false,
+ snapshotTime: 15 * time.Second,
+ cleanupTime: 1 * time.Minute,
+ },
+ }
+
+ for _, tt := range tests {
+ t.Run(tt.name, func(t *testing.T) {
+ c := setupTestCache[string, string](t)
+
+ err := c.SetConfig(tt.options...)
+ if (err != nil) != tt.wantErr {
+ t.Fatalf("SetConfig() error = %v, wantErr = %v", err, tt.wantErr)
+ }
+
+ if !tt.wantErr {
+ if c.Store.Policy.Type != tt.expectedPolicy {
+ t.Errorf("Expected policy %v, got %v", tt.expectedPolicy, c.Store.Policy.Type)
+ }
+ if tt.expectedMaxCost != 0 && c.Store.MaxCost != tt.expectedMaxCost {
+ t.Errorf("Expected MaxCost %d, got %d", tt.expectedMaxCost, c.Store.MaxCost)
+ }
+ if tt.snapshotTime != 0 && c.Store.SnapshotTicker.GetDuration() != tt.snapshotTime {
+ t.Errorf("Expected SnapshotTime %v, got %v", tt.snapshotTime, c.Store.SnapshotTicker.GetDuration())
+ }
+ if tt.cleanupTime != 0 && c.Store.CleanupTicker.GetDuration() != tt.cleanupTime {
+ t.Errorf("Expected CleanupTime %v, got %v", tt.cleanupTime, c.Store.CleanupTicker.GetDuration())
+ }
+ }
+ })
+ }
+}
+
+func TestCacheGetSet(t *testing.T) {
t.Parallel()
t.Run("Exists", func(t *testing.T) {
t.Parallel()
- db := setupTestDB[string, string](t)
+ db := setupTestCache[string, string](t)
want := "Value"
@@ -55,7 +133,7 @@ func TestDBGetSet(t *testing.T) {
t.Run("Not Exists", func(t *testing.T) {
t.Parallel()
- db := setupTestDB[string, string](t)
+ db := setupTestCache[string, string](t)
if _, _, err := db.GetValue("Key"); !errors.Is(err, ErrKeyNotFound) {
t.Fatalf("expected error: %v, got: %v", ErrKeyNotFound, err)
@@ -65,7 +143,7 @@ func TestDBGetSet(t *testing.T) {
t.Run("Update", func(t *testing.T) {
t.Parallel()
- db := setupTestDB[string, string](t)
+ db := setupTestCache[string, string](t)
if err := db.Set("Key", "Other", 0); err != nil {
t.Fatalf("expected no error, got: %v", err)
@@ -89,7 +167,7 @@ func TestDBGetSet(t *testing.T) {
t.Run("Key Expiry", func(t *testing.T) {
t.Parallel()
- db := setupTestDB[string, string](t)
+ db := setupTestCache[string, string](t)
if err := db.Set("Key", "Value", 500*time.Millisecond); err != nil {
t.Fatalf("unexpected error: %v", err)
@@ -103,13 +181,13 @@ func TestDBGetSet(t *testing.T) {
})
}
-func TestDBDelete(t *testing.T) {
+func TestCacheDelete(t *testing.T) {
t.Parallel()
t.Run("Exists", func(t *testing.T) {
t.Parallel()
- db := setupTestDB[string, string](t)
+ db := setupTestCache[string, string](t)
want := "Value"
if err := db.Set("Key", want, 0); err != nil {
@@ -128,7 +206,7 @@ func TestDBDelete(t *testing.T) {
t.Run("Not Exists", func(t *testing.T) {
t.Parallel()
- db := setupTestDB[string, string](t)
+ db := setupTestCache[string, string](t)
if err := db.Delete("Key"); !errors.Is(err, ErrKeyNotFound) {
t.Fatalf("expected error: %v, got: %v", ErrKeyNotFound, err)
@@ -136,13 +214,13 @@ func TestDBDelete(t *testing.T) {
})
}
-func TestDBUpdateInPlace(t *testing.T) {
+func TestCacheUpdateInPlace(t *testing.T) {
t.Parallel()
t.Run("Exists", func(t *testing.T) {
t.Parallel()
- store := setupTestDB[string, string](t)
+ store := setupTestCache[string, string](t)
want := "Value"
@@ -171,7 +249,7 @@ func TestDBUpdateInPlace(t *testing.T) {
t.Run("Not Exists", func(t *testing.T) {
t.Parallel()
- store := setupTestDB[string, string](t)
+ store := setupTestCache[string, string](t)
want := "Value"
@@ -185,13 +263,13 @@ func TestDBUpdateInPlace(t *testing.T) {
})
}
-func TestDBMemoize(t *testing.T) {
+func TestCacheMemoize(t *testing.T) {
t.Parallel()
t.Run("Cache Miss", func(t *testing.T) {
t.Parallel()
- store := setupTestDB[string, string](t)
+ store := setupTestCache[string, string](t)
want := "Value"
@@ -221,7 +299,7 @@ func TestDBMemoize(t *testing.T) {
t.Run("Cache Hit", func(t *testing.T) {
t.Parallel()
- store := setupTestDB[string, string](t)
+ store := setupTestCache[string, string](t)
want := "NewValue"
@@ -244,10 +322,10 @@ func TestDBMemoize(t *testing.T) {
})
}
-func BenchmarkDBGet(b *testing.B) {
+func BenchmarkCacheGet(b *testing.B) {
for n := 1; n <= 100000; n *= 10 {
b.Run(strconv.Itoa(n), func(b *testing.B) {
- db := setupTestDB[int, int](b)
+ db := setupTestCache[int, int](b)
for i := range n {
if err := db.Set(i, i, 0); err != nil {
b.Fatalf("unexpected error: %v", err)
@@ -265,10 +343,10 @@ func BenchmarkDBGet(b *testing.B) {
}
}
-func BenchmarkDBSet(b *testing.B) {
+func BenchmarkCacheSet(b *testing.B) {
for n := 1; n <= 100000; n *= 10 {
b.Run(strconv.Itoa(n), func(b *testing.B) {
- db := setupTestDB[int, int](b)
+ db := setupTestCache[int, int](b)
for i := range n - 1 {
if err := db.Set(i, i, 0); err != nil {
b.Fatalf("unexpected error: %v", err)
@@ -286,10 +364,10 @@ func BenchmarkDBSet(b *testing.B) {
}
}
-func BenchmarkDBDelete(b *testing.B) {
+func BenchmarkCacheDelete(b *testing.B) {
for n := 1; n <= 100000; n *= 10 {
b.Run(strconv.Itoa(n), func(b *testing.B) {
- db := setupTestDB[int, int](b)
+ db := setupTestCache[int, int](b)
for i := range n - 1 {
if err := db.Set(i, i, 0); err != nil {
b.Fatalf("unexpected error: %v", err)
generated by cgit v1.2.3-70-g09d2 (git 2.50.1) at 2025-08-17 01:26:20 +0000