diff --git a/pytest_tests/testsuites/container/test_policy.py b/pytest_tests/testsuites/container/test_policy.py index 52fe792f..2bf01480 100644 --- a/pytest_tests/testsuites/container/test_policy.py +++ b/pytest_tests/testsuites/container/test_policy.py @@ -1486,11 +1486,16 @@ class TestPolicy(ClusterTestBase): with allure.step(f"Check the object appearance"): netmap = parse_netmap_output(get_netmap_snapshot(node=resulting_copies[0], shell=self.shell)) netmap = self.get_netmap_param(netmap) + list_of_location = [] for node in resulting_copies: node_address = node.get_rpc_endpoint().split(":")[0] - assert expected_params["location"][0] == netmap[node_address]["location"] or ( - expected_params["location"][1] == netmap[node_address]["location"] - ), f"The node is selected from the wrong location. Expected {expected_params} and got {netmap[node_address]}" + list_of_location.append(netmap[node_address]["location"]) + + assert ( + expected_params["location"][0] in list_of_location + and expected_params["location"][1] in list_of_location + and len(resulting_copies) > 2 + ), f"The node is selected from the wrong location. Expected {expected_params} and got {netmap[node_address]}" self.check_for_the_uniqueness_of_the_nodes(resulting_copies)