integration test: allow pro service warnings (#4447)

Even if pro is giving us warnings, that is showing that cloud-init enabled things correctly.
canonical · Sep 19, 2023 · 67bb6aa · 67bb6aa
1 parent 5be5f58
commit 67bb6aa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/integration_tests/modules/test_ubuntu_advantage.py b/tests/integration_tests/modules/test_ubuntu_advantage.py
@@ -112,7 +112,8 @@ def get_services_status(client: IntegrationInstance) -> dict:
     assert status_resp.ok
     status = json.loads(status_resp.stdout)
     return {
-        svc["name"]: svc["status"] == "enabled" for svc in status["services"]
+        svc["name"]: svc["status"] in ["enabled", "warning"]
+        for svc in status["services"]
     }