diff --git a/nova/tests/unit/virt/libvirt/volume/test_remotefs.py b/nova/tests/unit/virt/libvirt/volume/test_remotefs.py index f36c79fce7c1..3be6a1764b76 100644 --- a/nova/tests/unit/virt/libvirt/volume/test_remotefs.py +++ b/nova/tests/unit/virt/libvirt/volume/test_remotefs.py @@ -74,11 +74,11 @@ class RemoteFSTestCase(test.NoDBTestCase): self.assertEqual(2, mock_execute.call_count) self.assertEqual(1, mock_mkdtemp.call_count) - @mock.patch('nova.utils.execute') - def test_remove_remote_file_ssh(self, mock_execute): + @mock.patch('nova.utils.ssh_execute') + def test_remove_remote_file_ssh(self, mock_ssh_execute): remotefs.SshDriver().remove_file('host', 'dest', None, None) - mock_execute.assert_called_once_with( - 'ssh', 'host', 'rm', 'dest', + mock_ssh_execute.assert_called_once_with( + 'host', 'rm', 'dest', on_completion=None, on_execute=None) @mock.patch('tempfile.mkdtemp', return_value='/tmp/Venus') @@ -101,11 +101,11 @@ class RemoteFSTestCase(test.NoDBTestCase): self.assertEqual(3, mock_execute.call_count) self.assertEqual(1, mock_mkdtemp.call_count) - @mock.patch('nova.utils.execute') - def test_remove_remote_dir_ssh(self, mock_execute): + @mock.patch('nova.utils.ssh_execute') + def test_remove_remote_dir_ssh(self, mock_ssh_execute): remotefs.SshDriver().remove_dir('host', 'dest', None, None) - mock_execute.assert_called_once_with( - 'ssh', 'host', 'rm', '-rf', 'dest', on_completion=None, + mock_ssh_execute.assert_called_once_with( + 'host', 'rm', '-rf', 'dest', on_completion=None, on_execute=None) @mock.patch('tempfile.mkdtemp', return_value='/tmp/Mars') @@ -128,13 +128,13 @@ class RemoteFSTestCase(test.NoDBTestCase): self.assertEqual(4, mock_execute.call_count) self.assertEqual(1, mock_mkdtemp.call_count) - @mock.patch('nova.utils.execute') - def test_create_remote_file_ssh(self, mock_execute): + @mock.patch('nova.utils.ssh_execute') + def test_create_remote_file_ssh(self, mock_ssh_execute): remotefs.SshDriver().create_file('host', 'dest_dir', None, None) - mock_execute.assert_called_once_with('ssh', 'host', - 'touch', 'dest_dir', - on_completion=None, - on_execute=None) + mock_ssh_execute.assert_called_once_with('host', 'touch', + 'dest_dir', + on_completion=None, + on_execute=None) @mock.patch('tempfile.mkdtemp', return_value='/tmp/Jupiter') @mock.patch('nova.utils.execute') @@ -153,13 +153,13 @@ class RemoteFSTestCase(test.NoDBTestCase): self.assertEqual(3, mock_execute.call_count) self.assertEqual(1, mock_mkdtemp.call_count) - @mock.patch('nova.utils.execute') - def test_create_remote_dir_ssh(self, mock_execute): + @mock.patch('nova.utils.ssh_execute') + def test_create_remote_dir_ssh(self, mock_ssh_execute): remotefs.SshDriver().create_dir('host', 'dest_dir', None, None) - mock_execute.assert_called_once_with('ssh', 'host', 'mkdir', - '-p', 'dest_dir', - on_completion=None, - on_execute=None) + mock_ssh_execute.assert_called_once_with('host', 'mkdir', + '-p', 'dest_dir', + on_completion=None, + on_execute=None) @mock.patch('nova.utils.execute') def test_remote_copy_file_rsync(self, mock_execute): diff --git a/nova/virt/libvirt/volume/remotefs.py b/nova/virt/libvirt/volume/remotefs.py index fe9b3d903e49..cb9060c148cd 100644 --- a/nova/virt/libvirt/volume/remotefs.py +++ b/nova/virt/libvirt/volume/remotefs.py @@ -189,20 +189,20 @@ class RemoteFilesystemDriver(object): class SshDriver(RemoteFilesystemDriver): def create_file(self, host, dst_path, on_execute, on_completion): - utils.execute('ssh', host, 'touch', dst_path, - on_execute=on_execute, on_completion=on_completion) + utils.ssh_execute(host, 'touch', dst_path, + on_execute=on_execute, on_completion=on_completion) def remove_file(self, host, dst, on_execute, on_completion): - utils.execute('ssh', host, 'rm', dst, - on_execute=on_execute, on_completion=on_completion) + utils.ssh_execute(host, 'rm', dst, + on_execute=on_execute, on_completion=on_completion) def create_dir(self, host, dst_path, on_execute, on_completion): - utils.execute('ssh', host, 'mkdir', '-p', dst_path, - on_execute=on_execute, on_completion=on_completion) + utils.ssh_execute(host, 'mkdir', '-p', dst_path, + on_execute=on_execute, on_completion=on_completion) def remove_dir(self, host, dst, on_execute, on_completion): - utils.execute('ssh', host, 'rm', '-rf', dst, - on_execute=on_execute, on_completion=on_completion) + utils.ssh_execute(host, 'rm', '-rf', dst, + on_execute=on_execute, on_completion=on_completion) def copy_file(self, src, dst, on_execute, on_completion, compression): utils.execute('scp', src, dst,