165 lines
7.8 KiB
Python
165 lines
7.8 KiB
Python
# Copyright 2013 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# 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 mock
|
|
import six
|
|
from six.moves import builtins as __builtin__
|
|
from six.moves import http_client
|
|
from swiftclient import client as swift_client
|
|
from swiftclient import exceptions as swift_exception
|
|
from swiftclient import utils as swift_utils
|
|
|
|
from ironic.common import exception
|
|
from ironic.common import swift
|
|
from ironic.tests import base
|
|
|
|
|
|
if six.PY3:
|
|
import io
|
|
file = io.BytesIO
|
|
|
|
|
|
@mock.patch.object(swift, '_get_swift_session')
|
|
@mock.patch.object(swift_client, 'Connection', autospec=True)
|
|
class SwiftTestCase(base.TestCase):
|
|
|
|
def setUp(self):
|
|
super(SwiftTestCase, self).setUp()
|
|
self.swift_exception = swift_exception.ClientException('', '')
|
|
|
|
def test___init__(self, connection_mock, keystone_mock):
|
|
sess = mock.Mock()
|
|
sess.get_endpoint.return_value = 'http://swift:8080'
|
|
sess.get_token.return_value = 'fake_token'
|
|
sess.verify = '/path/to/ca/file'
|
|
keystone_mock.return_value = sess
|
|
swift.SwiftAPI()
|
|
params = {'retries': 2,
|
|
'preauthurl': 'http://swift:8080',
|
|
'preauthtoken': 'fake_token',
|
|
'insecure': False,
|
|
'cacert': '/path/to/ca/file'}
|
|
connection_mock.assert_called_once_with(**params)
|
|
|
|
@mock.patch.object(__builtin__, 'open', autospec=True)
|
|
def test_create_object(self, open_mock, connection_mock, keystone_mock):
|
|
swiftapi = swift.SwiftAPI()
|
|
connection_obj_mock = connection_mock.return_value
|
|
mock_file_handle = mock.MagicMock(spec=file)
|
|
mock_file_handle.__enter__.return_value = 'file-object'
|
|
open_mock.return_value = mock_file_handle
|
|
|
|
connection_obj_mock.put_object.return_value = 'object-uuid'
|
|
|
|
object_uuid = swiftapi.create_object('container', 'object',
|
|
'some-file-location')
|
|
|
|
connection_obj_mock.put_container.assert_called_once_with('container')
|
|
connection_obj_mock.put_object.assert_called_once_with(
|
|
'container', 'object', 'file-object', headers=None)
|
|
self.assertEqual('object-uuid', object_uuid)
|
|
|
|
@mock.patch.object(__builtin__, 'open', autospec=True)
|
|
def test_create_object_create_container_fails(self, open_mock,
|
|
connection_mock,
|
|
keystone_mock):
|
|
swiftapi = swift.SwiftAPI()
|
|
connection_obj_mock = connection_mock.return_value
|
|
connection_obj_mock.put_container.side_effect = self.swift_exception
|
|
self.assertRaises(exception.SwiftOperationError,
|
|
swiftapi.create_object, 'container',
|
|
'object', 'some-file-location')
|
|
connection_obj_mock.put_container.assert_called_once_with('container')
|
|
self.assertFalse(connection_obj_mock.put_object.called)
|
|
|
|
@mock.patch.object(__builtin__, 'open', autospec=True)
|
|
def test_create_object_put_object_fails(self, open_mock, connection_mock,
|
|
keystone_mock):
|
|
swiftapi = swift.SwiftAPI()
|
|
mock_file_handle = mock.MagicMock(spec=file)
|
|
mock_file_handle.__enter__.return_value = 'file-object'
|
|
open_mock.return_value = mock_file_handle
|
|
connection_obj_mock = connection_mock.return_value
|
|
connection_obj_mock.head_account.side_effect = None
|
|
connection_obj_mock.put_object.side_effect = self.swift_exception
|
|
self.assertRaises(exception.SwiftOperationError,
|
|
swiftapi.create_object, 'container',
|
|
'object', 'some-file-location')
|
|
connection_obj_mock.put_container.assert_called_once_with('container')
|
|
connection_obj_mock.put_object.assert_called_once_with(
|
|
'container', 'object', 'file-object', headers=None)
|
|
|
|
@mock.patch.object(swift_utils, 'generate_temp_url', autospec=True)
|
|
def test_get_temp_url(self, gen_temp_url_mock, connection_mock,
|
|
keystone_mock):
|
|
swiftapi = swift.SwiftAPI()
|
|
connection_obj_mock = connection_mock.return_value
|
|
connection_obj_mock.url = 'http://host/v1/AUTH_tenant_id'
|
|
head_ret_val = {'x-account-meta-temp-url-key': 'secretkey'}
|
|
connection_obj_mock.head_account.return_value = head_ret_val
|
|
gen_temp_url_mock.return_value = 'temp-url-path'
|
|
temp_url_returned = swiftapi.get_temp_url('container', 'object', 10)
|
|
connection_obj_mock.head_account.assert_called_once_with()
|
|
object_path_expected = '/v1/AUTH_tenant_id/container/object'
|
|
gen_temp_url_mock.assert_called_once_with(object_path_expected, 10,
|
|
'secretkey', 'GET')
|
|
self.assertEqual('http://host/temp-url-path', temp_url_returned)
|
|
|
|
def test_delete_object(self, connection_mock, keystone_mock):
|
|
swiftapi = swift.SwiftAPI()
|
|
connection_obj_mock = connection_mock.return_value
|
|
swiftapi.delete_object('container', 'object')
|
|
connection_obj_mock.delete_object.assert_called_once_with('container',
|
|
'object')
|
|
|
|
def test_delete_object_exc_resource_not_found(self, connection_mock,
|
|
keystone_mock):
|
|
swiftapi = swift.SwiftAPI()
|
|
exc = swift_exception.ClientException(
|
|
"Resource not found", http_status=http_client.NOT_FOUND)
|
|
connection_obj_mock = connection_mock.return_value
|
|
connection_obj_mock.delete_object.side_effect = exc
|
|
self.assertRaises(exception.SwiftObjectNotFoundError,
|
|
swiftapi.delete_object, 'container', 'object')
|
|
connection_obj_mock.delete_object.assert_called_once_with('container',
|
|
'object')
|
|
|
|
def test_delete_object_exc(self, connection_mock, keystone_mock):
|
|
swiftapi = swift.SwiftAPI()
|
|
exc = swift_exception.ClientException("Operation error")
|
|
connection_obj_mock = connection_mock.return_value
|
|
connection_obj_mock.delete_object.side_effect = exc
|
|
self.assertRaises(exception.SwiftOperationError,
|
|
swiftapi.delete_object, 'container', 'object')
|
|
connection_obj_mock.delete_object.assert_called_once_with('container',
|
|
'object')
|
|
|
|
def test_head_object(self, connection_mock, keystone_mock):
|
|
swiftapi = swift.SwiftAPI()
|
|
connection_obj_mock = connection_mock.return_value
|
|
expected_head_result = {'a': 'b'}
|
|
connection_obj_mock.head_object.return_value = expected_head_result
|
|
actual_head_result = swiftapi.head_object('container', 'object')
|
|
connection_obj_mock.head_object.assert_called_once_with('container',
|
|
'object')
|
|
self.assertEqual(expected_head_result, actual_head_result)
|
|
|
|
def test_update_object_meta(self, connection_mock, keystone_mock):
|
|
swiftapi = swift.SwiftAPI()
|
|
connection_obj_mock = connection_mock.return_value
|
|
headers = {'a': 'b'}
|
|
swiftapi.update_object_meta('container', 'object', headers)
|
|
connection_obj_mock.post_object.assert_called_once_with(
|
|
'container', 'object', headers)
|