Improve tests

s12v · Jul 30, 2018 · 856e5f0 · 856e5f0
1 parent ea5fa8f
commit 856e5f0
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/cache_test.go b/cache_test.go
@@ -6,13 +6,13 @@ import (
 	"time"
 )
 
-func TestDefaultCache_Get(t *testing.T) {
+func TestDefaultCache_GetWithExpiration(t *testing.T) {
 	c := &defaultCache{
-		cache.New(1, 0),
+		cache.New(time.Minute, 0),
 	}
 
-	c.cache.Set("key", "val", time.Minute)
-	val, expTime, found := c.cache.GetWithExpiration("key")
+	c.Set("key", "val")
+	val, expTime, found := c.GetWithExpiration("key")
 
 	if expTime.Before(time.Now()) {
 		t.Fatalf("expTime should be after now: %v", expTime)
@@ -27,14 +27,14 @@ func TestDefaultCache_Get(t *testing.T) {
 	}
 }
 
-func TestDefaultCache_GetExpired(t *testing.T) {
+func TestDefaultCache_GetWithExpiration_Expired(t *testing.T) {
 	c := &defaultCache{
-		cache.New(1, 0),
+		cache.New(time.Nanosecond, 0),
 	}
 
-	c.cache.Set("key", "val", time.Nanosecond)
+	c.Set("key", "val")
 	time.Sleep(10 * time.Millisecond)
-	val, expTime, found := c.cache.GetWithExpiration("key")
+	val, expTime, found := c.GetWithExpiration("key")
 
 	if expTime.After(time.Now()) {
 		t.Fatalf("expTime should be before now: %v", expTime)