aboutsummaryrefslogtreecommitdiff
path: root/tests/test_ttl.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_ttl.py')
-rw-r--r--tests/test_ttl.py12
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/test_ttl.py b/tests/test_ttl.py
index f677c9b..6e51a59 100644
--- a/tests/test_ttl.py
+++ b/tests/test_ttl.py
@@ -182,3 +182,15 @@ class TTLCacheTest(unittest.TestCase, CacheTestMixin):
with self.assertRaises(KeyError):
cache[(1, 2, 3)]
self.assertNotIn((1, 2, 3), cache)
+
+ def test_ttl_datetime(self):
+ from datetime import datetime, timedelta
+
+ cache = TTLCache(maxsize=1, ttl=timedelta(days=1), timer=datetime.now)
+
+ cache[1] = 1
+ self.assertEqual(1, len(cache))
+ cache.expire(datetime.now())
+ self.assertEqual(1, len(cache))
+ cache.expire(datetime.now() + timedelta(days=1))
+ self.assertEqual(0, len(cache))