Merge "Fix object storage bulk middleware client return value"

This commit is contained in:
Jenkins 2017-07-24 10:41:33 +00:00 committed by Gerrit Code Review
commit 2a1271b924
4 changed files with 97 additions and 27 deletions

View File

@ -58,9 +58,9 @@ class BulkTest(base.BaseObjectTest):
# upload an archived file
with open(filepath) as fh:
mydata = fh.read()
resp, body = self.bulk_client.upload_archive(
resp = self.bulk_client.upload_archive(
upload_path='', data=mydata, archive_file_format='tar')
return resp, body
return resp
def _check_contents_deleted(self, container_name):
param = {'format': 'txt'}
@ -73,21 +73,20 @@ class BulkTest(base.BaseObjectTest):
def test_extract_archive(self):
# Test bulk operation of file upload with an archived file
filepath, container_name, object_name = self._create_archive()
resp, _ = self._upload_archive(filepath)
resp = self._upload_archive(filepath)
self.containers.append(container_name)
# When uploading an archived file with the bulk operation, the response
# does not contain 'content-length' header. This is the special case,
# therefore the existence of response headers is checked without
# custom matcher.
self.assertIn('transfer-encoding', resp)
self.assertIn('content-type', resp)
self.assertIn('x-trans-id', resp)
self.assertIn('date', resp)
self.assertIn('transfer-encoding', resp.response)
self.assertIn('content-type', resp.response)
self.assertIn('x-trans-id', resp.response)
self.assertIn('date', resp.response)
# Check only the format of common headers with custom matcher
self.assertThat(resp, custom_matchers.AreAllWellFormatted())
self.assertThat(resp.response, custom_matchers.AreAllWellFormatted())
param = {'format': 'json'}
resp, body = self.account_client.list_account_containers(param)
@ -112,19 +111,19 @@ class BulkTest(base.BaseObjectTest):
self._upload_archive(filepath)
data = '%s/%s\n%s' % (container_name, object_name, container_name)
resp, _ = self.bulk_client.delete_bulk_data(data=data)
resp = self.bulk_client.delete_bulk_data(data=data)
# When deleting multiple files using the bulk operation, the response
# does not contain 'content-length' header. This is the special case,
# therefore the existence of response headers is checked without
# custom matcher.
self.assertIn('transfer-encoding', resp)
self.assertIn('content-type', resp)
self.assertIn('x-trans-id', resp)
self.assertIn('date', resp)
self.assertIn('transfer-encoding', resp.response)
self.assertIn('content-type', resp.response)
self.assertIn('x-trans-id', resp.response)
self.assertIn('date', resp.response)
# Check only the format of common headers with custom matcher
self.assertThat(resp, custom_matchers.AreAllWellFormatted())
self.assertThat(resp.response, custom_matchers.AreAllWellFormatted())
# Check if uploaded contents are completely deleted
self._check_contents_deleted(container_name)
@ -138,19 +137,19 @@ class BulkTest(base.BaseObjectTest):
data = '%s/%s\n%s' % (container_name, object_name, container_name)
resp, _ = self.bulk_client.delete_bulk_data_with_post(data=data)
resp = self.bulk_client.delete_bulk_data_with_post(data=data)
# When deleting multiple files using the bulk operation, the response
# does not contain 'content-length' header. This is the special case,
# therefore the existence of response headers is checked without
# custom matcher.
self.assertIn('transfer-encoding', resp)
self.assertIn('content-type', resp)
self.assertIn('x-trans-id', resp)
self.assertIn('date', resp)
self.assertIn('transfer-encoding', resp.response)
self.assertIn('content-type', resp.response)
self.assertIn('x-trans-id', resp.response)
self.assertIn('date', resp.response)
# Check only the format of common headers with custom matcher
self.assertThat(resp, custom_matchers.AreAllWellFormatted())
self.assertThat(resp.response, custom_matchers.AreAllWellFormatted())
# Check if uploaded contents are completely deleted
self._check_contents_deleted(container_name)

View File

@ -31,7 +31,7 @@ class BulkMiddlewareClient(rest_client.RestClient):
headers = {}
resp, body = self.put(url, data, headers)
self.expected_success(200, resp.status)
return resp, body
return rest_client.ResponseBodyData(resp, body)
def delete_bulk_data(self, data=None, headers=None):
"""Delete multiple objects or containers from their account.
@ -43,9 +43,9 @@ class BulkMiddlewareClient(rest_client.RestClient):
if headers is None:
headers = {}
resp, body = self.delete(url, headers=headers, body=data)
resp, body = self.delete(url, headers, data)
self.expected_success(200, resp.status)
return resp, body
return rest_client.ResponseBodyData(resp, body)
def delete_bulk_data_with_post(self, data=None, headers=None):
"""Delete multiple objects or containers with POST request.
@ -57,6 +57,6 @@ class BulkMiddlewareClient(rest_client.RestClient):
if headers is None:
headers = {}
resp, body = self.post(url, headers=headers, body=data)
resp, body = self.post(url, data, headers)
self.expected_success([200, 204], resp.status)
return resp, body
return rest_client.ResponseBodyData(resp, body)

View File

@ -32,6 +32,7 @@ class BaseServiceTest(base.TestCase):
def check_service_client_function(self, function, function2mock,
body, to_utf=False, status=200,
headers=None, mock_args=None,
resp_as_string=False,
**kwargs):
"""Mock a service client function for unit testing.
@ -53,6 +54,9 @@ class BaseServiceTest(base.TestCase):
``assert_called_once_with(foo='bar')`` is called.
* If mock_args='foo' then ``assert_called_once_with('foo')``
is called.
:param resp_as_string: Whether response body is retruned as string.
This is for service client methods which return ResponseBodyData
object.
:param kwargs: kwargs that are passed to function.
"""
mocked_response = self.create_response(body, to_utf, status, headers)
@ -62,8 +66,9 @@ class BaseServiceTest(base.TestCase):
resp = function(**kwargs)
else:
resp = function()
if resp_as_string:
resp = resp.data
self.assertEqual(body, resp)
if isinstance(mock_args, list):
fixture.mock.assert_called_once_with(*mock_args)
elif isinstance(mock_args, dict):

View File

@ -0,0 +1,66 @@
# Copyright 2017 NEC Corporation. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
from tempest.services.object_storage import bulk_middleware_client
from tempest.tests.lib import fake_auth_provider
from tempest.tests.lib.services import base
class TestBulkMiddlewareClient(base.BaseServiceTest):
def setUp(self):
super(TestBulkMiddlewareClient, self).setUp()
fake_auth = fake_auth_provider.FakeAuthProvider()
self.client = bulk_middleware_client.BulkMiddlewareClient(
fake_auth, 'object-storage', 'regionOne')
def test_upload_archive(self):
url = 'test_path?extract-archive=tar'
data = 'test_data'
self.check_service_client_function(
self.client.upload_archive,
'tempest.lib.common.rest_client.RestClient.put',
{},
mock_args=[url, data, {}],
resp_as_string=True,
upload_path='test_path', data=data, archive_file_format='tar')
def test_delete_bulk_data(self):
url = '?bulk-delete'
data = 'test_data'
self.check_service_client_function(
self.client.delete_bulk_data,
'tempest.lib.common.rest_client.RestClient.delete',
{},
mock_args=[url, {}, data],
resp_as_string=True,
data=data)
def _test_delete_bulk_data_with_post(self, status):
url = '?bulk-delete'
data = 'test_data'
self.check_service_client_function(
self.client.delete_bulk_data_with_post,
'tempest.lib.common.rest_client.RestClient.post',
{},
mock_args=[url, data, {}],
resp_as_string=True,
status=status,
data=data)
def test_delete_bulk_data_with_post_200(self):
self._test_delete_bulk_data_with_post(200)
def test_delete_bulk_data_with_post_204(self):
self._test_delete_bulk_data_with_post(204)