Merge pull request #20 from fastbill/fix-cache-mock

Fix mock for cache TTL method
fastbill · Feb 28, 2020 · 011c372 · 011c372
2 parents 27ac336 + 692e813
commit 011c372
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/cache/cachemock/cache.go b/cache/cachemock/cache.go
@@ -92,11 +92,7 @@ func (m *Cache) SetInt(key string, value int64, expiration time.Duration) error
 func (m *Cache) TTL(key string) (time.Duration, error) {
 	args := m.Called(key)
 
-	if args.Get(0) != nil {
-		return args.Get(0).(time.Duration), args.Error(1)
-	}
-
-	return 0, args.Error(1)
+	return args.Get(0).(time.Duration), args.Error(1)
 }
 
 // SetJSON is a mock implementation of cache.Cache#SetJSON.