218 lines
8.1 KiB
Python
218 lines
8.1 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2012 OpenStack, LLC
|
|
# 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.
|
|
|
|
import hashlib
|
|
import hmac
|
|
import urlparse
|
|
|
|
from tempest.common import http
|
|
from tempest.common.rest_client import RestClient
|
|
from tempest import exceptions
|
|
|
|
|
|
class ObjectClient(RestClient):
|
|
def __init__(self, config, username, password, auth_url, tenant_name=None):
|
|
super(ObjectClient, self).__init__(config, username, password,
|
|
auth_url, tenant_name)
|
|
|
|
self.service = self.config.object_storage.catalog_type
|
|
|
|
def create_object(self, container, object_name, data):
|
|
"""Create storage object."""
|
|
|
|
headers = dict(self.headers)
|
|
if not data:
|
|
headers['content-length'] = '0'
|
|
url = "%s/%s" % (str(container), str(object_name))
|
|
resp, body = self.put(url, data, headers)
|
|
return resp, body
|
|
|
|
def update_object(self, container, object_name, data):
|
|
"""Upload data to replace current storage object."""
|
|
return self.create_object(container, object_name, data)
|
|
|
|
def delete_object(self, container, object_name):
|
|
"""Delete storage object."""
|
|
url = "%s/%s" % (str(container), str(object_name))
|
|
resp, body = self.delete(url)
|
|
return resp, body
|
|
|
|
def update_object_metadata(self, container, object_name, metadata,
|
|
metadata_prefix='X-Object-Meta-'):
|
|
"""Add, remove, or change X-Object-Meta metadata for storage object."""
|
|
|
|
headers = {}
|
|
for key in metadata:
|
|
headers["%s%s" % (str(metadata_prefix), str(key))] = metadata[key]
|
|
|
|
url = "%s/%s" % (str(container), str(object_name))
|
|
resp, body = self.post(url, None, headers=headers)
|
|
return resp, body
|
|
|
|
def list_object_metadata(self, container, object_name):
|
|
"""List all storage object X-Object-Meta- metadata."""
|
|
|
|
url = "%s/%s" % (str(container), str(object_name))
|
|
resp, body = self.head(url)
|
|
return resp, body
|
|
|
|
def get_object(self, container, object_name):
|
|
"""Retrieve object's data."""
|
|
|
|
url = "{0}/{1}".format(container, object_name)
|
|
resp, body = self.get(url)
|
|
return resp, body
|
|
|
|
def copy_object_in_same_container(self, container, src_object_name,
|
|
dest_object_name, metadata=None):
|
|
"""Copy storage object's data to the new object using PUT."""
|
|
|
|
url = "{0}/{1}".format(container, dest_object_name)
|
|
headers = {}
|
|
headers['X-Copy-From'] = "%s/%s" % (str(container),
|
|
str(src_object_name))
|
|
headers['content-length'] = '0'
|
|
if metadata:
|
|
for key in metadata:
|
|
headers[str(key)] = metadata[key]
|
|
|
|
resp, body = self.put(url, None, headers=headers)
|
|
return resp, body
|
|
|
|
def copy_object_across_containers(self, src_container, src_object_name,
|
|
dst_container, dst_object_name,
|
|
metadata=None):
|
|
"""Copy storage object's data to the new object using PUT."""
|
|
|
|
url = "{0}/{1}".format(dst_container, dst_object_name)
|
|
headers = {}
|
|
headers['X-Copy-From'] = "%s/%s" % (str(src_container),
|
|
str(src_object_name))
|
|
headers['content-length'] = '0'
|
|
if metadata:
|
|
for key in metadata:
|
|
headers[str(key)] = metadata[key]
|
|
|
|
resp, body = self.put(url, None, headers=headers)
|
|
return resp, body
|
|
|
|
def copy_object_2d_way(self, container, src_object_name, dest_object_name,
|
|
metadata=None):
|
|
"""Copy storage object's data to the new object using COPY."""
|
|
|
|
url = "{0}/{1}".format(container, src_object_name)
|
|
headers = {}
|
|
headers['Destination'] = "%s/%s" % (str(container),
|
|
str(dest_object_name))
|
|
if metadata:
|
|
for key in metadata:
|
|
headers[str(key)] = metadata[key]
|
|
|
|
resp, body = self.copy(url, headers=headers)
|
|
return resp, body
|
|
|
|
def get_object_using_temp_url(self, container, object_name, expires, key):
|
|
"""Retrieve object's data using temp URL."""
|
|
|
|
self._set_auth()
|
|
method = 'GET'
|
|
path = "%s/%s/%s" % (urlparse.urlparse(self.base_url).path, container,
|
|
object_name)
|
|
hmac_body = '%s\n%s\n%s' % (method, expires, path)
|
|
sig = hmac.new(key, hmac_body, hashlib.sha1).hexdigest()
|
|
|
|
url = "%s/%s?temp_url_sig=%s&temp_url_expires=%s" % (container,
|
|
object_name,
|
|
sig, expires)
|
|
|
|
resp, body = self.get(url)
|
|
return resp, body
|
|
|
|
def create_object_segments(self, container, object_name, segment, data):
|
|
"""Creates object segments."""
|
|
url = "{0}/{1}/{2}".format(container, object_name, segment)
|
|
resp, body = self.put(url, data, self.headers)
|
|
return resp, body
|
|
|
|
|
|
class ObjectClientCustomizedHeader(RestClient):
|
|
|
|
def __init__(self, config, username, password, auth_url, tenant_name=None):
|
|
super(ObjectClientCustomizedHeader, self).__init__(config, username,
|
|
password, auth_url,
|
|
tenant_name)
|
|
# Overwrites json-specific header encoding in RestClient
|
|
self.service = self.config.object_storage.catalog_type
|
|
self.format = 'json'
|
|
|
|
def request(self, method, url, headers=None, body=None):
|
|
"""A simple HTTP request interface."""
|
|
dscv = self.config.identity.disable_ssl_certificate_validation
|
|
self.http_obj = http.ClosingHttp(
|
|
disable_ssl_certificate_validation=dscv)
|
|
if headers is None:
|
|
headers = {}
|
|
if self.base_url is None:
|
|
self._set_auth()
|
|
|
|
req_url = "%s/%s" % (self.base_url, url)
|
|
self._log_request(method, req_url, headers, body)
|
|
resp, resp_body = self.http_obj.request(req_url, method,
|
|
headers=headers, body=body)
|
|
self._log_response(resp, resp_body)
|
|
if resp.status == 401 or resp.status == 403:
|
|
raise exceptions.Unauthorized()
|
|
|
|
return resp, resp_body
|
|
|
|
def get_object(self, container, object_name, metadata=None):
|
|
"""Retrieve object's data."""
|
|
headers = {}
|
|
if metadata:
|
|
for key in metadata:
|
|
headers[str(key)] = metadata[key]
|
|
|
|
url = "{0}/{1}".format(container, object_name)
|
|
resp, body = self.get(url, headers=headers)
|
|
return resp, body
|
|
|
|
def create_object(self, container, object_name, data, metadata=None):
|
|
"""Create storage object."""
|
|
|
|
headers = {}
|
|
if metadata:
|
|
for key in metadata:
|
|
headers[str(key)] = metadata[key]
|
|
|
|
if not data:
|
|
headers['content-length'] = '0'
|
|
url = "%s/%s" % (str(container), str(object_name))
|
|
resp, body = self.put(url, data, headers=headers)
|
|
return resp, body
|
|
|
|
def delete_object(self, container, object_name, metadata=None):
|
|
"""Delete storage object."""
|
|
|
|
headers = {}
|
|
if metadata:
|
|
for key in metadata:
|
|
headers[str(key)] = metadata[key]
|
|
|
|
url = "%s/%s" % (str(container), str(object_name))
|
|
resp, body = self.delete(url, headers=headers)
|
|
return resp, body
|