def test_vsperf_dpdk_setup(self, mock_ssh, mock_subprocess):
p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
# setup() specific mocks
mock_subprocess.call().execute.return_value = None
p.setup()
self.assertIsNotNone(p.client)
self.assertEqual(p.setup_done, True)
def test_vsperf_dpdk_dpdk_setup_fail(self, mock_time, mock_ssh, mock_subprocess):
p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
# setup() specific mocks
mock_ssh.SSH.from_node().execute.return_value = (0, '', '')
mock_subprocess.call().execute.return_value = None
p.setup()
self.assertIsNotNone(p.client)
mock_ssh.SSH.from_node().execute.return_value = (1, '', '')
self.assertEqual(p.setup_done, True)
self.assertRaises(RuntimeError, p.dpdk_setup)
def test_vsperf_dpdk_is_dpdk_setup_no(self, mock_ssh, mock_subprocess):
p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
# setup() specific mocks
mock_subprocess.call().execute.return_value = None
p.setup()
self.assertIsNotNone(p.client)
self.assertEqual(p.setup_done, True)
# is_dpdk_setup() specific mocks
mock_ssh.SSH.from_node().execute.return_value = (0, 'dummy', '')
result = p._is_dpdk_setup()
self.assertEqual(result, False)
def setUp(self):
self.ctx = {
"host": {
"ip": "10.229.47.137",
"user": "******",
"password": "******",
},
}
self.args = {
'task_id': "1234-5678",
'options': {
'testname': 'pvp_tput',
'traffic_type': 'rfc2544_throughput',
'frame_size': '64',
'test_params': 'TRAFFICGEN_DURATION=30;',
'trafficgen_port1': 'ens4',
'trafficgen_port2': 'ens5',
'conf_file': 'vsperf-yardstick.conf',
'setup_script': 'setup_yardstick.sh',
'moongen_helper_file': '~/moongen.py',
'moongen_host_ip': '10.5.201.151',
'moongen_port1_mac': '8c:dc:d4:ae:7c:5c',
'moongen_port2_mac': '8c:dc:d4:ae:7c:5d',
'trafficgen_port1_nw': 'test2',
'trafficgen_port2_nw': 'test3',
},
'sla': {
'metrics': 'throughput_rx_fps',
'throughput_rx_fps': 500000,
'action': 'monitor',
}
}
self._mock_ssh = mock.patch.object(ssh, 'SSH')
self.mock_ssh = self._mock_ssh.start()
self._mock_subprocess_call = mock.patch.object(subprocess, 'call')
self.mock_subprocess_call = self._mock_subprocess_call.start()
mock_call_obj = mock.Mock()
mock_call_obj.execute.return_value = None
self.mock_subprocess_call.return_value = mock_call_obj
self._mock_log_info = mock.patch.object(vsperf_dpdk.LOG, 'info')
self.mock_log_info = self._mock_log_info.start()
self.addCleanup(self._cleanup)
self.scenario = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
self.scenario.setup()
def test_vsperf_dpdk_run_falied_vsperf_execution(self, mock_ssh,
mock_subprocess):
p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
# setup() specific mocks
mock_subprocess.call().execute.return_value = None
p.setup()
self.assertIsNotNone(p.client)
self.assertEqual(p.setup_done, True)
# run() specific mocks
mock_subprocess.call().execute.return_value = None
mock_subprocess.call().execute.return_value = None
mock_ssh.SSH.from_node().execute.return_value = (1, '', '')
result = {}
self.assertRaises(RuntimeError, p.run, result)
def test_vsperf_dpdk_run_ok(self, mock_ssh, mock_subprocess):
p = vsperf_dpdk.VsperfDPDK(self.args, self.ctx)
# setup() specific mocks
mock_subprocess.call().execute.return_value = None
p.setup()
self.assertIsNotNone(p.client)
self.assertEqual(p.setup_done, True)
# run() specific mocks
mock_subprocess.call().execute.return_value = None
mock_subprocess.call().execute.return_value = None
mock_ssh.SSH.from_node().execute.return_value = (
0, 'throughput_rx_fps\r\n14797660.000\r\n', '')
result = {}
p.run(result)
self.assertEqual(result['throughput_rx_fps'], '14797660.000')
