Vladimir Domnich
bfd02531ef
Replace service_helper with hosting class from the testlib. Instead of invoking commands on remote via ssh_helper, we now use shell from the hosting. Signed-off-by: Vladimir Domnich <v.domnich@yadro.com>
16 lines
610 B
Python
16 lines
610 B
Python
from ssh_helper import HostClient
|
|
|
|
|
|
# TODO: convert to shell from hosting
|
|
class IpTablesHelper:
|
|
@staticmethod
|
|
def drop_input_traffic_to_port(client: HostClient, ports: list[str]):
|
|
for port in ports:
|
|
cmd_output = client.exec(cmd=f"sudo iptables -A INPUT -p tcp --dport {port} -j DROP")
|
|
assert cmd_output.rc == 0
|
|
|
|
@staticmethod
|
|
def restore_input_traffic_to_port(client: HostClient, ports: list[str]):
|
|
for port in ports:
|
|
cmd_output = client.exec(cmd=f"sudo iptables -D INPUT -p tcp --dport {port} -j DROP")
|
|
assert cmd_output.rc == 0
|