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