Add raise unit test for job and backup in sqlalchemy

Change-Id: I7e22e6a3ac53eee6ab653a62be21829a42162623

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)

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)

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)

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)