fixed some tests

Author: Code-Hex

cache.go

@@ -157,7 +157,7 @@ func WithJanitorInterval[K comparable, V any](d time.Duration) Option[K, V] {
 // There are several Cache replacement policies available with you specified any options.
 func New[K comparable, V any](opts ...Option[K, V]) *Cache[K, V] {
 	ctx, cancel := context.WithCancel(context.Background())
-	cache := NewContext[K, V](ctx)
+	cache := NewContext(ctx, opts...)
 	runtime.SetFinalizer(cache, func(self *Cache[K, V]) {
 		cancel()
 	})

cache_internal_test.go

@@ -2,6 +2,7 @@ package cache
 
 import (
 	"context"
+	"runtime"
 	"testing"
 	"time"
 )
@@ -26,3 +27,29 @@ func TestDeletedCache(t *testing.T) {
 		t.Fatal("want false")
 	}
 }
+
+func TestFinalizeCache(t *testing.T) {
+	if runtime.GOARCH != "amd64" {
+		t.Skipf("Skipping on non-amd64 machine")
+	}
+
+	done := make(chan struct{})
+	wait := make(chan struct{})
+	go func() {
+		c := New[string, int]()
+		c.janitor.run(func() {
+			close(done)
+		})
+		c = nil
+		close(wait)
+	}()
+	<-wait
+
+	runtime.GC()
+
+	select {
+	case <-done:
+	case <-time.After(10 * time.Second):
+		t.Fatal("expected to call a function which is set as finalizer")
+	}
+}

cache_test.go

@@ -2,8 +2,10 @@ package cache_test
 
 import (
 	"math/rand"
+	"runtime"
 	"sync"
 	"testing"
+	"time"
 
 	cache "github.com/Code-Hex/go-generics-cache"
 	"github.com/Code-Hex/go-generics-cache/policy/clock"
@@ -103,3 +105,22 @@ func TestMultiThread(t *testing.T) {
 		})
 	}
 }
+
+func TestCallJanitor(t *testing.T) {
+	c := cache.New(
+		cache.WithJanitorInterval[string, int](100 * time.Millisecond),
+	)
+
+	c.Set("1", 10, cache.WithExpiration(10*time.Millisecond))
+	c.Set("2", 20, cache.WithExpiration(20*time.Millisecond))
+	c.Set("3", 30, cache.WithExpiration(30*time.Millisecond))
+
+	<-time.After(300 * time.Millisecond)
+
+	keys := c.Keys()
+	if len(keys) != 0 {
+		t.Errorf("want items is empty but got %d", len(keys))
+	}
+
+	runtime.GC()
+}