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_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()