class RoleManagerTests(TestCase):
def setUp(self):
super(RoleManagerTests, self).setUp()
self.role_manager = RoleManager()
self.template_store = TemplateStore()
def test_list_roles(self):
# Test
self._populate_roles()
all_roles = self.role_manager.list_roles()
# Verify
self.assertEqual(3, len(all_roles))
self.assertTrue(isinstance(all_roles[0], Role))
all_roles.sort(key=lambda x: '%s-%s' % (x.name, x.version))
self.assertEqual('r1', all_roles[0].name)
self.assertEqual(1, all_roles[0].version)
self.assertEqual('r1', all_roles[1].name)
self.assertEqual(2, all_roles[1].version)
self.assertEqual('r2', all_roles[2].name)
self.assertEqual(1, all_roles[2].version)
def test_list_roles_only_latest(self):
# Setup
list_mock = mock.MagicMock()
self.role_manager.template_store.list = list_mock
list_mock.return_value = []
# Test
self.role_manager.list_roles(only_latest=True)
# Verify
list_mock.assert_called_once_with(only_latest=True)
def test_retrieve_role_by_uuid(self):
# Test
added_roles = self._populate_roles()
found = self.role_manager.retrieve_role_by_uuid(added_roles[0].uuid)
# Verify
self.assertTrue(found is not None)
self.assertTrue(isinstance(found, Role))
self.assertEqual(found.name, 'r1')
self.assertEqual(found.version, 2)
def test_retrieve_role_by_fake_uuid(self):
self.assertRaises(UnknownUUID,
self.role_manager.retrieve_role_by_uuid,
'fake')
def _populate_roles(self):
r1 = self.template_store.create('r1', TEST_TEMPLATE)
r1 = self.template_store.update(r1.uuid, TEST_TEMPLATE)
r2 = self.template_store.create('r2', TEST_TEMPLATE)
return [r1, r2]
class RoleManagerTests(TestCase):
def setUp(self):
super(RoleManagerTests, self).setUp()
self.role_manager = RoleManager()
self.template_store = TemplateStore()
def test_list_roles(self):
# Test
self._populate_roles()
all_roles = self.role_manager.list_roles()
# Verify
self.assertEqual(3, len(all_roles))
self.assertTrue(isinstance(all_roles[0], Role))
all_roles.sort(key=lambda x: '%s-%s' % (x.name, x.version))
self.assertEqual('r1', all_roles[0].name)
self.assertEqual(1, all_roles[0].version)
self.assertEqual('r1', all_roles[1].name)
self.assertEqual(2, all_roles[1].version)
self.assertEqual('r2', all_roles[2].name)
self.assertEqual(1, all_roles[2].version)
def test_list_roles_only_latest(self):
# Setup
list_mock = mock.MagicMock()
self.role_manager.template_store.list = list_mock
list_mock.return_value = []
# Test
self.role_manager.list_roles(only_latest=True)
# Verify
list_mock.assert_called_once_with(only_latest=True)
def test_retrieve_role_by_uuid(self):
# Test
added_roles = self._populate_roles()
found = self.role_manager.retrieve_role_by_uuid(added_roles[0].uuid)
# Verify
self.assertTrue(found is not None)
self.assertTrue(isinstance(found, Role))
self.assertEqual(found.name, 'r1')
self.assertEqual(found.version, 2)
def test_retrieve_role_by_fake_uuid(self):
self.assertRaises(UnknownUUID, self.role_manager.retrieve_role_by_uuid,
'fake')
def _populate_roles(self):
r1 = self.template_store.create('r1', TEST_TEMPLATE)
r1 = self.template_store.update(r1.uuid, TEST_TEMPLATE)
r2 = self.template_store.create('r2', TEST_TEMPLATE)
return [r1, r2]
def get_all(self):
"""Returns all roles.
An empty list is returned if no roles are present.
:return: list of roles; empty list if none are found
:rtype: list of tuskar.api.controllers.v2.models.Role
"""
LOG.debug('Retrieving all roles')
manager = RoleManager()
all_roles = manager.list_roles(only_latest=False)
transfer_roles = [models.Role.from_tuskar_model(r) for r in all_roles]
return transfer_roles
def get_all(self):
"""Returns all roles.
An empty list is returned if no roles are present.
:return: list of roles; empty list if none are found
:rtype: list of tuskar.api.controllers.v2.models.Role
"""
LOG.debug('Retrieving all roles')
manager = RoleManager()
all_roles = manager.list_roles(only_latest=False)
transfer_roles = [models.Role.from_tuskar_model(r) for r in all_roles]
return transfer_roles
