diff options
Diffstat (limited to 'troveclient/tests/osc/v1/test_database_backups.py')
-rw-r--r-- | troveclient/tests/osc/v1/test_database_backups.py | 30 |
1 files changed, 22 insertions, 8 deletions
diff --git a/troveclient/tests/osc/v1/test_database_backups.py b/troveclient/tests/osc/v1/test_database_backups.py index a552d25..a23622f 100644 --- a/troveclient/tests/osc/v1/test_database_backups.py +++ b/troveclient/tests/osc/v1/test_database_backups.py @@ -14,6 +14,7 @@ from unittest import mock from osc_lib import exceptions from osc_lib import utils +from oslo_utils import uuidutils from troveclient import common from troveclient.osc.v1 import database_backups @@ -181,25 +182,38 @@ class TestDatabaseBackupDelete(TestBackups): super(TestDatabaseBackupDelete, self).setUp() self.cmd = database_backups.DeleteDatabaseBackup(self.app, None) - @mock.patch.object(utils, 'find_resource') - def test_backup_delete(self, mock_find): + @mock.patch("troveclient.utils.get_resource_id_by_name") + def test_backup_delete(self, mock_getid): args = ['backup1'] - mock_find.return_value = args[0] + mock_getid.return_value = args[0] parsed_args = self.check_parser(self.cmd, args, []) - result = self.cmd.take_action(parsed_args) + self.cmd.take_action(parsed_args) self.backup_client.delete.assert_called_with('backup1') - self.assertIsNone(result) - @mock.patch.object(utils, 'find_resource') - def test_backup_delete_with_exception(self, mock_find): + @mock.patch("troveclient.utils.get_resource_id_by_name") + def test_backup_delete_with_exception(self, mock_getid): args = ['fakebackup'] parsed_args = self.check_parser(self.cmd, args, []) - mock_find.side_effect = exceptions.CommandError + mock_getid.side_effect = exceptions.CommandError self.assertRaises(exceptions.CommandError, self.cmd.take_action, parsed_args) + @mock.patch("troveclient.utils.get_resource_id_by_name") + def test_backup_bulk_delete(self, mock_getid): + backup_1 = uuidutils.generate_uuid() + backup_2 = uuidutils.generate_uuid() + mock_getid.return_value = backup_1 + + args = ["fake_backup", backup_2] + parsed_args = self.check_parser(self.cmd, args, []) + self.cmd.take_action(parsed_args) + + mock_getid.assert_called_once_with(self.backup_client, "fake_backup") + calls = [mock.call(backup_1), mock.call(backup_2)] + self.backup_client.delete.assert_has_calls(calls) + class TestBackupCreate(TestBackups): |