diff --git a/db_test.go b/db_test.go index 736f906..e7ebc80 100644 --- a/db_test.go +++ b/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) }) -} \ No newline at end of file + 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) + }) +}