Add raise unit test for session in sqlalchemy
Change-Id: I1fc045010732f178abb4c5ea75e02b96e7a83b3d
This commit is contained in:
parent
2dc322497e
commit
ac48abe735
|
@ -18,8 +18,11 @@
|
||||||
"""Tests for manipulating job via the DB API"""
|
"""Tests for manipulating job via the DB API"""
|
||||||
|
|
||||||
import copy
|
import copy
|
||||||
|
import mock
|
||||||
|
from mock import patch
|
||||||
from oslo_config import cfg
|
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 import common
|
||||||
from freezer_api.tests.unit.sqlalchemy import base
|
from freezer_api.tests.unit.sqlalchemy import base
|
||||||
|
|
||||||
|
@ -31,6 +34,7 @@ class DbSessionTestCase(base.DbTestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
super(DbSessionTestCase, self).setUp()
|
super(DbSessionTestCase, self).setUp()
|
||||||
self.fake_session_0 = common.get_fake_session_0()
|
self.fake_session_0 = common.get_fake_session_0()
|
||||||
|
self.fake_user_id = self.fake_session_0.get('user_id')
|
||||||
self.fake_session_0.pop('session_id')
|
self.fake_session_0.pop('session_id')
|
||||||
self.fake_session_2 = common.get_fake_session_2()
|
self.fake_session_2 = common.get_fake_session_2()
|
||||||
self.fake_session_2.pop('session_id')
|
self.fake_session_2.pop('session_id')
|
||||||
|
@ -405,3 +409,16 @@ class DbSessionTestCase(base.DbTestCase):
|
||||||
for index in range(len(result)):
|
for index in range(len(result)):
|
||||||
sessionmap = result[index]
|
sessionmap = result[index]
|
||||||
self.assertEqual(100, sessionmap['hold_off'])
|
self.assertEqual(100, sessionmap['hold_off'])
|
||||||
|
|
||||||
|
@patch('freezer_api.db.sqlalchemy.api.get_session')
|
||||||
|
def test_raise_add_session_exist(self, mock_get_session):
|
||||||
|
mock_get_session.return_value = mock.MagicMock()
|
||||||
|
self.assertRaises(freezer_api_exc.DocumentExists,
|
||||||
|
self.dbapi.add_session, self.fake_user_id,
|
||||||
|
self.fake_session_0)
|
||||||
|
|
||||||
|
def test_raise_update_session_noexist(self):
|
||||||
|
self.assertRaises(freezer_api_exc.DocumentNotFound,
|
||||||
|
self.dbapi.update_session, self.fake_user_id,
|
||||||
|
self.fake_session_id,
|
||||||
|
self.fake_session_0)
|
||||||
|
|
|
@ -18,7 +18,10 @@
|
||||||
"""Tests for manipulating job via the DB API"""
|
"""Tests for manipulating job via the DB API"""
|
||||||
|
|
||||||
import copy
|
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 import common
|
||||||
from freezer_api.tests.unit.sqlalchemy import base
|
from freezer_api.tests.unit.sqlalchemy import base
|
||||||
|
|
||||||
|
@ -28,6 +31,8 @@ class DbSessionTestCase(base.DbTestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
super(DbSessionTestCase, self).setUp()
|
super(DbSessionTestCase, self).setUp()
|
||||||
self.fake_session_0 = common.get_fake_session_0()
|
self.fake_session_0 = common.get_fake_session_0()
|
||||||
|
self.fake_project_id = self.fake_session_0.get('project_id')
|
||||||
|
self.fake_user_id = self.fake_session_0.get('user_id')
|
||||||
self.fake_session_0.pop('session_id')
|
self.fake_session_0.pop('session_id')
|
||||||
self.fake_session_2 = common.get_fake_session_2()
|
self.fake_session_2 = common.get_fake_session_2()
|
||||||
self.fake_session_2.pop('session_id')
|
self.fake_session_2.pop('session_id')
|
||||||
|
@ -461,3 +466,18 @@ class DbSessionTestCase(base.DbTestCase):
|
||||||
for index in range(len(result)):
|
for index in range(len(result)):
|
||||||
sessionmap = result[index]
|
sessionmap = result[index]
|
||||||
self.assertEqual(100, sessionmap['hold_off'])
|
self.assertEqual(100, sessionmap['hold_off'])
|
||||||
|
|
||||||
|
@patch('freezer_api.db.sqlalchemy.api.get_session')
|
||||||
|
def test_raise_add_session_exist(self, mock_get_session):
|
||||||
|
mock_get_session.return_value = mock.MagicMock()
|
||||||
|
self.assertRaises(freezer_api_exc.DocumentExists,
|
||||||
|
self.dbapi.add_session, self.fake_user_id,
|
||||||
|
self.fake_session_0,
|
||||||
|
project_id=self.fake_project_id)
|
||||||
|
|
||||||
|
def test_raise_update_session_noexist(self):
|
||||||
|
self.assertRaises(freezer_api_exc.DocumentNotFound,
|
||||||
|
self.dbapi.update_session, self.fake_user_id,
|
||||||
|
self.fake_session_id,
|
||||||
|
self.fake_session_0,
|
||||||
|
project_id=self.fake_project_id)
|
||||||
|
|
Loading…
Reference in New Issue