diff --git a/freezer_api/tests/unit/sqlalchemy/v1/test_action.py b/freezer_api/tests/unit/sqlalchemy/v1/test_action.py index 82f4ec8d..be374ffd 100644 --- a/freezer_api/tests/unit/sqlalchemy/v1/test_action.py +++ b/freezer_api/tests/unit/sqlalchemy/v1/test_action.py @@ -16,8 +16,11 @@ """Tests for manipulating Action via the DB API""" import copy +import mock +from mock import patch from oslo_config import cfg +from freezer_api.common import exceptions as freezer_api_exc from freezer_api.tests.unit import common from freezer_api.tests.unit.sqlalchemy import base @@ -33,6 +36,7 @@ class DbActionTestCase(base.DbTestCase): self.fake_action_3 = common.get_fake_action_3() self.freezer_action_0 = self.fake_action_0.get('freezer_action') self.freezer_action_2 = self.fake_action_2.get('freezer_action') + self.fake_user_id = self.fake_action_0.get('user_id') self.fake_action_id = common.get_fake_action_id() CONF.enable_v1_api = True @@ -414,3 +418,10 @@ class DbActionTestCase(base.DbTestCase): self.assertIsNotNone(result) self.assertEqual(len(result), 20) + + @patch('freezer_api.db.sqlalchemy.api.get_action') + def test_raise_add_action(self, mock_get_action): + mock_get_action.return_value = mock.MagicMock() + self.assertRaises(freezer_api_exc.DocumentExists, + self.dbapi.add_action, self.fake_user_id, + self.fake_action_0) diff --git a/freezer_api/tests/unit/sqlalchemy/v1/test_backup.py b/freezer_api/tests/unit/sqlalchemy/v1/test_backup.py index 8ff2948c..97a34015 100644 --- a/freezer_api/tests/unit/sqlalchemy/v1/test_backup.py +++ b/freezer_api/tests/unit/sqlalchemy/v1/test_backup.py @@ -110,6 +110,13 @@ class DbBackupTestCase(base.DbTestCase): backupmap = result[index] self.assertEqual(backupids[index], backupmap['backup_id']) + @patch('freezer_api.db.sqlalchemy.api.get_backup') + def test_raise_add_backup_exist(self, mock_get_backup): + mock_get_backup.return_value = mock.MagicMock() + self.assertRaises(freezer_api_exc.DocumentExists, + self.dbapi.add_backup, self.fake_user_id, + '12343', self.fake_backup_metadata) + @patch('freezer_api.common.elasticv2_utils.BackupMetadataDoc') @patch('freezer_api.common.utils.BackupMetadataDoc') def test_raise_add_backup(self, mock1_BackupMetadataDoc, diff --git a/freezer_api/tests/unit/sqlalchemy/v2/test_action.py b/freezer_api/tests/unit/sqlalchemy/v2/test_action.py index 35dbc653..a39b388b 100644 --- a/freezer_api/tests/unit/sqlalchemy/v2/test_action.py +++ b/freezer_api/tests/unit/sqlalchemy/v2/test_action.py @@ -16,7 +16,10 @@ """Tests for manipulating Action via the DB API""" import copy +import mock +from mock import patch +from freezer_api.common import exceptions as freezer_api_exc from freezer_api.tests.unit import common from freezer_api.tests.unit.sqlalchemy import base @@ -31,6 +34,7 @@ class DbActionTestCase(base.DbTestCase): self.freezer_action_0 = self.fake_action_0.get('freezer_action') self.freezer_action_2 = self.fake_action_2.get('freezer_action') self.fake_project_id = self.fake_action_0.get('project_id') + self.fake_user_id = self.fake_action_0.get('user_id') self.fake_action_id = common.get_fake_action_id() def test_add_and_get_action(self): @@ -438,3 +442,11 @@ class DbActionTestCase(base.DbTestCase): self.assertIsNotNone(result) self.assertEqual(len(result), 20) + + @patch('freezer_api.db.sqlalchemy.api.get_action') + def test_raise_add_action(self, mock_get_action): + mock_get_action.return_value = mock.MagicMock() + self.assertRaises(freezer_api_exc.DocumentExists, + self.dbapi.add_action, self.fake_user_id, + self.fake_action_0, + project_id=self.fake_project_id) diff --git a/freezer_api/tests/unit/sqlalchemy/v2/test_backup.py b/freezer_api/tests/unit/sqlalchemy/v2/test_backup.py index 1882a506..f5945892 100644 --- a/freezer_api/tests/unit/sqlalchemy/v2/test_backup.py +++ b/freezer_api/tests/unit/sqlalchemy/v2/test_backup.py @@ -110,6 +110,14 @@ class DbBackupTestCase(base.DbTestCase): backupmap = result[index] self.assertEqual(backupids[index], backupmap['backup_id']) + @patch('freezer_api.db.sqlalchemy.api.get_backup') + def test_raise_add_backup_exist(self, mock_get_backup): + mock_get_backup.return_value = mock.MagicMock() + self.assertRaises(freezer_api_exc.DocumentExists, + self.dbapi.add_backup, self.fake_user_id, + '12343', self.fake_backup_metadata, + project_id=self.fake_project_id) + @patch('freezer_api.common.elasticv2_utils.BackupMetadataDoc') @patch('freezer_api.common.utils.BackupMetadataDoc') def test_raise_add_backup(self, mock1_BackupMetadataDoc,