diff --git a/scripts/k8s_scripts/test_fabric_forwarding.py b/scripts/k8s_scripts/test_fabric_forwarding.py index e0a76db37..b5c556b8c 100644 --- a/scripts/k8s_scripts/test_fabric_forwarding.py +++ b/scripts/k8s_scripts/test_fabric_forwarding.py @@ -8,12 +8,7 @@ class TestFabricFWD(BaseK8sTest): @classmethod def setUpClass(cls): super(TestFabricFWD, cls).setUpClass() - if cls.inputs.get_service_ip(cls.inputs.bgp_control_ips[0],service="CONTROL"): - cls.ip_to_ping = cls.inputs.get_service_ip(cls.inputs.bgp_control_ips[0], - service="CONTROL") - else: - cls.ip_to_ping = cls.inputs.get_service_ip(cls.inputs.bgp_control_ips[0], - service="CONTROLLER") + cls.ip_to_ping = cls.inputs.bgp_control_ips[0] @classmethod def tearDownClass(cls): diff --git a/scripts/k8s_scripts/test_snat.py b/scripts/k8s_scripts/test_snat.py index 95322ecc9..1905a8f75 100644 --- a/scripts/k8s_scripts/test_snat.py +++ b/scripts/k8s_scripts/test_snat.py @@ -8,12 +8,7 @@ class TestSNAT(BaseK8sTest): @classmethod def setUpClass(cls): super(TestSNAT, cls).setUpClass() - if cls.inputs.get_service_ip(cls.inputs.bgp_control_ips[0],service="CONTROL"): - cls.ip_to_ping = cls.inputs.get_service_ip(cls.inputs.bgp_control_ips[0], - service="CONTROL") - else: - cls.ip_to_ping = cls.inputs.get_service_ip(cls.inputs.bgp_control_ips[0], - service="CONTROLLER") + cls.ip_to_ping = cls.inputs.bgp_control_ips[0] @classmethod def tearDownClass(cls): diff --git a/serial_scripts/k8s_scripts/test_fabric_forwarding.py b/serial_scripts/k8s_scripts/test_fabric_forwarding.py index 2b3bcfef9..6052fdaf9 100644 --- a/serial_scripts/k8s_scripts/test_fabric_forwarding.py +++ b/serial_scripts/k8s_scripts/test_fabric_forwarding.py @@ -8,12 +8,7 @@ class TestFabricFWDRestarts(BaseK8sTest): @classmethod def setUpClass(cls): super(TestFabricFWDRestarts, cls).setUpClass() - if cls.inputs.get_service_ip(cls.inputs.bgp_control_ips[0],service="CONTROL"): - cls.ip_to_ping = cls.inputs.get_service_ip(cls.inputs.bgp_control_ips[0], - service="CONTROL") - else: - cls.ip_to_ping = cls.inputs.get_service_ip(cls.inputs.bgp_control_ips[0], - service="CONTROLLER") + cls.ip_to_ping = cls.inputs.bgp_control_ips[0] @classmethod def tearDownClass(cls): diff --git a/serial_scripts/k8s_scripts/test_fabric_snat_restart_scenarios.py b/serial_scripts/k8s_scripts/test_fabric_snat_restart_scenarios.py index 4b1cef8df..949717476 100644 --- a/serial_scripts/k8s_scripts/test_fabric_snat_restart_scenarios.py +++ b/serial_scripts/k8s_scripts/test_fabric_snat_restart_scenarios.py @@ -14,12 +14,7 @@ class TestFabricSNATRestarts(BaseK8sTest): @classmethod def setUpClass(cls): super(TestFabricSNATRestarts, cls).setUpClass() - if cls.inputs.get_service_ip(cls.inputs.bgp_control_ips[0],service="CONTROL"): - cls.ip_to_ping = cls.inputs.get_service_ip(cls.inputs.bgp_control_ips[0], - service="CONTROL") - else: - cls.ip_to_ping = cls.inputs.get_service_ip(cls.inputs.bgp_control_ips[0], - service="CONTROLLER") + cls.ip_to_ping = cls.inputs.bgp_control_ips[0] @classmethod def tearDownClass(cls):