From a4d2958ddb727c2aaf46956cbdbf7718bfd0be89 Mon Sep 17 00:00:00 2001 From: Ross Brattain Date: Mon, 24 Apr 2017 23:25:48 +0200 Subject: PROX VNF and TG PROX was added to samplevnf project https://git.opnfv.org/samplevnf/tree/VNFs/DPPD-PROX JIRA: YARDSTICK-638 Change-Id: If9875b1130c6bed87deb8720b0d8b28ede9289d9 Signed-off-by: Ross Brattain --- tests/unit/network_services/vnf_generic/vnf/test_base.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'tests/unit/network_services/vnf_generic/vnf/test_base.py') diff --git a/tests/unit/network_services/vnf_generic/vnf/test_base.py b/tests/unit/network_services/vnf_generic/vnf/test_base.py index e4f4450ce..8a5d836e0 100644 --- a/tests/unit/network_services/vnf_generic/vnf/test_base.py +++ b/tests/unit/network_services/vnf_generic/vnf/test_base.py @@ -59,8 +59,9 @@ def mock_ssh(ssh, spec=None, exec_result=_LOCAL_OBJECT, run_result=_LOCAL_OBJECT run_result = 0, "", "" ssh_mock = mock.Mock(autospec=spec) - ssh_mock.execute = mock.Mock(return_value=exec_result) - ssh_mock.run = mock.Mock(return_value=run_result) + ssh_mock._get_client.return_value = mock.Mock() + ssh_mock.execute.return_value = exec_result + ssh_mock.run.return_value = run_result ssh.from_node.return_value = ssh_mock -- cgit 1.2.3-korg