92590ca743
Moves unit tests under the ims directory to the root test directory Change-Id: Idc8e356ce74970bacf16c3046b72e57fe9cbe5eb
41 lines
1022 B
Python
41 lines
1022 B
Python
from orm.services.image_manager.ims.persistency.wsme import models
|
|
from orm.tests.unit.ims import FunctionalTest
|
|
|
|
import mock
|
|
|
|
GROUP_REGIONS = [
|
|
"DPK",
|
|
"SNA1",
|
|
"SNA2"
|
|
]
|
|
|
|
|
|
class TestModels(FunctionalTest):
|
|
|
|
def setUp(self):
|
|
FunctionalTest.setUp(self)
|
|
models.get_regions_of_group = mock.MagicMock(return_value=GROUP_REGIONS)
|
|
models.set_utils_conf = mock.MagicMock()
|
|
|
|
def test_handle_group_success(self):
|
|
image = get_image_model()
|
|
image.handle_region_group()
|
|
|
|
self.assertEqual(len(image.regions), 3)
|
|
|
|
def test_handle_group_not_found(self):
|
|
models.get_regions_of_group = mock.MagicMock(return_value=None)
|
|
image = get_image_model()
|
|
|
|
self.assertRaises(models.ErrorStatus, image.handle_region_group,)
|
|
|
|
|
|
def get_image_model():
|
|
"""this function create a customer model object for testing
|
|
:return: new customer object
|
|
"""
|
|
|
|
image = models.Image(id='a', regions=[models.Region(name='r1', type='group')])
|
|
|
|
return image
|