- 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.from_node.return_value = ssh_mock
+ mock_ssh_instance = mock.Mock(autospec=spec)
+ mock_ssh_instance._get_client.return_value = mock.Mock()
+ mock_ssh_instance.execute.return_value = exec_result
+ mock_ssh_instance.run.return_value = run_result
+ mock_ssh_type.from_node.return_value = mock_ssh_instance
+ return mock_ssh_instance