def test_check_image_service_client_already_set(self, mock_gclient, mock_sess, mock_adapter, mock_sauth, mock_auth): def func(self): return True self.service.client = True wrapped_func = image_service.check_image_service(func) self.assertTrue(wrapped_func(self.service)) self.assertEqual(0, mock_gclient.call_count) self.assertEqual(0, mock_sess.call_count) self.assertEqual(0, mock_adapter.call_count) self.assertEqual(0, mock_auth.call_count) self.assertEqual(0, mock_sauth.call_count)
def test_check_image_service__no_auth(self, mock_gclient, mock_sess, mock_adapter, mock_sauth, mock_auth): def func(service, *args, **kwargs): return args, kwargs self.config(endpoint_override='foo', auth_type='none', group='glance') mock_adapter.return_value = adapter = mock.Mock() adapter.get_endpoint.return_value = 'foo' uuid = uuidutils.generate_uuid() params = {'image_href': uuid} wrapped_func = image_service.check_image_service(func) self.assertEqual(((), params), wrapped_func(self.service, **params)) self.assertEqual('none', image_service.CONF.glance.auth_type) self._assert_client_call(mock_gclient, 'foo') mock_sess.assert_called_once_with('glance') mock_adapter.assert_called_once_with('glance', session=mock.sentinel.session, auth=mock.sentinel.auth) self.assertEqual(0, mock_sauth.call_count)
def test_check_image_service__token_auth(self, mock_gclient, mock_sess, mock_adapter, mock_sauth, mock_auth): def func(service, *args, **kwargs): return args, kwargs self.service.context = context.RequestContext( auth_token='token', global_request_id='global') mock_adapter.return_value = adapter = mock.Mock() adapter.get_endpoint.return_value = 'glance_url' uuid = uuidutils.generate_uuid() params = {'image_href': uuid} wrapped_func = image_service.check_image_service(func) self.assertEqual(((), params), wrapped_func(self.service, **params)) self._assert_client_call(mock_gclient, 'glance_url', user=True) mock_sess.assert_called_once_with('glance') mock_adapter.assert_called_once_with('glance', session=mock.sentinel.session, auth=mock.sentinel.auth) mock_sauth.assert_called_once_with(self.service.context, 'glance_url', mock.sentinel.auth) mock_auth.assert_called_once_with('glance')