X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;ds=sidebyside;f=tests%2Funit%2Fnetwork_services%2Fvnf_generic%2Fvnf%2Ftest_base.py;h=8a5d836e0739dd3c9a2caf497193c3ca222af8c4;hb=58b399dee7bdff055ca8cfb712d1e0c753d2f58c;hp=e4f4450cee93029d85346ae88024787f464a3f8e;hpb=4d301304687229c841c15e325474ad946d883ff0;p=yardstick.git

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