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)