test: add container name for DNS

This commit is contained in:
Roman 2025-01-07 13:13:11 +08:00
parent b2502740e7
commit 97266edf61
No known key found for this signature in database
GPG Key ID: B8FE070B54E11B75
3 changed files with 12 additions and 6 deletions

View File

@ -35,7 +35,7 @@ class DockerManager:
logger.debug(f"Network {network_name} created")
return network
def start_container(self, image_name, port_bindings, args, log_path, container_ip, volumes, entrypoint, remove_container=True):
def start_container(self, image_name, port_bindings, args, log_path, container_ip, volumes, entrypoint, remove_container=True, name=None):
cli_args = []
for key, value in args.items():
if isinstance(value, list): # Check if value is a list
@ -58,6 +58,7 @@ class DockerManager:
auto_remove=remove_container,
volumes=volumes,
entrypoint=entrypoint,
name=name,
)
network = self._client.networks.get(NETWORK_NAME)

View File

@ -21,15 +21,16 @@ def sanitize_docker_flags(input_flags):
class NomosNode:
def __init__(self, node_type, docker_log_prefix=""):
def __init__(self, node_type, container_name=""):
logger.debug(f"Node is going to be initialized with this config {nomos_nodes[node_type]}")
self._image_name = nomos_nodes[node_type]["image"]
self._internal_ports = nomos_nodes[node_type]["ports"]
self._volumes = nomos_nodes[node_type]["volumes"]
self._entrypoint = nomos_nodes[node_type]["entrypoint"]
self._log_path = os.path.join(DOCKER_LOG_DIR, f"{docker_log_prefix}__{self._image_name.replace('/', '_')}.log")
self._log_path = os.path.join(DOCKER_LOG_DIR, f"{container_name}__{self._image_name.replace('/', '_')}.log")
self._docker_manager = DockerManager(self._image_name)
self._container_name = container_name
self._container = None
cwd = os.getcwd()
@ -77,6 +78,7 @@ class NomosNode:
volumes=self._volumes,
entrypoint=self._entrypoint,
remove_container=True,
name=self._container_name,
)
logger.debug(f"Container returned {self._container}")
logger.debug(f"Started container from image {self._image_name}")

View File

@ -1,14 +1,17 @@
from src.env_vars import CFGSYNC, NOMOS, NOMOS_EXECUTOR
from src.node.nomos_node import NomosNode
from src.libs.common import delay
class Test2NodeClAlive:
def test_cluster_start(self):
self.node1 = NomosNode(CFGSYNC, f"node1_{1}")
self.node2 = NomosNode(NOMOS, f"node2_{2}")
self.node3 = NomosNode(NOMOS_EXECUTOR, f"node3_{3}")
self.node1 = NomosNode(CFGSYNC, "cfgsync")
self.node2 = NomosNode(NOMOS, "nomos_node_0")
self.node3 = NomosNode(NOMOS_EXECUTOR, "nomos_node_0")
self.node1.start()
self.node2.start()
self.node3.start()
delay(60)