staging

src/aiogram_dialog/context/media_storage.py

@@ -21,6 +21,8 @@ async def get_media_id(
         if not path and not url:
             return None
         cached = self.cache.get((path, url, type))
+        if cached is None:
+            return None
         if cached[1] is not None:
             mtime = self._get_file_mtime(path)
             if mtime is not None and mtime > cached[1]:

tests/widgets/media/test_media_storage.py

@@ -0,0 +1,44 @@
+import os
+import tempfile
+
+import pytest
+from aiogram.enums import ContentType
+
+from aiogram_dialog.context.media_storage import MediaIdStorage
+
+
+@pytest.mark.asyncio
+async def test_get_media_id():
+    manager = MediaIdStorage()
+    with tempfile.NamedTemporaryFile() as file:
+        os.fsync(file)
+        media_id = await manager.get_media_id(
+            file.name,
+            None,
+            ContentType.DOCUMENT,
+        )
+        assert media_id is None
+
+        await manager.save_media_id(
+            file.name,
+            None,
+            ContentType.DOCUMENT,
+            "test1",
+        )
+
+        media_id = await manager.get_media_id(
+            file.name,
+            None,
+            ContentType.DOCUMENT,
+        )
+        assert media_id == "test1"
+
+        file.write(b"new info")
+        file.flush()
+
+        media_id = await manager.get_media_id(
+            file.name,
+            None,
+            ContentType.DOCUMENT,
+        )
+        assert media_id is None