feat: complete client runtime foundation

tests/test_services.py

@@ -0,0 +1,92 @@
+import io
+import os
+import tempfile
+import unittest
+from pathlib import Path, PurePosixPath
+from unittest.mock import patch
+
+from archive_clients.config import ServiceConfig
+from archive_clients.services import probe_qbittorrent, probe_syncthing
+from archive_control.v1 import common_pb2
+
+
+class _Response:
+    def __init__(self, value: str):
+        self._source = io.BytesIO(value.encode("utf-8"))
+
+    def read(self, size: int = -1) -> bytes:
+        return self._source.read(size)
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *_):
+        return None
+
+
+class _Opener:
+    def __init__(self, responses: list[str]):
+        self.responses = iter(responses)
+        self.requests = []
+
+    def open(self, call, timeout):
+        self.requests.append((call, timeout))
+        return _Response(next(self.responses))
+
+
+class ServiceProbeTests(unittest.TestCase):
+    def test_qbittorrent_versions_are_normalized(self):
+        with tempfile.TemporaryDirectory() as directory:
+            root = Path(directory)
+            password = self._secret(root, "password", "private")
+            config = ServiceConfig(
+                "http://qb:8080", PurePosixPath("/downloads"), root,
+                username="admin", password_file=password,
+            )
+            opener = _Opener([
+                "Ok.", "v5.0.4", "2.11.4", '{"libtorrent":"2.0.11"}',
+            ])
+            with patch(
+                "archive_clients.services.request.build_opener",
+                return_value=opener,
+            ):
+                result = probe_qbittorrent(config)
+            self.assertEqual(result.state, common_pb2.HEALTH_STATE_HEALTHY)
+            self.assertEqual(result.version, "v5.0.4")
+            self.assertEqual(result.api_version, "2.11.4")
+            self.assertEqual(result.libtorrent_version, "2.0.11")
+            self.assertIn(b"password=private", opener.requests[0][0].data)
+
+    def test_syncthing_identity_is_normalized(self):
+        with tempfile.TemporaryDirectory() as directory:
+            root = Path(directory)
+            api_key = self._secret(root, "api-key", "private")
+            config = ServiceConfig(
+                "http://syncthing:8384", PurePosixPath("/sync"), root,
+                api_key_file=api_key,
+            )
+            opener = _Opener([
+                '{"version":"v2.0.1","longVersion":"syncthing v2.0.1"}',
+                '{"myID":"DEVICE-ID"}',
+            ])
+            with patch(
+                "archive_clients.services.request.build_opener",
+                return_value=opener,
+            ):
+                result = probe_syncthing(config)
+            self.assertEqual(result.state, common_pb2.HEALTH_STATE_HEALTHY)
+            self.assertEqual(result.device_id, "DEVICE-ID")
+            self.assertEqual(
+                opener.requests[0][0].headers["X-api-key"], "private"
+            )
+
+    @staticmethod
+    def _secret(root: Path, name: str, value: str) -> Path:
+        path = root / name
+        path.write_text(value, encoding="utf-8")
+        os.chmod(path, 0o600)
+        return path
+
+
+if __name__ == "__main__":
+    unittest.main()