mihomo/common/cache/cache_test.go

package cache

import (
	"runtime"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestCache_Basic(t *testing.T) {
	interval := 200 * time.Millisecond
	ttl := 20 * time.Millisecond
	c := New[string, int](interval)
	c.Put("int", 1, ttl)

	d := New[string, string](interval)
	d.Put("string", "a", ttl)

	i := c.Get("int")
	assert.Equal(t, i, 1, "should recv 1")

	s := d.Get("string")
	assert.Equal(t, s, "a", "should recv 'a'")
}

func TestCache_TTL(t *testing.T) {
	interval := 200 * time.Millisecond
	ttl := 20 * time.Millisecond
	now := time.Now()
	c := New[string, int](interval)
	c.Put("int", 1, ttl)
	c.Put("int2", 2, ttl)

	i := c.Get("int")
	_, expired := c.GetWithExpire("int2")
	assert.Equal(t, i, 1, "should recv 1")
	assert.True(t, now.Before(expired))

	time.Sleep(ttl * 2)
	i = c.Get("int")
	j, _ := c.GetWithExpire("int2")
	assert.True(t, i == 0, "should recv 0")
	assert.True(t, j == 0, "should recv 0")
}

func TestCache_AutoCleanup(t *testing.T) {
	interval := 10 * time.Millisecond
	ttl := 15 * time.Millisecond
	c := New[string, int](interval)
	c.Put("int", 1, ttl)

	time.Sleep(ttl * 2)
	i := c.Get("int")
	j, _ := c.GetWithExpire("int")
	assert.True(t, i == 0, "should recv 0")
	assert.True(t, j == 0, "should recv 0")
}

func TestCache_AutoGC(t *testing.T) {
	sign := make(chan struct{})
	go func() {
		interval := 10 * time.Millisecond
		ttl := 15 * time.Millisecond
		c := New[string, int](interval)
		c.Put("int", 1, ttl)
		sign <- struct{}{}
	}()

	<-sign
	runtime.GC()
}