Fix Ixload KPIs collection & hang issues.

[yardstick.git] / 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 e4f4450..8a5d836 100644 (file)
--- 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