diff options
| -rw-r--r-- | tests/unit/benchmark/scenarios/networking/test_pktgen.py | 2 | ||||
| -rw-r--r-- | yardstick/benchmark/scenarios/networking/pktgen.py | 5 |
2 files changed, 4 insertions, 3 deletions
diff --git a/tests/unit/benchmark/scenarios/networking/test_pktgen.py b/tests/unit/benchmark/scenarios/networking/test_pktgen.py index 2914c8e02..32ba255b2 100644 --- a/tests/unit/benchmark/scenarios/networking/test_pktgen.py +++ b/tests/unit/benchmark/scenarios/networking/test_pktgen.py @@ -68,7 +68,7 @@ class PktgenTestCase(unittest.TestCase): mock_ssh.SSH.from_node().execute.assert_called_with( "sudo iptables -F; " "sudo iptables -A INPUT -p udp --dport 1000:%s -j DROP" - % 1010) + % 1010, timeout=60) def test_pktgen_unsuccessful_iptables_setup(self, mock_ssh): diff --git a/yardstick/benchmark/scenarios/networking/pktgen.py b/yardstick/benchmark/scenarios/networking/pktgen.py index 8ca1ca60e..1e0a5fcbb 100644 --- a/yardstick/benchmark/scenarios/networking/pktgen.py +++ b/yardstick/benchmark/scenarios/networking/pktgen.py @@ -20,6 +20,7 @@ from yardstick.benchmark.scenarios import base LOG = logging.getLogger(__name__) +SSH_TIMEOUT = 60 VNIC_TYPE_LIST = ["ovs", "sriov"] SRIOV_DRIVER_LIST = ["ixgbevf", "i40evf"] @@ -287,7 +288,7 @@ class Pktgen(base.Scenario): "sudo iptables -A INPUT -p udp --dport 1000:%s -j DROP" \ % (1000 + self.number_of_ports) LOG.debug("Executing command: %s", cmd) - status, _, stderr = self.server.execute(cmd) + status, _, stderr = self.server.execute(cmd, timeout=SSH_TIMEOUT) if status: raise RuntimeError(stderr) @@ -349,7 +350,7 @@ class Pktgen(base.Scenario): duration, queue_number, pps) LOG.debug("Executing command: %s", cmd) - status, stdout, stderr = self.client.execute(cmd) + status, stdout, stderr = self.client.execute(cmd, timeout=SSH_TIMEOUT) if status: raise RuntimeError(stderr) |