Test cache more

db_test.go

@@ -2,10 +2,11 @@ package main
 
 import (
 	"testing"
+	"time"
 
+	logger "git.site.quack-lab.dev/dave/cylogger"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
-	logger "git.site.quack-lab.dev/dave/cylogger"
 )
 
 func TestExpandGroups(t *testing.T) {
@@ -41,7 +42,7 @@ func TestGetModuleSlots(t *testing.T) {
 		require.NoError(t, err)
 		slots, err := db.GetModuleSlots([]int64{11357})
 		require.NoError(t, err)
-		assert.Equal(t, ModuleSlotHigh, slots[11357])		
+		assert.Equal(t, ModuleSlotHigh, slots[11357])
 	})
 	t.Run("Get slots for 4391 -> Med", func(t *testing.T) {
 		// Large Ancillary Shield Booster
@@ -49,7 +50,7 @@ func TestGetModuleSlots(t *testing.T) {
 		require.NoError(t, err)
 		slots, err := db.GetModuleSlots([]int64{4391})
 		require.NoError(t, err)
-		assert.Equal(t, ModuleSlotMid, slots[4391])		
+		assert.Equal(t, ModuleSlotMid, slots[4391])
 	})
 	t.Run("Get slots for 4393-> Low", func(t *testing.T) {
 		// Drone Damage Amplifier I
@@ -57,7 +58,7 @@ func TestGetModuleSlots(t *testing.T) {
 		require.NoError(t, err)
 		slots, err := db.GetModuleSlots([]int64{4393})
 		require.NoError(t, err)
-		assert.Equal(t, ModuleSlotLow, slots[4393])		
+		assert.Equal(t, ModuleSlotLow, slots[4393])
 	})
 	t.Run("Get slots for 25908 -> Rig", func(t *testing.T) {
 		// Large Drone Control Range Augmentor I
@@ -65,7 +66,7 @@ func TestGetModuleSlots(t *testing.T) {
 		require.NoError(t, err)
 		slots, err := db.GetModuleSlots([]int64{25908})
 		require.NoError(t, err)
-		assert.Equal(t, ModuleSlotRig, slots[25908])		
+		assert.Equal(t, ModuleSlotRig, slots[25908])
 	})
 	t.Run("Get slots for all 3 combined -> High, Mid, Low, Rig", func(t *testing.T) {
 		db, err := GetDB()
@@ -84,7 +85,7 @@ func TestGetModuleSlots(t *testing.T) {
 		require.NoError(t, err)
 		slots, err := db.GetModuleSlots([]int64{2454})
 		require.NoError(t, err)
-		assert.Equal(t, ModuleSlotDrone, slots[2454])		
+		assert.Equal(t, ModuleSlotDrone, slots[2454])
 	})
 }
 
@@ -106,4 +107,22 @@ func TestCache(t *testing.T) {
 		require.NoError(t, err)
 		assert.Equal(t, []byte("test"), cache)
 	})
-}
+	t.Run("Test clean", func(t *testing.T) {
+		db, err := GetDB()
+		require.NoError(t, err)
+
+		err = db.CacheSet("test", []byte("test"))
+		require.NoError(t, err)
+
+		expiredTime := time.Now().Add(-4 * 24 * time.Hour)
+		db.Get().
+			Model(&CacheEntry{}).
+			Update("created_at", expiredTime)
+
+		err = db.CacheClean()
+		require.NoError(t, err)
+
+		_, err = db.CacheGet("test")
+		assert.ErrorIs(t, err, ErrCacheMiss)
+	})
+}