bf51655a7b
Signed-off-by: Jacek J. Łakis <jacek.lakis@intel.com> Signed-off-by: Samuel Ortiz <sameo@linux.intel.com>
68 lines
1.9 KiB
Go
68 lines
1.9 KiB
Go
/*
|
|
Copyright 2016 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package cache
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"k8s.io/client-go/util/clock"
|
|
|
|
"github.com/golang/groupcache/lru"
|
|
)
|
|
|
|
func expectEntry(t *testing.T, c *LRUExpireCache, key lru.Key, value interface{}) {
|
|
result, ok := c.Get(key)
|
|
if !ok || result != value {
|
|
t.Errorf("Expected cache[%v]: %v, got %v", key, value, result)
|
|
}
|
|
}
|
|
|
|
func expectNotEntry(t *testing.T, c *LRUExpireCache, key lru.Key) {
|
|
if result, ok := c.Get(key); ok {
|
|
t.Errorf("Expected cache[%v] to be empty, got %v", key, result)
|
|
}
|
|
}
|
|
|
|
func TestSimpleGet(t *testing.T) {
|
|
c := NewLRUExpireCache(10)
|
|
c.Add("long-lived", "12345", 10*time.Hour)
|
|
expectEntry(t, c, "long-lived", "12345")
|
|
}
|
|
|
|
func TestExpiredGet(t *testing.T) {
|
|
fakeClock := clock.NewFakeClock(time.Now())
|
|
c := NewLRUExpireCacheWithClock(10, fakeClock)
|
|
c.Add("short-lived", "12345", 1*time.Millisecond)
|
|
// ensure the entry expired
|
|
fakeClock.Step(2 * time.Millisecond)
|
|
expectNotEntry(t, c, "short-lived")
|
|
}
|
|
|
|
func TestLRUOverflow(t *testing.T) {
|
|
c := NewLRUExpireCache(4)
|
|
c.Add("elem1", "1", 10*time.Hour)
|
|
c.Add("elem2", "2", 10*time.Hour)
|
|
c.Add("elem3", "3", 10*time.Hour)
|
|
c.Add("elem4", "4", 10*time.Hour)
|
|
c.Add("elem5", "5", 10*time.Hour)
|
|
expectNotEntry(t, c, "elem1")
|
|
expectEntry(t, c, "elem2", "2")
|
|
expectEntry(t, c, "elem3", "3")
|
|
expectEntry(t, c, "elem4", "4")
|
|
expectEntry(t, c, "elem5", "5")
|
|
}
|