Интегрирован wesp в сайт

WESP_REL/tests/test_pi_boot_config.py

@@ -0,0 +1,67 @@
+"""Тесты правки disable_splash в config.txt."""
+
+from __future__ import annotations
+
+import tempfile
+import unittest
+from pathlib import Path
+
+from app.services.pi_boot_config_service import (
+    _parse_disable_splash_enabled,
+    _patch_disable_splash,
+    get_rainbow_splash_status,
+    set_rainbow_splash_disabled,
+)
+
+
+SAMPLE_CONFIG = """# comment
+dtparam=audio=on
+
+[cm4]
+otg_mode=1
+
+[all]
+enable_uart=1
+"""
+
+
+class PiBootConfigPatchTests(unittest.TestCase):
+    def test_parse_disabled(self) -> None:
+        self.assertFalse(_parse_disable_splash_enabled(SAMPLE_CONFIG))
+        self.assertTrue(
+            _parse_disable_splash_enabled(SAMPLE_CONFIG + "\ndisable_splash=1\n")
+        )
+
+    def test_patch_enable_inserts_under_all(self) -> None:
+        out = _patch_disable_splash(SAMPLE_CONFIG, True)
+        self.assertIn("disable_splash=1", out)
+        lines = out.splitlines()
+        all_idx = next(i for i, l in enumerate(lines) if l.strip() == "[all]")
+        splash_idx = next(i for i, l in enumerate(lines) if l.strip() == "disable_splash=1")
+        self.assertEqual(splash_idx, all_idx + 1)
+
+    def test_patch_disable_removes_line(self) -> None:
+        enabled = _patch_disable_splash(SAMPLE_CONFIG, True)
+        disabled = _patch_disable_splash(enabled, False)
+        self.assertNotIn("disable_splash", disabled)
+
+    def test_set_on_temp_file(self) -> None:
+        with tempfile.TemporaryDirectory() as tmp:
+            path = Path(tmp) / "config.txt"
+            path.write_text(SAMPLE_CONFIG, encoding="utf-8")
+            result = set_rainbow_splash_disabled(True, config_path=str(path))
+            self.assertTrue(result.get("ok"))
+            text = path.read_text(encoding="utf-8")
+            self.assertTrue(_parse_disable_splash_enabled(text))
+
+    def test_status_temp_file(self) -> None:
+        with tempfile.TemporaryDirectory() as tmp:
+            path = Path(tmp) / "config.txt"
+            path.write_text(SAMPLE_CONFIG, encoding="utf-8")
+            st = get_rainbow_splash_status(str(path))
+            self.assertTrue(st["config_found"])
+            self.assertFalse(st["disable_splash"])
+
+
+if __name__ == "__main__":
+    unittest.main()