Add raise unit test for action and backup in sqlalchemy

Change-Id: I3f0feeea92cc7ba97e9952ced3a56fd96abb03b2
This commit is contained in:
gengchc2 2018-12-21 01:35:23 -08:00
parent d41f3ab696
commit 02decd342d
4 changed files with 38 additions and 0 deletions

View File

@ -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)

View File

@ -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,

View File

@ -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)

View File

@ -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,