diff options
Diffstat (limited to 'tests/test_ttl.py')
-rw-r--r-- | tests/test_ttl.py | 12 |
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)) |