def test_get_server_dependent_resources(self, mock_server_get,
mock_image_get):
vm = server_info(id='server1',
type=constants.SERVER_RESOURCE_TYPE,
name='nameserver1',
image=dict(id='123', name='name123'))
image = image_info(id='123', name='name123', owner='abcd')
plugin = ImageProtectablePlugin(self._context)
mock_server_get.return_value = vm
mock_image_get.return_value = image
self.assertEqual(plugin.get_dependent_resources(self._context, vm),
[resource.Resource(type=constants.IMAGE_RESOURCE_TYPE,
id='123', name='name123')])
def test_get_project_dependent_resources(self, mock_image_list):
project = project_info(id='abcd', type=constants.PROJECT_RESOURCE_TYPE,
name='nameabcd')
plugin = ImageProtectablePlugin(self._context)
mock_image_list.return_value = \
[
image_info('123', 'abcd', 'nameabcd'),
image_info('456', 'efgh', 'nameefgh'),
]
self.assertEqual(
plugin.get_dependent_resources(self._context, project),
[resource.Resource(type=constants.IMAGE_RESOURCE_TYPE,
name='nameabcd',
id='123')])
def test_get_server_dependent_resources(self, mock_server_get,
mock_image_get):
vm = server_info(id='server1',
type=constants.SERVER_RESOURCE_TYPE,
name='nameserver1',
image=dict(id='123', name='name123'))
image = image_info(id='123', name='name123', owner='abcd')
plugin = ImageProtectablePlugin(self._context)
mock_server_get.return_value = vm
mock_image_get.return_value = image
self.assertEqual(plugin.get_dependent_resources(self._context, vm), [
resource.Resource(
type=constants.IMAGE_RESOURCE_TYPE, id='123', name='name123')
])
def test_get_project_dependent_resources(self, mock_image_list):
project = project_info(id='abcd',
type=constants.PROJECT_RESOURCE_TYPE,
name='nameabcd')
plugin = ImageProtectablePlugin(self._context)
mock_image_list.return_value = [
image_info('123', 'abcd', 'nameabcd', 'active'),
image_info('456', 'efgh', 'nameefgh', 'active'),
]
self.assertEqual(
plugin.get_dependent_resources(self._context, project), [
resource.Resource(type=constants.IMAGE_RESOURCE_TYPE,
name='nameabcd',
id='123')
])
def test_get_server_dependent_resources(self, mock_generate_session,
mock_server_get, mock_image_get):
vm = server_info(id='server1',
type=constants.SERVER_RESOURCE_TYPE,
name='nameserver1',
image=dict(id='123', name='name123'))
image = image_info(id='123',
name='name123',
owner='abcd',
status='active')
plugin = ImageProtectablePlugin(self._context)
mock_generate_session.return_value = keystone_session.Session(
auth=None)
mock_server_get.return_value = vm
mock_image_get.return_value = image
self.assertEqual(plugin.get_dependent_resources(self._context, vm), [
resource.Resource(type=constants.IMAGE_RESOURCE_TYPE,
id='123',
name='name123',
extra_info={'server_id': 'server1'})
])
