初始化奇妙小屏幕控制器项目

核桃派软件源代码/tests/test_device_storage.py

@@ -0,0 +1,63 @@
+from types import SimpleNamespace
+
+from app.monitoring.storage import DeviceStorageMonitor, directory_bytes
+
+
+def test_directory_bytes_deduplicates_nested_roots_and_counts_template_content(tmp_path):
+    program = tmp_path / "program"
+    data = program / "data"
+    templates = data / "templates"
+    animations = data / "animations"
+    templates.mkdir(parents=True)
+    animations.mkdir()
+    (program / "app.py").write_bytes(b"app")
+    (data / "config.json").write_bytes(b"config")
+    (templates / "one.json").write_bytes(b"template")
+    (animations / "frame.json").write_bytes(b"animation")
+
+    assert directory_bytes([program, data]) == 3 + 6 + 8 + 9
+    assert directory_bytes([templates, animations]) == 8 + 9
+
+
+def test_device_storage_monitor_caches_expires_and_invalidates(tmp_path):
+    program = tmp_path / "program"
+    data = tmp_path / "data"
+    program.mkdir()
+    data.mkdir()
+    now = [10.0]
+    calls = []
+
+    def directory_reader(paths):
+        resolved = tuple(str(path) for path in paths)
+        calls.append(resolved)
+        return 100 if len(calls) % 2 else 25
+
+    monitor = DeviceStorageMonitor(
+        program,
+        data,
+        cache_seconds=300,
+        clock=lambda: now[0],
+        usage_reader=lambda _path: SimpleNamespace(total=1000, used=400, free=600),
+        directory_reader=directory_reader,
+    )
+
+    first = monitor.get_status()
+    second = monitor.get_status()
+    assert first == second
+    assert first["device_total_bytes"] == 1000
+    assert first["device_free_bytes"] == 600
+    assert first["software_bytes"] == 100
+    assert first["templates_bytes"] == 25
+    assert len(calls) == 2
+
+    now[0] += 299
+    assert monitor.get_status() == first
+    assert len(calls) == 2
+
+    now[0] += 1
+    monitor.get_status()
+    assert len(calls) == 4
+
+    monitor.invalidate()
+    monitor.get_status()
+    assert len(calls) == 6