diff --git a/pytest_tests/testsuites/container/test_policy.py b/pytest_tests/testsuites/container/test_policy.py index 765be4c4..7342767e 100644 --- a/pytest_tests/testsuites/container/test_policy.py +++ b/pytest_tests/testsuites/container/test_policy.py @@ -182,7 +182,7 @@ class TestPolicy(ClusterTestBase): file_path = generate_file(simple_object_size.value) expected_copies = 1 endpoint = self.cluster.default_rpc_endpoint - expected_params = {"country": "Russia"} + placement_params = {"country": "Russia"} with reporter.step(f"Create container with policy {placement_rule}"): cid = create_container( @@ -205,9 +205,10 @@ class TestPolicy(ClusterTestBase): netmap = parse_netmap_output(get_netmap_snapshot(node=resulting_copies[0], shell=self.shell)) netmap = self.get_netmap_param(netmap) node_address = resulting_copies[0].get_rpc_endpoint().split(":")[0] - assert ( - expected_params["country"] == netmap[node_address]["country"] - ), f"The node is selected from the wrong country. Expected {expected_params['country']} and got {netmap[node_address]['country']}" + with allure.step(f"Check the node is selected from {placement_params['country']}"): + assert ( + placement_params["country"] == netmap[node_address]["country"] + ), f"The node is selected from the wrong country. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -225,8 +226,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with SELECT and Complex FILTER results with one node. """ placement_rule = "REP 1 IN RUS REP 1 IN RUS CBF 1 SELECT 1 FROM RU AS RUS FILTER Country NE Sweden AS NotSE FILTER @NotSE AND NOT (CountryCode EQ FI) AND Country EQ 'Russia' AS RU" - expected_params = {"country": ["Russia", "Sweden"]} - unexpected_params = {"country": "Sweden", "country_code": "FI"} + placement_params = {"country": ["Russia", "Sweden"], "country_code": "FI"} file_path = generate_file(simple_object_size.value) expected_copies = 1 endpoint = self.cluster.default_rpc_endpoint @@ -252,12 +252,15 @@ class TestPolicy(ClusterTestBase): netmap = parse_netmap_output(get_netmap_snapshot(node=resulting_copies[0], shell=self.shell)) netmap = self.get_netmap_param(netmap) node_address = resulting_copies[0].get_rpc_endpoint().split(":")[0] - assert ( - (expected_params["country"][1] == netmap[node_address]["country"]) - or not (unexpected_params["country"] == netmap[node_address]["country"]) - and not (unexpected_params["country_code"] == netmap[node_address]["country_code"]) - and (expected_params["country"][0] == netmap[node_address]["country"]) - ), f"The node is selected from the wrong country or country code. Expected {expected_params} and got {netmap[node_address]}" + with allure.step( + f"Check the node is selected from {placement_params['country'][0]}" + ): + assert ( + not (placement_params["country"][1] == netmap[node_address]["country"]) + or not (placement_params["country"][1] == netmap[node_address]["country"]) + and not (placement_params["country_code"] == netmap[node_address]["country_code"]) + and (placement_params["country"][0] == netmap[node_address]["country"]) + ), f"The node is selected from the wrong country or country code. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -276,8 +279,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with Multi SELECTs and FILTERs results with one nodes. """ placement_rule = "REP 1 IN RU REP 1 IN EU REP 1 IN RU CBF 1 SELECT 1 FROM RUS AS RU SELECT 1 FROM EUR AS EU FILTER Country EQ Russia AS RUS FILTER NOT (@RUS) AND Country EQ Sweden OR CountryCode EQ FI AS EUR" - expected_params = {"country": ["Sweden", "Russia"], "country_code": "FI"} - unexpected_params = {"country": "Russia"} + placement_params = {"country": ["Sweden", "Russia"], "country_code": "FI"} file_path = generate_file(simple_object_size.value) expected_copies = 2 endpoint = self.cluster.default_rpc_endpoint @@ -302,13 +304,14 @@ class TestPolicy(ClusterTestBase): with reporter.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) - for node in resulting_copies: - node_address = resulting_copies[0].get_rpc_endpoint().split(":")[0] - assert (expected_params["country"][1] == netmap[node_address]["country"]) or ( - not (unexpected_params["country"][1] == netmap[node_address]["country"]) - and (expected_params["country"][0] == netmap[node_address]["country"]) - or (expected_params["country_code"] == netmap[node_address]["country_code"]) - ), f"The node is selected from the wrong country or country code. Expected {expected_params} and got {netmap[node_address]}" + with allure.step(f"Check two nodes are selected from any country"): + for node in resulting_copies: + node_address = resulting_copies[0].get_rpc_endpoint().split(":")[0] + assert (placement_params["country"][1] == netmap[node_address]["country"]) or ( + not (placement_params["country"][1] == netmap[node_address]["country"]) + and (placement_params["country"][0] == netmap[node_address]["country"]) + or (placement_params["country_code"] == netmap[node_address]["country_code"]) + ), f"The node is selected from the wrong country or country code. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -327,12 +330,12 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with SELECT and FILTER results with UNIQUE nodes. """ placement_rule = "UNIQUE REP 1 IN MyRussianNodes REP 1 IN MyRussianNodes CBF 1 SELECT 1 FROM RussianNodes AS MyRussianNodes FILTER Country EQ 'Russia' AS RussianNodes" + placement_params = {"country": "Russia"} file_path = generate_file(simple_object_size.value) expected_copies = 2 endpoint = self.cluster.default_rpc_endpoint - expected_params = {"country": "Russia"} - - with reporter.step(f"Create container with policy {placement_rule}"): + + with allure.step(f"Create container with policy {placement_rule}"): cid = create_container( wallet=default_wallet, rule=placement_rule, basic_acl=PUBLIC_ACL, shell=self.shell, endpoint=endpoint ) @@ -349,15 +352,15 @@ class TestPolicy(ClusterTestBase): len(resulting_copies) == expected_copies ), f"Expected {expected_copies} copies, got {len(resulting_copies)}" - with reporter.step(f"Check the object appearance"): - self.check_for_the_uniqueness_of_the_nodes(resulting_copies) + 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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert ( - expected_params["country"] == netmap[node_address]["country"] - ), f"The node is selected from the wrong country. Expected {expected_params['country']} and got {netmap[node_address]['country']}" + with allure.step(f"Check two nodes are selected from {placement_params['country']}"): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert ( + placement_params["country"] == netmap[node_address]["country"] + ), f"The node is selected from the wrong country. Got {netmap[node_address]['country']}" with reporter.step(f"Delete the container"): delete_container(wallet=default_wallet, cid=cid, shell=self.shell, endpoint=endpoint) @@ -450,7 +453,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with SELECT and FILTER results with 25% of available nodes. """ placement_rule = "REP 1 IN Nodes25 SELECT 1 FROM LE10 AS Nodes25 FILTER Price LE 10 AS LE10" - expected_params = {"price": 10} + placement_params = {"price": 10} file_path = generate_file(simple_object_size.value) expected_copies = 1 endpoint = self.cluster.default_rpc_endpoint @@ -475,11 +478,11 @@ class TestPolicy(ClusterTestBase): with reporter.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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] + node_address = resulting_copies[0].get_rpc_endpoint().split(":")[0] + with allure.step(f"Check the node is selected with price <= {placement_params['price']}"): assert ( - int(netmap[node_address]["price"]) <= expected_params["price"] - ), f"The node is selected from the wrong price. Expected {expected_params} and got {netmap[node_address]}" + int(netmap[node_address]["price"]) <= placement_params["price"] + ), f"The node is selected with the wrong price. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -498,7 +501,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with SELECT and Complex FILTER results with 25% of available nodes. """ placement_rule = "REP 1 IN Nodes25 SELECT 1 FROM BET0AND10 AS Nodes25 FILTER Price LE 10 AS LE10 FILTER Price GT 0 AS GT0 FILTER @LE10 AND @GT0 AS BET0AND10" - expected_params = {"price": [10, 0]} + placement_params = {"price": [10, 0]} file_path = generate_file(simple_object_size.value) expected_copies = 1 endpoint = self.cluster.default_rpc_endpoint @@ -523,16 +526,13 @@ class TestPolicy(ClusterTestBase): with reporter.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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert ( - int(netmap[node_address]["price"]) > expected_params["price"][1] - or int(netmap[node_address]["price"]) <= expected_params["price"][0] - or ( - int(netmap[node_address]["price"]) > expected_params["price"][1] - and int(netmap[node_address]["price"]) <= expected_params["price"][0] - ) - ), f"The node is selected from the wrong price. Expected {expected_params} and got {netmap[node_address]}" + with allure.step(f"Check the node is selected with price between 1 and 10"): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert ( + int(netmap[node_address]["price"]) > placement_params["price"][1] + and int(netmap[node_address]["price"]) <= placement_params["price"][0] + ), f"The node is selected with the wrong price. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -551,7 +551,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with Multi SELECTs and FILTERs results with 25% of available nodes. """ placement_rule = "UNIQUE REP 1 IN One REP 1 IN One CBF 1 SELECT 1 FROM MINMAX AS One FILTER Price LT 15 AS LT15 FILTER Price GT 55 AS GT55 FILTER @LT15 OR @GT55 AS MINMAX" - expected_params = {"price": [15, 55]} + placement_params = {"price": [15, 55]} file_path = generate_file(simple_object_size.value) expected_copies = 2 endpoint = self.cluster.default_rpc_endpoint @@ -576,17 +576,13 @@ class TestPolicy(ClusterTestBase): with reporter.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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert ( - int(netmap[node_address]["price"]) > expected_params["price"][1] - or int(netmap[node_address]["price"]) < expected_params["price"][0] - or ( - int(netmap[node_address]["price"]) > expected_params["price"][1] - or int(netmap[node_address]["price"]) < expected_params["price"][0] - ) - ), f"The node is selected from the wrong price. Expected {expected_params} and got {netmap[node_address]}" - self.check_for_the_uniqueness_of_the_nodes(resulting_copies) + with allure.step(f"Check two nodes are selected with max and min prices"): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert ( + int(netmap[node_address]["price"]) > placement_params["price"][1] + or int(netmap[node_address]["price"]) < placement_params["price"][0] + ), f"The node is selected with the wrong price. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -679,7 +675,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with SELECT and FILTER results with 50% of available nodes. """ placement_rule = "REP 2 IN HALF CBF 1 SELECT 2 FROM GT15 AS HALF FILTER Price GT 15 AS GT15" - expected_params = {"price": 15} + placement_params = {"price": 15} file_path = generate_file(simple_object_size.value) expected_copies = 2 endpoint = self.cluster.default_rpc_endpoint @@ -704,11 +700,12 @@ class TestPolicy(ClusterTestBase): with reporter.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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert ( - int(netmap[node_address]["price"]) > expected_params["price"] - ), f"The node is selected from the wrong price. Expected {expected_params} and got {netmap[node_address]}" + with allure.step(f"Check two nodes are selected with price > {placement_params['price']}"): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert ( + int(netmap[node_address]["price"]) > placement_params["price"] + ), f"The node is selected with the wrong price. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -727,8 +724,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with SELECT and Complex FILTER results with 50% of available nodes. """ placement_rule = "REP 2 IN HALF CBF 2 SELECT 2 FROM GE15 AS HALF FILTER CountryCode NE RU AS NOTRU FILTER @NOTRU AND Price GE 15 AS GE15" - expected_params = {"price": 15} - unexpected_params = {"country_code": "RU"} + placement_params = {"price": 15, "country_code": "RU"} file_path = generate_file(simple_object_size.value) expected_copies = 2 endpoint = self.cluster.default_rpc_endpoint @@ -753,12 +749,13 @@ class TestPolicy(ClusterTestBase): with reporter.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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert (not netmap[node_address]["country_code"] == unexpected_params["country_code"]) or ( - not netmap[node_address]["country_code"] == unexpected_params["country_code"] - and int(netmap[node_address]["price"]) >= expected_params["price"] - ), f"The node is selected from the wrong price or country_code. Expected {expected_params} and got {netmap[node_address]}" + with allure.step(f"Check two nodes are selected not with country code '{placement_params['country_code']}'"): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert (not netmap[node_address]["country_code"] == placement_params["country_code"] + or not netmap[node_address]["country_code"] == placement_params["country_code"] + and int(netmap[node_address]["price"]) >= placement_params["price"] + ), f"The node is selected with the wrong price or country code. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -777,7 +774,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with Multi SELECTs and FILTERs results with 50% of available nodes. """ placement_rule = "REP 2 IN FH REP 1 IN SH CBF 2 SELECT 2 FROM LE55 AS FH SELECT 2 FROM GE15 AS SH FILTER 'UN-LOCODE' EQ RU_LED OR 'UN-LOCODE' EQ RU_MOW AS RU FILTER NOT(@RU) AS NOTRU FILTER @NOTRU AND Price GE 15 AS GE15 FILTER @RU AND Price LE 55 AS LE55" - expected_params = {"un_locode": ["RU_LED", "RU_MOW"], "price": [15, 55]} + placement_params = {"un_locode": ["RU_LED", "RU_MOW"], "price": [15, 55]} file_path = generate_file(simple_object_size.value) expected_copies = 4 endpoint = self.cluster.default_rpc_endpoint @@ -802,20 +799,21 @@ class TestPolicy(ClusterTestBase): with reporter.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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert ( - (netmap[node_address]["un_locode"] in expected_params["un_locode"]) - or (not netmap[node_address]["un_locode"] in expected_params["un_locode"]) - or ( - not netmap[node_address]["un_locode"] in expected_params["un_locode"] - and int(netmap[node_address]["price"]) >= expected_params["price"][0] - ) - or ( - netmap[node_address]["un_locode"] in expected_params["un_locode"] - and int(netmap[node_address]["price"]) <= expected_params["price"][1] - ) - ), f"The node is selected from the wrong price or un_locode. Expected {expected_params} and got {netmap[node_address]}" + with allure.step(f"Check all nodes are selected"): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert ( + netmap[node_address]["un_locode"] in placement_params["un_locode"] + or not netmap[node_address]["un_locode"] == placement_params["un_locode"][1] + or ( + not netmap[node_address]["un_locode"] == placement_params["un_locode"][1] + and int(netmap[node_address]["price"]) >= placement_params["price"][0] + ) + or ( + netmap[node_address]["un_locode"] == placement_params["un_locode"][1] + and int(netmap[node_address]["price"]) <= placement_params["price"][1] + ) + ), f"The node is selected with the wrong price or un_locode. Expected {placement_params} and got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -909,7 +907,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with SELECT and FILTER results with 75% of available nodes. """ placement_rule = "REP 2 IN NODES75 SELECT 2 FROM LT65 AS NODES75 FILTER Price LT 65 AS LT65" - expected_params = {"price": 65} + placement_params = {"price": 65} file_path = generate_file(simple_object_size.value) expected_copies = 2 endpoint = self.cluster.default_rpc_endpoint @@ -934,11 +932,12 @@ class TestPolicy(ClusterTestBase): with reporter.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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert ( - int(netmap[node_address]["price"]) < expected_params["price"] - ), f"The node is selected from the wrong price. Expected {expected_params} and got {netmap[node_address]}" + with allure.step(f"Check two nodes are selected with price < {placement_params['price']}"): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert ( + int(netmap[node_address]["price"]) < placement_params["price"] + ), f"The node is selected with the wrong price. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -957,8 +956,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with SELECT and Complex FILTER results with 75% of available nodes. """ placement_rule = "REP 2 IN NODES75 SELECT 2 FROM LT65 AS NODES75 FILTER Continent NE America AS NOAM FILTER @NOAM AND Price LT 65 AS LT65" - expected_params = {"price": 65} - unexpected_params = {"continent": "America"} + placement_params = {"price": 65, "continent": "America"} file_path = generate_file(simple_object_size.value) expected_copies = 3 endpoint = self.cluster.default_rpc_endpoint @@ -983,14 +981,16 @@ class TestPolicy(ClusterTestBase): with reporter.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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert ( - int(netmap[node_address]["price"]) < expected_params["price"] - and netmap[node_address]["continent"] == unexpected_params["continent"] - ) or ( - netmap[node_address]["continent"] == unexpected_params["continent"] - ), f"The node is selected from the wrong price or continent. Expected {expected_params} and got {netmap[node_address]}" + with allure.step(f"Check three nodes are selected not from {placement_params['continent']}" + ): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert ( + int(netmap[node_address]["price"]) < placement_params["price"] + and not netmap[node_address]["continent"] == placement_params["continent"] + ) or ( + not netmap[node_address]["continent"] == placement_params["continent"] + ), f"The node is selected with the wrong price or continent. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -1009,8 +1009,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with Multi SELECTs and FILTERs results with 75% of available nodes. """ placement_rule = "REP 2 IN EXPNSV REP 2 IN CHEAP SELECT 3 FROM GT10 AS EXPNSV SELECT 3 FROM LT65 AS CHEAP FILTER NOT(Continent EQ America) AS NOAM FILTER @NOAM AND Price LT 65 AS LT65 FILTER @NOAM AND Price GT 10 AS GT10" - expected_params = {"price": [65, 10]} - unexpected_params = {"continent": "America"} + placement_params = {"price": [65, 10], "continent": "America"} file_path = generate_file(simple_object_size.value) expected_copies = 4 endpoint = self.cluster.default_rpc_endpoint @@ -1035,19 +1034,20 @@ class TestPolicy(ClusterTestBase): with reporter.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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert ( - ( - int(netmap[node_address]["price"]) > expected_params["price"][1] - and netmap[node_address]["continent"] == unexpected_params["continent"] - ) - or ( - int(netmap[node_address]["price"]) < expected_params["price"][0] - and netmap[node_address]["continent"] == unexpected_params["continent"] - ) - or (netmap[node_address]["continent"] == unexpected_params["continent"]) - ), f"The node is selected from the wrong price or continent. Expected {expected_params} and got {netmap[node_address]}" + with allure.step(f"Check all nodes are selected"): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert ( + ( + int(netmap[node_address]["price"]) > placement_params["price"][1] + and not netmap[node_address]["continent"] == placement_params["continent"] + ) + or ( + int(netmap[node_address]["price"]) < placement_params["price"][0] + and not netmap[node_address]["continent"] == placement_params["continent"] + ) + or not (netmap[node_address]["continent"] == placement_params["continent"]) + ), f"The node is selected with the wrong price or continent. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -1141,7 +1141,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with SELECT and FILTER results with 100% of available nodes. """ placement_rule = "REP 1 IN All SELECT 4 FROM AllNodes AS All FILTER Price GE 0 AS AllNodes" - expected_params = {"price": 0} + placement_params = {"price": 0} file_path = generate_file(simple_object_size.value) expected_copies = 1 endpoint = self.cluster.default_rpc_endpoint @@ -1163,14 +1163,14 @@ class TestPolicy(ClusterTestBase): len(resulting_copies) == expected_copies ), f"Expected {expected_copies} copies, got {len(resulting_copies)}" - with reporter.step(f"Check the object appearance"): - self.check_for_the_uniqueness_of_the_nodes(resulting_copies) + 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) node_address = resulting_copies[0].get_rpc_endpoint().split(":")[0] - assert netmap[node_address]["price"] >= int( - expected_params["price"] - ), f"The node is selected from the wrong price. Expected {expected_params} and got {netmap[node_address]}" + with allure.step(f"Check the node is selected with price >= {placement_params['price']}"): + assert ( + netmap[node_address]["price"] >= int(placement_params["price"]) + ), f"The node is selected with the wrong price. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -1188,7 +1188,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with SELECT and Complex FILTER results with 100% of available nodes. """ placement_rule = "REP 2 IN All SELECT 4 FROM AllNodes AS All FILTER Country EQ Russia OR Country EQ Sweden OR Country EQ Finland AS AllCountries FILTER @AllCountries AND Continent EQ Europe AS AllNodes" - expected_params = {"country": ["Russia", "Sweden", "Finland"], "continent": "Europe"} + placement_params = {"country": ["Russia", "Sweden", "Finland"], "continent": "Europe"} file_path = generate_file(simple_object_size.value) expected_copies = 2 endpoint = self.cluster.default_rpc_endpoint @@ -1213,13 +1213,14 @@ class TestPolicy(ClusterTestBase): with reporter.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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert ( - (netmap[node_address]["country"] in expected_params["country"]) - or (netmap[node_address]["country"] in expected_params["country"]) - and (netmap[node_address]["continent"] in expected_params["continent"]) - ), f"The node is selected from the wrong country or continent. Expected {expected_params} and got {netmap[node_address]}" + with allure.step(f"Check two nodes are selected from {' or '.join(placement_params['country'])}"): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert ( + (netmap[node_address]["country"] in placement_params["country"]) + or (netmap[node_address]["country"] in placement_params["country"]) + and (netmap[node_address]["continent"] == placement_params["continent"]) + ), f"The node is selected from the wrong country or continent. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -1237,7 +1238,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with Multi SELECTs and FILTERs results with 100% of available nodes. """ placement_rule = "REP 4 IN AllOne REP 4 IN AllTwo CBF 4 SELECT 2 FROM GEZero AS AllOne SELECT 2 FROM AllCountries AS AllTwo FILTER Country EQ Russia OR Country EQ Sweden OR Country EQ Finland AS AllCountries FILTER Price GE 0 AS GEZero" - expected_params = {"country": ["Russia", "Sweden", "Finland"], "price": "0"} + placement_params = {"country": ["Russia", "Sweden", "Finland"], "price": 0} file_path = generate_file(simple_object_size.value) expected_copies = 4 endpoint = self.cluster.default_rpc_endpoint @@ -1262,11 +1263,12 @@ class TestPolicy(ClusterTestBase): with reporter.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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert (netmap[node_address]["country"] in expected_params["country"]) or ( - netmap[node_address]["price"] in expected_params["price"] - ), f"The node is selected from the wrong country or price. Expected {expected_params} and got {netmap[node_address]}" + with allure.step(f"Check all node are selected"): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert (netmap[node_address]["country"] in placement_params["country"]) or ( + int(netmap[node_address]["price"]) >= placement_params["price"] + ), f"The node is selected from the wrong country or with wrong price. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -1305,10 +1307,6 @@ class TestPolicy(ClusterTestBase): assert ( len(resulting_copies) == expected_copies ), f"Expected {expected_copies} copies, got {len(resulting_copies)}" - self.check_for_the_uniqueness_of_the_nodes(resulting_copies) - - with reporter.step(f"Check the object appearance"): - self.check_for_the_uniqueness_of_the_nodes(resulting_copies) with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -1347,10 +1345,7 @@ class TestPolicy(ClusterTestBase): len(resulting_copies) == expected_copies ), f"Expected {expected_copies} copies, got {len(resulting_copies)}" - with reporter.step(f"Check the object appearance"): - self.check_for_the_uniqueness_of_the_nodes(resulting_copies) - - with reporter.step(f"Delete the object from the container"): + with allure.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) with reporter.step(f"Delete the container"): @@ -1367,8 +1362,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with SELECT and Complex FILTER results with UNIQUE nodes. """ placement_rule = "UNIQUE REP 1 IN RUS REP 1 IN RUS CBF 1 SELECT 1 FROM RU AS RUS FILTER Country NE Sweden AS NotSE FILTER @NotSE AND NOT (CountryCode EQ FI) AND Country EQ 'Russia' AS RU" - expected_params = {"country": "Russia"} - unexpected_params = {"country_code": "FI", "country": "Sweden"} + placement_params = {"country_code": "FI", "country": ["Sweden", "Russia"]} file_path = generate_file(simple_object_size.value) expected_copies = 2 endpoint = self.cluster.default_rpc_endpoint @@ -1390,17 +1384,17 @@ class TestPolicy(ClusterTestBase): len(resulting_copies) == expected_copies ), f"Expected {expected_copies} copies, got {len(resulting_copies)}" - with reporter.step(f"Check the object appearance"): - self.check_for_the_uniqueness_of_the_nodes(resulting_copies) + 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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert not (unexpected_params["country"] == netmap[node_address]["country"]) or ( - not (unexpected_params["country"] == netmap[node_address]["country"]) - and not (expected_params["country_code"] == netmap[node_address]["country_code"]) - or (expected_params["country"] == netmap[node_address]["country"]) - ), f"The node is selected from the wrong country or country code. Expected {expected_params} and got {netmap[node_address]}" + with allure.step(f"Check two nodes are selected not from {placement_params['country'][0]}"): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert not (placement_params["country"][0] == netmap[node_address]["country"]) or ( + not (placement_params["country"][0] == netmap[node_address]["country"]) + and not (placement_params["country_code"] == netmap[node_address]["country_code"]) + and placement_params["country"][1] == netmap[node_address]["country"] + ), f"The node is selected from the wrong country or with wrong country code. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -1419,8 +1413,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy with Multi SELECTs and FILTERs results with UNIQUE nodes. """ placement_rule = "UNIQUE REP 1 IN RU REP 1 IN EU REP 1 IN RU CBF 1 SELECT 1 FROM RUS AS RU SELECT 1 FROM EUR AS EU FILTER Country EQ Russia AS RUS FILTER NOT (@RUS) AND Country EQ Sweden OR CountryCode EQ FI AS EUR" - expected_params = {"country": ["Russia", "Sweden"], "country_code": "FI"} - unexpected_params = {"country": "Russia"} + placement_params = {"country": ["Russia", "Sweden"], "country_code": "FI"} file_path = generate_file(simple_object_size.value) expected_copies = 3 endpoint = self.cluster.default_rpc_endpoint @@ -1442,17 +1435,19 @@ class TestPolicy(ClusterTestBase): len(resulting_copies) == expected_copies ), f"Expected {expected_copies} copies, got {len(resulting_copies)}" - with reporter.step(f"Check the object appearance"): - self.check_for_the_uniqueness_of_the_nodes(resulting_copies) + 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) - for node in resulting_copies: - node_address = node.get_rpc_endpoint().split(":")[0] - assert (expected_params["country"][0] == netmap[node_address]["country"]) or ( - not (unexpected_params["country"][0] == netmap[node_address]["country"]) - and (expected_params["country"][1] == netmap[node_address]["country"]) - or (expected_params["country_code"] == netmap[node_address]["country_code"]) - ), f"The node is selected from the wrong country or country code. Expected {expected_params} and got {netmap[node_address]}" + with allure.step( + f"Check three nodes are selected from any country" + ): + for node in resulting_copies: + node_address = node.get_rpc_endpoint().split(":")[0] + assert (placement_params["country"][0] == netmap[node_address]["country"]) or ( + not (placement_params["country"][0] == netmap[node_address]["country"]) + and (placement_params["country"][1] == netmap[node_address]["country"]) + or (placement_params["country_code"] == netmap[node_address]["country_code"]) + ), f"The node is selected from the wrong country or with wrong country code. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -1470,7 +1465,7 @@ class TestPolicy(ClusterTestBase): This test checks object's copies based on container's placement policy: REP 1 IN SPB REP 1 IN MSK REP 3. """ placement_rule = "REP 1 IN SPB REP 1 IN MSK REP 3 CBF 1 SELECT 2 FROM LED AS SPB SELECT 2 FROM MOW AS MSK FILTER Location EQ 'Saint Petersburg (ex Leningrad)' AS LED FILTER Location EQ Moskva AS MOW" - expected_params = {"location": ["Saint Petersburg (ex Leningrad)", "Moskva"]} + placement_params = {"location": ["Saint Petersburg (ex Leningrad)", "Moskva"]} file_path = generate_file(simple_object_size.value) expected_copies = 3 endpoint = self.cluster.default_rpc_endpoint @@ -1500,13 +1495,12 @@ class TestPolicy(ClusterTestBase): node_address = node.get_rpc_endpoint().split(":")[0] 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) + with allure.step(f"Check two or three nodes are selected from Russia and from any other country"): + assert ( + placement_params["location"][0] in list_of_location + and placement_params["location"][1] in list_of_location + and len(resulting_copies) > 2 + ), f"The node is selected from the wrong location. Got {netmap[node_address]}" with reporter.step(f"Delete the object from the container"): delete_object(wallet=default_wallet, cid=cid, oid=oid, shell=self.shell, endpoint=endpoint) @@ -1522,9 +1516,6 @@ class TestPolicy(ClusterTestBase): "'", "" ), f"Expected \n{placement_rule} and got policy \n{got_policy} are the same" - def check_for_the_uniqueness_of_the_nodes(self, nodes: list): - assert len(set(nodes)) == len(nodes), f"The nodes found must be unique, but {nodes}" - def get_netmap_param(self, netmap_info: list[NodeNetmapInfo]) -> dict: dict_external = dict() for node in netmap_info: