chore: add test

Author: nathanverse

7-lfu-cache/cache_test.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"slices"
 	"testing"
 )
@@ -34,3 +35,44 @@ func TestCache(t *testing.T) {
 		t.Errorf("keys should not contain luan")
 	}
 }
+
+func TestCache1(t *testing.T) {
+	cache, err := NewLFUCache(3, func(key string) (string, error) {
+		return "", nil
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = cache.Set("vu", "10")
+	err = cache.Set("nghia", "20")
+	err = cache.Set("luan", "5")
+
+	for i := 0; i < 10; i++ {
+		cache.Get("vu")
+	}
+
+	for i := 0; i < 9; i++ {
+		cache.Get("nghia")
+	}
+
+	for i := 0; i < 8; i++ {
+		cache.Get("luan")
+	}
+
+	i := 8
+	for e := cache.GetBuckets().Front(); e != nil; e = e.Next() {
+		fmt.Printf("Value: %v (Type: %T)\n", e.Value, e.Value)
+		bucketFreq := cache.GetFreq(e)
+		if bucketFreq != i {
+			t.Errorf("bucketFreq should be %d, got %d", i, bucketFreq)
+		}
+		i += 1
+	}
+
+	err = cache.Set("xanh", "30")
+	keys := cache.GetKeys()
+	if slices.Contains(keys, "luan") {
+		t.Errorf("keys should not contain luan")
+	}
+}

7-lfu-cache/main.go

@@ -61,6 +61,15 @@ func NewLFUCache(size int, loaderFunc LoaderFunc) (*LFUCache, error) {
 	return cache, nil
 }
 
+// For testing
+func (cache *LFUCache) GetBuckets() *list.List {
+	return cache.list
+}
+
+func (cache *LFUCache) GetFreq(buckets *list.Element) int {
+	return buckets.Value.(*lruEntry).freq
+}
+
 func (cache *LFUCache) Get(key string) (string, error) {
 	if item, ok := cache.cache[key]; ok {
 		// Move item to the higher bucket