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'}) ])