From 5c411b28acd49582b27e3ad7ac26412a63df22ba Mon Sep 17 00:00:00 2001
From: lorneli <lorneli@163.com>
Date: Tue, 31 Oct 2017 22:00:35 +0800
Subject: [PATCH] lru: add evict test

Test evict cached keys in least-recently-used way.
---
 lru/lru_test.go | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/lru/lru_test.go b/lru/lru_test.go
index 98a2656..b7d9d8a 100644
--- a/lru/lru_test.go
+++ b/lru/lru_test.go
@@ -17,6 +17,7 @@ limitations under the License.
 package lru
 
 import (
+	"fmt"
 	"testing"
 )
 
@@ -71,3 +72,26 @@ func TestRemove(t *testing.T) {
 		t.Fatal("TestRemove returned a removed entry")
 	}
 }
+
+func TestEvict(t *testing.T) {
+	evictedKeys := make([]Key, 0)
+	onEvictedFun := func(key Key, value interface{}) {
+		evictedKeys = append(evictedKeys, key)
+	}
+
+	lru := New(20)
+	lru.OnEvicted = onEvictedFun
+	for i := 0; i < 22; i++ {
+		lru.Add(fmt.Sprintf("myKey%d", i), 1234)
+	}
+
+	if len(evictedKeys) != 2 {
+		t.Fatalf("got %d evicted keys; want 2", len(evictedKeys))
+	}
+	if evictedKeys[0] != Key("myKey0") {
+		t.Fatalf("got %v in first evicted key; want %s", evictedKeys[0], "myKey0")
+	}
+	if evictedKeys[1] != Key("myKey1") {
+		t.Fatalf("got %v in second evicted key; want %s", evictedKeys[1], "myKey1")
+	}
+}