cli org test fix (#13887)

SatelliteQE · Jan 24, 2024 · e6df221 · e6df221
1 parent ff8b7f6
commit e6df221
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/foreman/cli/test_organization.py b/tests/foreman/cli/test_organization.py
@@ -520,7 +520,13 @@ def test_positive_add_and_remove_locations(module_org, module_target_sat):
         {'location': locations[1]['name'], 'id': module_org.id}
     )
     org_info = module_target_sat.cli.Org.info({'id': module_org.id})
-    assert not org_info.get('locations'), "Failed to remove locations"
+    found_locations = (
+        org_info.get('locations')
+        if isinstance(org_info.get('locations'), list)
+        else [org_info.get('locations')]
+    )
+    assert locations[0]['name'] not in found_locations, "Failed to remove locations"
+    assert locations[1]['name'] not in found_locations, "Failed to remove locations"
 
 
 @pytest.mark.tier1