feat: background remote agent updates survive SAC navigation (0.20.19)

backend/tests/test_agent_update.py

@@ -1,5 +1,6 @@
 """Tests for agent update settings, poll, request and ingest lifecycle."""
 
+import time
 import uuid
 from datetime import datetime, timedelta, timezone
 from unittest.mock import patch
@@ -17,6 +18,21 @@ from app.services.ingest import ingest_event
 from tests.test_ingest import VALID_EVENT
 
 
+def wait_remote_job(client, host_id, headers, timeout=5.0):
+    deadline = time.monotonic() + timeout
+    while time.monotonic() < deadline:
+        response = client.get(
+            f"/api/v1/hosts/{host_id}/actions/remote-job",
+            headers=headers,
+        )
+        assert response.status_code == 200
+        body = response.json()
+        if not body.get("active") and body.get("status") != "running":
+            return body
+        time.sleep(0.05)
+    raise AssertionError("remote job did not finish in time")
+
+
 def test_agent_update_settings_sac_mode(db_session):
     cfg = upsert_agent_update_settings(
         db_session,
@@ -203,6 +219,8 @@ def test_api_agent_update_fallback_ssh(jwt_headers, client, db_session, monkeypa
             headers=jwt_headers,
         )
 
-    assert response.status_code == 200
-    assert response.json()["ok"] is True
-    assert response.json()["product_version"] == "2.1.0-SAC"
+    assert response.status_code == 202
+    assert response.json()["status"] == "running"
+    job = wait_remote_job(client, host.id, jwt_headers)
+    assert job["ok"] is True
+    assert job["product_version"] == "2.1.0-SAC"