diff --git a/freezer_api/tests/unit/sqlalchemy/v1/test_backup.py b/freezer_api/tests/unit/sqlalchemy/v1/test_backup.py index e4f7bc47..8ff2948c 100644 --- a/freezer_api/tests/unit/sqlalchemy/v1/test_backup.py +++ b/freezer_api/tests/unit/sqlalchemy/v1/test_backup.py @@ -17,8 +17,11 @@ 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 @@ -106,3 +109,14 @@ class DbBackupTestCase(base.DbTestCase): for index in range(len(result)): backupmap = result[index] self.assertEqual(backupids[index], backupmap['backup_id']) + + @patch('freezer_api.common.elasticv2_utils.BackupMetadataDoc') + @patch('freezer_api.common.utils.BackupMetadataDoc') + def test_raise_add_backup(self, mock1_BackupMetadataDoc, + mock_BackupMetadataDoc): + mock1_BackupMetadataDoc().is_valid.return_value = None + mock_BackupMetadataDoc().is_valid.return_value = None + mock_doc = mock.MagicMock() + self.assertRaises(freezer_api_exc.BadDataFormat, + self.dbapi.add_backup, self.fake_user_id, '12343', + mock_doc) diff --git a/freezer_api/tests/unit/sqlalchemy/v1/test_job.py b/freezer_api/tests/unit/sqlalchemy/v1/test_job.py index bc1c9290..865d11f7 100644 --- a/freezer_api/tests/unit/sqlalchemy/v1/test_job.py +++ b/freezer_api/tests/unit/sqlalchemy/v1/test_job.py @@ -18,8 +18,11 @@ """Tests for manipulating job 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 @@ -37,6 +40,7 @@ class DbJobTestCase(base.DbTestCase): self.fake_job_3 = common.get_fake_job_3() self.fake_job_3.pop('job_id') self.fake_job_id = common.get_fake_job_id() + self.fake_user_id = self.fake_job_0.get('user_id') CONF.enable_v1_api = True def tearDown(self): @@ -403,3 +407,10 @@ class DbJobTestCase(base.DbTestCase): for index in range(len(result)): jobmap = result[index] self.assertEqual('node1', jobmap['client_id']) + + @patch('freezer_api.db.sqlalchemy.api.get_job') + def test_raise_add_job(self, mock_get_job): + mock_get_job.return_value = mock.MagicMock() + self.assertRaises(freezer_api_exc.DocumentExists, + self.dbapi.add_job, self.fake_user_id, + self.fake_job_0) diff --git a/freezer_api/tests/unit/sqlalchemy/v2/test_backup.py b/freezer_api/tests/unit/sqlalchemy/v2/test_backup.py index 57a4316e..1882a506 100644 --- a/freezer_api/tests/unit/sqlalchemy/v2/test_backup.py +++ b/freezer_api/tests/unit/sqlalchemy/v2/test_backup.py @@ -17,7 +17,10 @@ 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 @@ -106,3 +109,15 @@ class DbBackupTestCase(base.DbTestCase): for index in range(len(result)): backupmap = result[index] self.assertEqual(backupids[index], backupmap['backup_id']) + + @patch('freezer_api.common.elasticv2_utils.BackupMetadataDoc') + @patch('freezer_api.common.utils.BackupMetadataDoc') + def test_raise_add_backup(self, mock1_BackupMetadataDoc, + mock_BackupMetadataDoc): + mock1_BackupMetadataDoc().is_valid.return_value = None + mock_BackupMetadataDoc().is_valid.return_value = None + mock_doc = mock.MagicMock() + self.assertRaises(freezer_api_exc.BadDataFormat, + self.dbapi.add_backup, self.fake_user_id, '12343', + mock_doc, + project_id=self.fake_project_id) diff --git a/freezer_api/tests/unit/sqlalchemy/v2/test_job.py b/freezer_api/tests/unit/sqlalchemy/v2/test_job.py index 6b854e29..eeb399bd 100644 --- a/freezer_api/tests/unit/sqlalchemy/v2/test_job.py +++ b/freezer_api/tests/unit/sqlalchemy/v2/test_job.py @@ -18,7 +18,10 @@ """Tests for manipulating job 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 @@ -34,6 +37,7 @@ class DbJobTestCase(base.DbTestCase): self.fake_job_3 = common.get_fake_job_3() self.fake_job_3.pop('job_id') self.fake_project_id = self.fake_job_0.get('project_id') + self.fake_user_id = self.fake_job_0.get('user_id') self.fake_job_id = common.get_fake_job_id() def test_add_and_get_job(self): @@ -427,3 +431,11 @@ class DbJobTestCase(base.DbTestCase): for index in range(len(result)): jobmap = result[index] self.assertEqual('node1', jobmap['client_id']) + + @patch('freezer_api.db.sqlalchemy.api.get_job') + def test_raise_add_job(self, mock_get_job): + mock_get_job.return_value = mock.MagicMock() + self.assertRaises(freezer_api_exc.DocumentExists, + self.dbapi.add_job, self.fake_user_id, + self.fake_job_0, + project_id=self.fake_project_id)