feat: complete client runtime foundation

tests/test_daemon.py

@@ -2,6 +2,7 @@ import asyncio
 import os
 import tempfile
 import unittest
+from datetime import datetime, timezone
 from pathlib import Path, PurePosixPath
 from uuid import uuid4
 
@@ -10,13 +11,15 @@ from websockets.asyncio.server import serve
 from archive_clients.config import ClientConfig, ConnectionConfig, ServiceConfig
 from archive_clients.daemon import ArchiveClientDaemon
 from archive_clients.probes import FilesystemProbe
-from archive_clients.protocol import decode, encode, new_envelope
-from archive_control.v1 import client_pb2, common_pb2, control_pb2
+from archive_clients.services import ServiceProbe
+from archive_clients.protocol import decode, encode, encode_message, new_envelope
+from archive_control.v1 import client_pb2, common_pb2, control_pb2, job_pb2
 
 
 class DaemonTransportTests(unittest.IsolatedAsyncioTestCase):
     async def test_registration_heartbeat_and_duplicate_command(self):
         observed = {}
+        job_id = str(uuid4())
 
         async def control(websocket):
             registration = decode(await websocket.recv())
@@ -29,6 +32,13 @@ class DaemonTransportTests(unittest.IsolatedAsyncioTestCase):
                 registration.register_request.capabilities
                 .syncthing_advertised_addresses
             )
+            observed["device_id"] = (
+                registration.register_request.capabilities.syncthing_device_id
+            )
+            observed["services"] = [
+                item.service
+                for item in registration.register_request.capabilities.services
+            ]
             response = new_envelope()
             response.correlation_id = registration.message_id
             response.register_response.status = client_pb2.REGISTRATION_STATUS_ACCEPTED
@@ -42,16 +52,21 @@ class DaemonTransportTests(unittest.IsolatedAsyncioTestCase):
             command = new_envelope()
             command.command.command_id = command_id
             command.command.created_at.CopyFrom(command.sent_at)
-            command.command.request_job_snapshot.job_ids.append(str(uuid4()))
+            command.command.request_job_snapshot.job_ids.append(job_id)
             await websocket.send(encode(command))
             observed["first"] = decode(await websocket.recv()).command_ack.status
-            observed["snapshot"] = (
-                decode(await websocket.recv()).WhichOneof("payload")
+            snapshot = decode(await websocket.recv())
+            observed["snapshot"] = snapshot.WhichOneof("payload")
+            observed["snapshot_job_id"] = (
+                snapshot.job_snapshot.job.definition.job_id
             )
             duplicate = new_envelope()
             duplicate.command.CopyFrom(command.command)
             await websocket.send(encode(duplicate))
             observed["second"] = decode(await websocket.recv()).command_ack.status
+            observed["duplicate_snapshot"] = (
+                decode(await websocket.recv()).WhichOneof("payload")
+            )
             unsupported = new_envelope()
             unsupported.command.command_id = str(uuid4())
             unsupported.command.created_at.CopyFrom(unsupported.sent_at)
@@ -83,18 +98,42 @@ class DaemonTransportTests(unittest.IsolatedAsyncioTestCase):
                     root / "state.db", root / "backups", service, service,
                     ConnectionConfig(registration_timeout=2),
                 )
-                probe = FilesystemProbe(root, True, True, True, True)
-                daemon = ArchiveClientDaemon(config, [probe, probe])
+                probe = FilesystemProbe(root, True, True, True, True, True)
+                service_probe = ServiceProbe(
+                    "syncthing", common_pb2.HEALTH_STATE_HEALTHY,
+                    datetime.now(timezone.utc), version="v2", device_id="DEVICE",
+                )
+                daemon = ArchiveClientDaemon(
+                    config, [probe, probe], [service_probe]
+                )
                 await asyncio.to_thread(daemon.store.initialize)
+                definition = job_pb2.JobDefinition(
+                    job_id=job_id,
+                    operation=job_pb2.JOB_OPERATION_ARCHIVE,
+                )
+                definition.created_at.GetCurrentTime()
+                await asyncio.to_thread(
+                    daemon.store.save_job,
+                    job_id,
+                    encode_message(definition),
+                    "JOB_STATE_WAITING",
+                    2,
+                    3,
+                    False,
+                )
                 await daemon._connection()
 
         self.assertEqual(observed["token"], "shared-secret")
         self.assertEqual(observed["root_names"], ["qbittorrent", "syncthing"])
         self.assertEqual(observed["addresses"], ["dynamic"])
+        self.assertEqual(observed["device_id"], "DEVICE")
+        self.assertEqual(observed["services"], ["syncthing"])
         self.assertEqual(observed["heartbeat"], 7)
         self.assertEqual(observed["first"], control_pb2.COMMAND_ACK_STATUS_ACCEPTED)
         self.assertEqual(observed["second"], control_pb2.COMMAND_ACK_STATUS_DUPLICATE)
-        self.assertEqual(observed["snapshot"], "client_state_snapshot")
+        self.assertEqual(observed["snapshot"], "job_snapshot")
+        self.assertEqual(observed["snapshot_job_id"], job_id)
+        self.assertEqual(observed["duplicate_snapshot"], "job_snapshot")
         self.assertEqual(
             observed["rejected"],
             (