test: test cache expiry when fetching zones/schedules

jnsgruk · Dec 3, 2024 · 17dbb66 · 17dbb66
1 parent df2dafd
commit 17dbb66
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tests/test_module.py b/tests/test_module.py
@@ -45,6 +45,16 @@ async def test_zones_cache(test_module):
     assert initial_fetch_time == test_module._last_fetched
 
 
+@pytest.mark.asyncio
+async def test_zones_cache_expired(test_module_short_cache):
+    await test_module_short_cache.zones()
+    initial_fetch_time = test_module_short_cache._last_fetched
+    await asyncio.sleep(0.25)
+
+    await test_module_short_cache.zones()
+    assert initial_fetch_time != test_module_short_cache._last_fetched
+
+
 @pytest.mark.asyncio
 async def test_zones_force_refresh(test_module):
     await test_module.zones()
@@ -119,6 +129,16 @@ async def test_schedules_cache(test_module):
     assert initial_fetch_time == test_module._last_fetched
 
 
+@pytest.mark.asyncio
+async def test_schedules_cache_expired(test_module_short_cache):
+    await test_module_short_cache.schedules()
+    initial_fetch_time = test_module_short_cache._last_fetched
+    await asyncio.sleep(0.5)
+
+    await test_module_short_cache.schedules()
+    assert initial_fetch_time != test_module_short_cache._last_fetched
+
+
 @pytest.mark.asyncio
 async def test_schedules_force_refresh(test_module):
     await test_module.schedules()