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': '' })