chore: test cache

pkg/cache/cache.go

@@ -17,11 +17,10 @@ type Cache struct {
 	Entries   map[string]CacheEntry
 }
 
-func NewCache(logger *zerolog.Logger) *Cache {
+func NewCache() *Cache {
 	return &Cache{
 		StoreTime: 10 * time.Minute,
-		Entries:   make(map[string]CacheEntry, 0),
-		Logger:    logger.With().Str("component", "cache").Logger(),
+		Entries:   make(map[string]CacheEntry),
 	}
 }
 

pkg/cache/cache_test.go

@@ -0,0 +1,38 @@
+package cache_test
+
+import (
+	"github.com/stretchr/testify/require"
+	cachePkg "main/pkg/cache"
+	"testing"
+	"time"
+)
+
+func TestCacheSet(t *testing.T) {
+	t.Parallel()
+
+	cache := cachePkg.NewCache()
+	cache.Set("key", "value")
+
+	entry, found := cache.Get("key")
+	require.Equal(t, "value", entry)
+	require.True(t, found)
+}
+
+func TestCacheGetNotExists(t *testing.T) {
+	t.Parallel()
+
+	cache := cachePkg.NewCache()
+	_, found := cache.Get("key")
+	require.False(t, found)
+}
+
+func TestCacheGetExpired(t *testing.T) {
+	t.Parallel()
+
+	cache := cachePkg.NewCache()
+	cache.Entries["key"] = cachePkg.CacheEntry{
+		Value: "test", StoredAt: time.Now().Add(-24 * time.Hour),
+	}
+	_, found := cache.Get("key")
+	require.False(t, found)
+}

pkg/data_fetcher/data_fetcher.go

@@ -48,7 +48,7 @@ func NewDataFetcher(
 		Logger: logger.With().
 			Str("component", "data_fetcher").
 			Logger(),
-		Cache:                 cache.NewCache(logger),
+		Cache:                 cache.NewCache(),
 		PriceFetchers:         map[string]priceFetchers.PriceFetcher{},
 		Config:                config,
 		TendermintApiClients:  tendermintApiClients,