c86c404c31
Moves all tests under orm/services/flavor_manager to top level tests folder, and makes changes to get them passing. Change-Id: I6e1bff032a1a90e60529bd0ed42f1c0b822687df
42 lines
1.0 KiB
Python
Executable File
42 lines
1.0 KiB
Python
Executable File
from orm.services.flavor_manager.fms_rest.data.wsme import models
|
|
from orm.tests.unit.fms 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):
|
|
flavor = get_flavor_model()
|
|
flavor.handle_region_groups()
|
|
|
|
self.assertEqual(len(flavor.regions), 3)
|
|
|
|
def test_handle_group_not_found(self):
|
|
models.get_regions_of_group = mock.MagicMock(return_value=None)
|
|
flavor = get_flavor_model()
|
|
|
|
self.assertRaises(ValueError, flavor.handle_region_groups,)
|
|
|
|
|
|
def get_flavor_model():
|
|
"""this function create a customer model object for testing
|
|
|
|
:return: new customer object
|
|
"""
|
|
|
|
flavor = models.Flavor(id='1', regions=[models.Region(name='r1', type='group')])
|
|
|
|
return flavor
|