From 0c3bb20af5c353887cec98f6c0dc203f2b3ed26c Mon Sep 17 00:00:00 2001
From: Dmitriy Zayakin <d.zayakin@yadro.com>
Date: Tue, 24 Oct 2023 13:57:11 +0300
Subject: [PATCH] Add method to interfaces

Signed-off-by: Dmitriy Zayakin <d.zayakin@yadro.com>
---
 src/frostfs_testlib/storage/cluster.py        | 35 +++++++++++++++++++
 .../storage/dataclasses/frostfs_services.py   |  8 +++--
 .../dataclasses/storage_object_info.py        | 11 ++++++
 3 files changed, 52 insertions(+), 2 deletions(-)

diff --git a/src/frostfs_testlib/storage/cluster.py b/src/frostfs_testlib/storage/cluster.py
index 0e24ebb..7a48a1d 100644
--- a/src/frostfs_testlib/storage/cluster.py
+++ b/src/frostfs_testlib/storage/cluster.py
@@ -17,6 +17,7 @@ from frostfs_testlib.storage.dataclasses.frostfs_services import (
     StorageNode,
 )
 from frostfs_testlib.storage.dataclasses.node_base import NodeBase, ServiceClass
+from frostfs_testlib.storage.dataclasses.storage_object_info import Interfaces
 from frostfs_testlib.storage.service_registry import ServiceRegistry
 
 reporter = get_reporter()
@@ -121,6 +122,40 @@ class ClusterNode:
             config.attributes[ConfigAttributes.SERVICE_NAME] for config in self.host.config.services
         ]
 
+    def get_all_interfaces(self) -> dict[str, str]:
+        return self.host.config.interfaces
+
+    def get_interface(self, interface: Interfaces) -> str:
+        return self.host.config.interfaces[interface.value]
+
+    def get_data_interfaces(self) -> list[str]:
+        return [
+            ip_address
+            for name_interface, ip_address in self.host.config.interfaces.items()
+            if "data" in name_interface
+        ]
+
+    def get_data_interface(self, search_interface: str) -> list[str]:
+        return [
+            self.host.config.interfaces[interface]
+            for interface in self.host.config.interfaces.keys()
+            if search_interface == interface
+        ]
+
+    def get_internal_interfaces(self) -> list[str]:
+        return [
+            ip_address
+            for name_interface, ip_address in self.host.config.interfaces.items()
+            if "internal" in name_interface
+        ]
+
+    def get_internal_interface(self, search_internal: str) -> list[str]:
+        return [
+            self.host.config.interfaces[interface]
+            for interface in self.host.config.interfaces.keys()
+            if search_internal == interface
+        ]
+
 
 class Cluster:
     """
diff --git a/src/frostfs_testlib/storage/dataclasses/frostfs_services.py b/src/frostfs_testlib/storage/dataclasses/frostfs_services.py
index ac2885b..9e6783c 100644
--- a/src/frostfs_testlib/storage/dataclasses/frostfs_services.py
+++ b/src/frostfs_testlib/storage/dataclasses/frostfs_services.py
@@ -110,6 +110,10 @@ class MorphChain(NodeBase):
     def label(self) -> str:
         return f"{self.name}: {self.get_endpoint()}"
 
+    def get_http_endpoint(self) -> str:
+        return self._get_attribute("http_endpoint")
+
+
 class StorageNode(NodeBase):
     """
     Class represents storage node in a storage cluster
@@ -149,10 +153,10 @@ class StorageNode(NodeBase):
 
     def get_data_directory(self) -> str:
         return self.host.get_data_directory(self.name)
-    
+
     def get_http_hostname(self) -> str:
         return self._get_attribute(ConfigAttributes.HTTP_HOSTNAME)
-    
+
     def get_s3_hostname(self) -> str:
         return self._get_attribute(ConfigAttributes.S3_HOSTNAME)
 
diff --git a/src/frostfs_testlib/storage/dataclasses/storage_object_info.py b/src/frostfs_testlib/storage/dataclasses/storage_object_info.py
index f7d51db..d670d8e 100644
--- a/src/frostfs_testlib/storage/dataclasses/storage_object_info.py
+++ b/src/frostfs_testlib/storage/dataclasses/storage_object_info.py
@@ -1,6 +1,9 @@
 from dataclasses import dataclass
+from enum import Enum
 from typing import Optional
 
+from frostfs_testlib.testing.readable import HumanReadableEnum
+
 
 @dataclass
 class ObjectRef:
@@ -42,3 +45,11 @@ class NodeNetmapInfo:
     sub_div_code: int = None
     un_locode: str = None
     role: str = None
+
+
+class Interfaces(HumanReadableEnum):
+    DATA_O: str = "data0"
+    DATA_1: str = "data1"
+    MGMT: str = "mgmt"
+    INTERNAL_0: str = "internal0"
+    INTERNAL_1: str = "internal1"