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)
