From 12b592713b2d797e96da1a5d85f5e7790484b02f Mon Sep 17 00:00:00 2001
From: Vladimir Avdeev <v.avdeev@yadro.com>
Date: Tue, 6 Dec 2022 19:23:21 +0300
Subject: [PATCH] Add control shards test

Signed-off-by: Vladimir Avdeev <v.avdeev@yadro.com>
---
 pytest_tests/testsuites/conftest.py           |   2 +-
 .../testsuites/shard/test_control_shard.py    | 103 ++++++++++++++++++
 requirements.txt                              |   2 +-
 3 files changed, 105 insertions(+), 2 deletions(-)
 create mode 100644 pytest_tests/testsuites/shard/test_control_shard.py

diff --git a/pytest_tests/testsuites/conftest.py b/pytest_tests/testsuites/conftest.py
index 869bd719..b5b4a26b 100644
--- a/pytest_tests/testsuites/conftest.py
+++ b/pytest_tests/testsuites/conftest.py
@@ -162,7 +162,7 @@ def run_health_check(collect_logs, cluster: Cluster):
 
 
 @pytest.fixture(scope="session")
-def background_grpc_load(client_shell, default_wallet):
+def background_grpc_load(client_shell):
     registry_file = os.path.join("/tmp/", f"{str(uuid.uuid4())}.bolt")
     prepare_file = os.path.join("/tmp/", f"{str(uuid.uuid4())}.json")
     allure.dynamic.title(
diff --git a/pytest_tests/testsuites/shard/test_control_shard.py b/pytest_tests/testsuites/shard/test_control_shard.py
new file mode 100644
index 00000000..62996d9c
--- /dev/null
+++ b/pytest_tests/testsuites/shard/test_control_shard.py
@@ -0,0 +1,103 @@
+import json
+from dataclasses import dataclass
+
+import allure
+import pytest
+import yaml
+from cluster import Cluster
+from common import NEOFS_CLI_EXEC, WALLET_CONFIG
+from neofs_testlib.cli import NeofsCli
+from neofs_testlib.hosting import Host, Hosting, ServiceConfig
+from neofs_testlib.shell import Shell
+
+
+@dataclass
+class Blobstor:
+    path: str
+    path_type: str
+
+    def __eq__(self, other) -> bool:
+        if not isinstance(other, self.__class__):
+            raise RuntimeError(f"Only two {self.__class__.__name__} instances can be compared")
+        return self.path == other.path and self.path_type == other.path_type
+
+    def __hash__(self):
+        return hash((self.path, self.path_type))
+
+
+@dataclass
+class Shard:
+    blobstor: list[Blobstor]
+    metabase: str
+    writecache: str
+
+    def __eq__(self, other) -> bool:
+        if not isinstance(other, self.__class__):
+            raise RuntimeError(f"Only two {self.__class__.__name__} instances can be compared")
+        return (
+            set(self.blobstor) == set(other.blobstor)
+            and self.metabase == other.metabase
+            and self.writecache == other.writecache
+        )
+
+    def __hash__(self):
+        return hash((self.metabase, self.writecache))
+
+
+@pytest.mark.sanity
+@pytest.mark.shard
+class TestControlShard:
+    @staticmethod
+    def get_shards_from_config(host: Host, service_config: ServiceConfig) -> list[Shard]:
+        config_file = service_config.attributes["config_path"]
+        config = yaml.safe_load(host.get_shell().exec(f"cat {config_file}").stdout)
+        config["storage"]["shard"].pop("default")
+        return [
+            Shard(
+                blobstor=[
+                    Blobstor(path=blobstor["path"], path_type=blobstor["type"])
+                    for blobstor in shard["blobstor"]
+                ],
+                metabase=shard["metabase"]["path"],
+                writecache=shard["writecache"]["path"],
+            )
+            for shard in config["storage"]["shard"].values()
+        ]
+
+    @staticmethod
+    def get_shards_from_cli(host: Host, service_config: ServiceConfig) -> list[Shard]:
+        wallet_path = service_config.attributes["wallet_path"]
+        wallet_password = service_config.attributes["wallet_password"]
+        control_endpoint = service_config.attributes["control_endpoint"]
+
+        cli = NeofsCli(host.get_shell(), NEOFS_CLI_EXEC, WALLET_CONFIG)
+        result = cli.shards.list(
+            endpoint=control_endpoint,
+            wallet=wallet_path,
+            wallet_password=wallet_password,
+            json_mode=True,
+        )
+        return [
+            Shard(
+                blobstor=[
+                    Blobstor(path=blobstor["path"], path_type=blobstor["type"])
+                    for blobstor in shard["blobstor"]
+                ],
+                metabase=shard["metabase"],
+                writecache=shard["writecache"],
+            )
+            for shard in json.loads(result.stdout.split(">", 1)[1])
+        ]
+
+    @allure.title("All shards are available")
+    def test_control_shard(self, hosting: Hosting, client_shell: Shell, cluster: Cluster):
+        for stroage_host in cluster.storage_nodes:
+            shards_from_config = self.get_shards_from_config(
+                hosting.get_host_by_service(stroage_host.name),
+                hosting.get_service_config(stroage_host.name),
+            )
+            shards_from_cli = self.get_shards_from_cli(
+                hosting.get_host_by_service(stroage_host.name),
+                hosting.get_service_config(stroage_host.name),
+            )
+            assert set(shards_from_config) == set(shards_from_cli)
diff --git a/requirements.txt b/requirements.txt
index bd291498..7aeb0d60 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -35,7 +35,7 @@ neo-mamba==0.10.0
 neo3crypto==0.2.1
 neo3vm==0.9.0
 neo3vm-stubs==0.9.0
-neofs-testlib==0.7.0
+neofs-testlib==0.8.0
 netaddr==0.8.0
 orjson==3.6.8
 packaging==21.3