Skip to content

Commit

Permalink
add docstring to SinkConnector class
Browse files Browse the repository at this point in the history
  • Loading branch information
Selfeer committed Apr 18, 2024
1 parent f532e84 commit 2a8c7e3
Showing 1 changed file with 10 additions and 2 deletions.
12 changes: 10 additions & 2 deletions sink-connector-lightweight/tests/integration/helpers/cluster.py
Original file line number Diff line number Diff line change
Expand Up @@ -954,25 +954,28 @@ def query(
class SinkConnector(DatabaseNode):
"""ClickHouse Sink Connector node."""

sink_connector_cli = "./sink-connector-client "
sink_connector_cli = "./sink-connector-client"

def __init__(self, cluster, name):
"""Install procps package on the node to have all the basic system tools."""
super().__init__(cluster, name)
self.command(command="microdnf install procps")

@staticmethod
def parse_value(input_string):
"""Parse the output value to get the PID of the ClickHouse Sink Connector."""
match = re.search(r"\[\d+\]\s*(\d+)", str(input_string))
if match:
return int(match.group(1))
else:
raise ValueError(f"Failed to parse value from string: {input_string}")

def start_sink_connector(self, timeout=300, config_file=None):
"""Start ClickHouse Sink Connector."""
if config_file is None:
config_file = "config.yml"

config_file = config_file.rsplit('/', 1)[-1]
config_file = config_file.rsplit("/", 1)[-1]

with Given("I start ClickHouse Sink Connector"):
start_command = self.command(
Expand Down Expand Up @@ -1025,34 +1028,39 @@ def restart_sink_connector(self, timeout=300, config_file=None):
self.start_sink_connector(timeout=timeout, config_file=config_file)

def stop_replication(self, timeout=300):
"""Stop ClickHouse Sink Connector replication using sink-connector-client script."""
with Given("I stop ClickHouse Sink Connector replication"):
self.command(
command=f"{self.sink_connector_cli} stop_replica",
timeout=timeout,
)

def start_replication(self):
"""Start ClickHouse Sink Connector replication using sink-connector-client script."""
with Given("I start ClickHouse Sink Connector replication"):
self.command(
command=f"{self.sink_connector_cli} start_replica",
timeout=300,
)

def change_replica_source(self):
"""Change ClickHouse Sink Connector replica source using sink-connector-client script."""
with Given("I change ClickHouse Sink Connector replica source"):
self.command(
command=f"{self.sink_connector_cli} change_replica_source",
timeout=300,
)

def show_replication_status(self):
"""Show ClickHouse Sink Connector replication status using sink-connector-client script."""
with Given("I change ClickHouse Sink Connector replica source"):
self.command(
command=f"{self.sink_connector_cli} show_replica_status",
timeout=300,
)

def get_container_id(self):
"""Get the container ID of the ClickHouse Sink Connector node."""
with Given("I get the container ID of the ClickHouse Sink Connector"):
container_id = self.cluster.node_container_id(self.name)
return container_id
Expand Down

0 comments on commit 2a8c7e3

Please sign in to comment.