Improve(fakeip): use lru cache to avoid outdate

common/cache/cache_test.go

@@ -4,6 +4,8 @@ import (
 	"runtime"
 	"testing"
 	"time"
+
+	"github.com/stretchr/testify/assert"
 )
 
 func TestCache_Basic(t *testing.T) {
@@ -14,32 +16,30 @@ func TestCache_Basic(t *testing.T) {
 	c.Put("string", "a", ttl)
 
 	i := c.Get("int")
-	if i.(int) != 1 {
-		t.Error("should recv 1")
-	}
+	assert.Equal(t, i.(int), 1, "should recv 1")
 
 	s := c.Get("string")
-	if s.(string) != "a" {
-		t.Error("should recv 'a'")
-	}
+	assert.Equal(t, s.(string), "a", "should recv 'a'")
 }
 
 func TestCache_TTL(t *testing.T) {
 	interval := 200 * time.Millisecond
 	ttl := 20 * time.Millisecond
+	now := time.Now()
 	c := New(interval)
 	c.Put("int", 1, ttl)
+	c.Put("int2", 2, ttl)
 
 	i := c.Get("int")
-	if i.(int) != 1 {
-		t.Error("should recv 1")
-	}
+	_, expired := c.GetWithExpire("int2")
+	assert.Equal(t, i.(int), 1, "should recv 1")
+	assert.True(t, now.Before(expired))
 
 	time.Sleep(ttl * 2)
 	i = c.Get("int")
-	if i != nil {
-		t.Error("should recv nil")
-	}
+	j, _ := c.GetWithExpire("int2")
+	assert.Nil(t, i, "should recv nil")
+	assert.Nil(t, j, "should recv nil")
 }
 
 func TestCache_AutoCleanup(t *testing.T) {
@@ -50,9 +50,9 @@ func TestCache_AutoCleanup(t *testing.T) {
 
 	time.Sleep(ttl * 2)
 	i := c.Get("int")
-	if i != nil {
-		t.Error("should recv nil")
-	}
+	j, _ := c.GetWithExpire("int")
+	assert.Nil(t, i, "should recv nil")
+	assert.Nil(t, j, "should recv nil")
 }
 
 func TestCache_AutoGC(t *testing.T) {