def setUp(self): mock_clients = self._create_service_mock('resource_management') self.rms = ResourceManagementService() self.rms.clients = mock_clients self.object_definition = '''
def setUp(self): mock_clients = self._create_service_mock('resource_management') self.rms = ResourceManagementService() self.rms.clients = mock_clients self.object_definition = '''
class TestResourceManagementServiceUnit(PyonTestCase): def setUp(self): mock_clients = self._create_service_mock('resource_management') self.rms = ResourceManagementService() self.rms.clients = mock_clients self.object_definition = ''' Policy9: !Extends_InformationResource enabled: True definition: {} ''' def test_create_resource(self): # Create ResourceType rt = Mock() rt.name = "bad name" with self.assertRaises(BadRequest): self.rms.create_resource_type(rt, "123") rt.name = "good_name" bad_object_id = None with self.assertRaises(BadRequest): self.rms.create_resource_type(rt, bad_object_id) # Create ObjectType ot = Mock() ot.definition = self.object_definition ot.name = "good_name" ot.description = "This is just a test. No need to panic" self.rms.clients.resource_registry.read.return_value = ot resource_id_return_value = '123' version_return_value = 1 self.rms.clients.resource_registry.create.return_value = [resource_id_return_value, version_return_value] self.rms.clients.resource_registry.create_association.return_value = '999' object_id = "444" resource_id = self.rms.create_resource_type(rt, object_id) self.assertEqual(resource_id, '123') self.rms.clients.resource_registry.read.assert_called_once_with(object_id, '') self.rms.clients.resource_registry.create.assert_called_once_with(rt) self.rms.clients.resource_registry.create_association.assert_called_once_with(resource_id_return_value, PRED.hasObjectType, object_id, 'H2H') def test_read_resource(self): with self.assertRaises(BadRequest): self.rms.read_resource_type(None) # Create ResourceType rt = Mock() rt.name = "good_name" self.rms.clients.resource_registry.read.return_value = rt rt_read = self.rms.read_resource_type("123") self.assertTrue(rt_read is rt) self.rms.clients.resource_registry.read.assert_called_once_with('123','') def test_read_not_found(self): self.rms.clients.resource_registry.read.side_effect = NotFound with self.assertRaises(NotFound): self.rms.read_resource_type("0xBadC0ffee") self.rms.clients.resource_registry.read.assert_called_once_with("0xBadC0ffee", '') def test_delete_resource(self): with self.assertRaises(BadRequest): self.rms.delete_resource_type(None, "123") with self.assertRaises(BadRequest): self.rms.delete_resource_type("123", None) self.rms.clients.resource_registry.delete.return_value = True association_id = '999' self.rms.clients.resource_registry.get_association.return_value = association_id self.rms.clients.resource_registry.delete_association.return_value = True resource_id = '123' object_id = '456' status = self.rms.delete_resource_type(resource_id, object_id) self.assertTrue(status, True) self.rms.clients.resource_registry.delete.assert_called_once_with(resource_id) self.rms.clients.resource_registry.get_association.assert_called_once_with(resource_id, PRED.hasObjectType, object_id, None, False) self.rms.clients.resource_registry.delete_association.assert_called_once_with(association_id) def test_delete_not_found(self): self.rms.clients.resource_registry.get_association.side_effect = NotFound resource_id = "123" object_id = '456' with self.assertRaises(NotFound): self.rms.delete_resource_type(resource_id, object_id) self.rms.clients.resource_registry.get_association.assert_called_once_with(resource_id, PRED.hasObjectType, object_id, None, False) association_id = '999' self.rms.clients.resource_registry.delete.side_effect = NotFound self.rms.clients.resource_registry.get_association.side_effect = None self.rms.clients.resource_registry.get_association.return_value = association_id self.rms.clients.resource_registry.delete_association.return_value = True with self.assertRaises(NotFound): self.rms.delete_resource_type(resource_id, object_id) self.rms.clients.resource_registry.get_association.assert_called_with(resource_id, PRED.hasObjectType, object_id, None, False) self.rms.clients.resource_registry.delete_association.assert_called_with(association_id) self.rms.clients.resource_registry.delete.assert_called_with(resource_id)
class TestResourceManagementServiceUnit(PyonTestCase): def setUp(self): mock_clients = self._create_service_mock('resource_management') self.rms = ResourceManagementService() self.rms.clients = mock_clients self.object_definition = ''' Policy9: !Extends_InformationResource enabled: True definition: {} ''' def test_create_resource(self): # Create ResourceType rt = Mock() rt.name = "bad name" with self.assertRaises(BadRequest): self.rms.create_resource_type(rt, "123") rt.name = "good_name" bad_object_id = None with self.assertRaises(BadRequest): self.rms.create_resource_type(rt, bad_object_id) # Create ObjectType ot = Mock() ot.definition = self.object_definition ot.name = "good_name" ot.description = "This is just a test. No need to panic" self.rms.clients.resource_registry.read.return_value = ot resource_id_return_value = '123' version_return_value = 1 self.rms.clients.resource_registry.create.return_value = [ resource_id_return_value, version_return_value ] self.rms.clients.resource_registry.create_association.return_value = '999' object_id = "444" resource_id = self.rms.create_resource_type(rt, object_id) self.assertEqual(resource_id, '123') self.rms.clients.resource_registry.read.assert_called_once_with( object_id, '') self.rms.clients.resource_registry.create.assert_called_once_with(rt) self.rms.clients.resource_registry.create_association.assert_called_once_with( resource_id_return_value, PRED.hasObjectType, object_id, None) def test_read_resource(self): with self.assertRaises(BadRequest): self.rms.read_resource_type(None) # Create ResourceType rt = Mock() rt.name = "good_name" self.rms.clients.resource_registry.read.return_value = rt rt_read = self.rms.read_resource_type("123") self.assertTrue(rt_read is rt) self.rms.clients.resource_registry.read.assert_called_once_with( '123', '') def test_read_not_found(self): self.rms.clients.resource_registry.read.side_effect = NotFound with self.assertRaises(NotFound): self.rms.read_resource_type("0xBadC0ffee") self.rms.clients.resource_registry.read.assert_called_once_with( "0xBadC0ffee", '') def test_delete_resource(self): with self.assertRaises(BadRequest): self.rms.delete_resource_type(None, "123") with self.assertRaises(BadRequest): self.rms.delete_resource_type("123", None) self.rms.clients.resource_registry.delete.return_value = True association_id = '999' self.rms.clients.resource_registry.get_association.return_value = association_id self.rms.clients.resource_registry.delete_association.return_value = True resource_id = '123' object_id = '456' status = self.rms.delete_resource_type(resource_id, object_id) self.assertTrue(status, True) self.rms.clients.resource_registry.delete.assert_called_once_with( resource_id) self.rms.clients.resource_registry.get_association.assert_called_once_with( resource_id, PRED.hasObjectType, object_id, None, False) self.rms.clients.resource_registry.delete_association.assert_called_once_with( association_id) def test_delete_not_found(self): self.rms.clients.resource_registry.get_association.side_effect = NotFound resource_id = "123" object_id = '456' with self.assertRaises(NotFound): self.rms.delete_resource_type(resource_id, object_id) self.rms.clients.resource_registry.get_association.assert_called_once_with( resource_id, PRED.hasObjectType, object_id, None, False) association_id = '999' self.rms.clients.resource_registry.delete.side_effect = NotFound self.rms.clients.resource_registry.get_association.side_effect = None self.rms.clients.resource_registry.get_association.return_value = association_id self.rms.clients.resource_registry.delete_association.return_value = True with self.assertRaises(NotFound): self.rms.delete_resource_type(resource_id, object_id) self.rms.clients.resource_registry.get_association.assert_called_with( resource_id, PRED.hasObjectType, object_id, None, False) self.rms.clients.resource_registry.delete_association.assert_called_with( association_id) self.rms.clients.resource_registry.delete.assert_called_with( resource_id)