def test_run_udp_replay(self, mock_open, eval, hex, mock_process):
with mock.patch("yardstick.ssh.SSH") as ssh:
ssh_mock = mock.Mock(autospec=ssh.SSH)
ssh_mock.execute = \
mock.Mock(return_value=(0, "", ""))
ssh_mock.run = \
mock.Mock(return_value=(0, "", ""))
ssh.from_node.return_value = ssh_mock
vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
udp_approx_vnf = UdpReplayApproxVnf(NAME, vnfd)
udp_approx_vnf._build_config = mock.MagicMock()
udp_approx_vnf.queue_wrapper = mock.MagicMock()
udp_approx_vnf.ssh_helper = mock.MagicMock()
udp_approx_vnf.ssh_helper.run = mock.MagicMock()
udp_approx_vnf.vnf_cfg = {
'lb_config': 'SW',
'lb_count': 1,
'worker_config': '1C/1T',
'worker_threads': 1
}
udp_approx_vnf.options = {
'traffic_type': '4',
'topology': 'nsb_test_case.yaml'
}
udp_approx_vnf._run()
udp_approx_vnf.ssh_helper.run.assert_called_once()
def test__build_pipeline_kwargs(self, mock_open, mock_process):
with mock.patch("yardstick.ssh.SSH") as ssh:
ssh_mock = mock.Mock(autospec=ssh.SSH)
ssh.from_node.return_value = ssh_mock
vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
udp_approx_vnf = UdpReplayApproxVnf(NAME, vnfd)
udp_approx_vnf._build_config = mock.MagicMock()
udp_approx_vnf.queue_wrapper = mock.MagicMock()
udp_approx_vnf.nfvi_type = "baremetal"
udp_approx_vnf.bound_pci = []
udp_approx_vnf.all_ports = [0, 1]
udp_approx_vnf.ssh_helper = mock.MagicMock(
**{"provision_tool.return_value": "tool_path"})
udp_approx_vnf.vnf_cfg = {
'lb_config': 'SW',
'lb_count': 1,
'worker_config': '1C/1T',
'worker_threads': 1
}
udp_approx_vnf.options = {
'traffic_type': '4',
'topology': 'nsb_test_case.yaml'
}
udp_approx_vnf._build_pipeline_kwargs()
self.assertEqual(
udp_approx_vnf.pipeline_kwargs, {
'config': '(0, 0, 1)(1, 0, 2)',
'cpu_mask_hex': '0x6',
'hw_csum': '',
'ports_len_hex': '0x3',
'tool_path': 'tool_path',
'whitelist': ''
})
def test_run_udp_replay(self, ssh, _):
mock_ssh(ssh)
udp_replay_approx_vnf = UdpReplayApproxVnf(NAME, self.VNFD_0)
udp_replay_approx_vnf._build_config = mock.MagicMock()
udp_replay_approx_vnf.queue_wrapper = mock.MagicMock()
udp_replay_approx_vnf._run()
udp_replay_approx_vnf.ssh_helper.run.assert_called_once()
def test__build_config(self, ssh, mock_context, *args):
mock_ssh(ssh)
udp_replay_approx_vnf = UdpReplayApproxVnf(NAME, self.VNFD_0)
udp_replay_approx_vnf.queue_wrapper = mock.MagicMock()
udp_replay_approx_vnf.nfvi_context = mock_context
udp_replay_approx_vnf.nfvi_context.attrs = {'nfvi_type': 'baremetal'}
udp_replay_approx_vnf.setup_helper.bound_pci = []
udp_replay_approx_vnf.ssh_helper.provision_tool = mock.MagicMock(
return_value="tool_path")
udp_replay_approx_vnf.scenario_helper = ScenarioHelper(name='vnf__1')
udp_replay_approx_vnf.scenario_helper.scenario_cfg = self.SCENARIO_CFG
cmd_line = udp_replay_approx_vnf._build_config()
expected = \
"sudo tool_path --log-level=5 -c 0x7 -n 4 -w -- -p 0x3 --config='(0,0,1),(1,0,2)'"
self.assertEqual(cmd_line, expected)