diff --git a/ai-hub/integration_tests/test_node_registration.py b/ai-hub/integration_tests/test_node_registration.py index 6c2c50a..e6c325b 100644 --- a/ai-hub/integration_tests/test_node_registration.py +++ b/ai-hub/integration_tests/test_node_registration.py @@ -191,9 +191,11 @@ del_r = client.delete(f"{BASE_URL}/nodes/admin/{node_id}", params={"admin_id": user_id}, headers=_headers()) assert del_r.status_code == 200 - # POST /nodes/admin/mesh/reset - reset_r = client.post(f"{BASE_URL}/nodes/admin/mesh/reset", params={"admin_id": user_id}, headers=_headers()) - assert reset_r.status_code == 200 - finally: subprocess.run(["docker", "rm", "-f", node_id], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL) + # Re-map the user preferences back to test-node-1 which is the stable background runner + try: + with httpx.Client() as client: + client.patch(f"{BASE_URL}/nodes/preferences", params={"user_id": user_id}, json={"default_node_ids": ["test-node-1"]}, headers=_headers()) + except Exception: + pass