horizon/openstack_dashboard/test/unit/api/test_swift.py
Hervé Beraud 17d8ab6103 Stop to use the __future__ module.
The __future__ module [1] was used in this context to ensure compatibility
between python 2 and python 3.

We previously dropped the support of python 2.7 [2] and now we only support
python 3 so we don't need to continue to use this module and the imports
listed below.

Imports commonly used and their related PEPs:
- `division` is related to PEP 238 [3]
- `print_function` is related to PEP 3105 [4]
- `unicode_literals` is related to PEP 3112 [5]
- `with_statement` is related to PEP 343 [6]
- `absolute_import` is related to PEP 328 [7]

[1] https://docs.python.org/3/library/__future__.html
[2] https://governance.openstack.org/tc/goals/selected/ussuri/drop-py27.html
[3] https://www.python.org/dev/peps/pep-0238
[4] https://www.python.org/dev/peps/pep-3105
[5] https://www.python.org/dev/peps/pep-3112
[6] https://www.python.org/dev/peps/pep-0343
[7] https://www.python.org/dev/peps/pep-0328

Change-Id: Ic03754dcaaa4f1c0018294aa52bb05d956aa5d10
2020-06-03 10:45:18 +02:00

303 lines
12 KiB
Python

# Copyright 2012 United States Government as represented by the
# Administrator of the National Aeronautics and Space Administration.
# All Rights Reserved.
#
# Copyright 2012 Nebula, Inc.
#
# 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 unittest import mock
from horizon import exceptions
from openstack_dashboard import api
from openstack_dashboard.test import helpers as test
@mock.patch('swiftclient.client.Connection')
class SwiftApiTests(test.APIMockTestCase):
def test_swift_get_containers(self, mock_swiftclient):
containers = self.containers.list()
cont_data = [c._apidict for c in containers]
swift_api = mock_swiftclient.return_value
swift_api.get_account.return_value = [{}, cont_data]
(conts, more) = api.swift.swift_get_containers(self.request)
self.assertEqual(len(containers), len(conts))
self.assertFalse(more)
swift_api.get_account.assert_called_once_with(
limit=1001, marker=None, prefix=None, full_listing=True)
def test_swift_get_container_with_data(self, mock_swiftclient):
container = self.containers.first()
objects = self.objects.list()
swift_api = mock_swiftclient.return_value
swift_api.get_object.return_value = (container, objects)
cont = api.swift.swift_get_container(self.request, container.name)
self.assertEqual(container.name, cont.name)
self.assertEqual(len(objects), len(cont.data))
swift_api.get_object.assert_called_once_with(container.name, "")
def test_swift_get_container_without_data(self, mock_swiftclient):
container = self.containers.first()
swift_api = mock_swiftclient.return_value
swift_api.head_container.return_value = container
cont = api.swift.swift_get_container(self.request,
container.name,
with_data=False)
self.assertEqual(cont.name, container.name)
self.assertIsNone(cont.data)
swift_api.head_container.assert_called_once_with(container.name)
def test_swift_create_duplicate_container(self, mock_swiftclient):
metadata = {'is_public': False}
container = self.containers.first()
headers = api.swift._metadata_to_header(metadata=(metadata))
swift_api = mock_swiftclient.return_value
# Check for existence, then create
swift_api.head_container.side_effect = self.exceptions.swift
swift_api.put_container.return_value = container
api.swift.swift_create_container(self.request,
container.name,
metadata=(metadata))
swift_api.head_container.assert_called_once_with(container.name)
swift_api.put_container.assert_called_once_with(container.name,
headers=headers)
def test_swift_create_container(self, mock_swiftclient):
metadata = {'is_public': True}
container = self.containers.first()
swift_api = mock_swiftclient.return_value
swift_api.head_container.return_value = container
with self.assertRaises(exceptions.AlreadyExists):
api.swift.swift_create_container(self.request,
container.name,
metadata=(metadata))
swift_api.head_container.assert_called_once_with(container.name)
def test_swift_create_container_with_storage_policy(self, mock_swiftclient):
metadata = {'is_public': True, 'storage_policy': 'nz-o1-mr-r3'}
container = self.containers.first()
swift_api = mock_swiftclient.return_value
swift_api.head_container.return_value = container
headers = api.swift._metadata_to_header(metadata=(metadata))
self.assertEqual(headers["x-storage-policy"], 'nz-o1-mr-r3')
with self.assertRaises(exceptions.AlreadyExists):
api.swift.swift_create_container(self.request,
container.name,
metadata=(metadata))
swift_api.head_container.assert_called_once_with(container.name)
def test_metadata_to_headers(self, mock_swiftclient):
metadata = {'is_public': True, 'storage_policy': 'nz-o1-mr-r3'}
headers = api.swift._metadata_to_header(metadata=(metadata))
self.assertEqual(headers["x-storage-policy"], 'nz-o1-mr-r3')
self.assertEqual(headers["x-container-read"], '.r:*,.rlistings')
def test_metadata_to_headers_without_metadata(self, mock_swiftclient):
metadata = {}
headers = api.swift._metadata_to_header(metadata=(metadata))
self.assertNotIn("x-storage-policy", headers)
self.assertNotIn("x-container-read", headers)
def test_swift_update_container(self, mock_swiftclient):
metadata = {'is_public': True}
container = self.containers.first()
swift_api = mock_swiftclient.return_value
headers = api.swift._metadata_to_header(metadata=(metadata))
swift_api.post_container.return_value = container
api.swift.swift_update_container(self.request,
container.name,
metadata=(metadata))
swift_api.post_container.assert_called_once_with(container.name,
headers=headers)
def test_swift_get_objects(self, mock_swiftclient):
container = self.containers.first()
objects = self.objects.list()
swift_api = mock_swiftclient.return_value
swift_api.get_container.return_value = [{}, objects]
(objs, more) = api.swift.swift_get_objects(self.request,
container.name)
self.assertEqual(len(objects), len(objs))
self.assertFalse(more)
swift_api.get_container.assert_called_once_with(
container.name,
limit=1001,
marker=None,
prefix=None,
delimiter='/',
full_listing=True)
def test_swift_get_object_with_data_non_chunked(self, mock_swiftclient):
container = self.containers.first()
object = self.objects.first()
swift_api = mock_swiftclient.return_value
swift_api.get_object.return_value = [object, object.data]
obj = api.swift.swift_get_object(self.request, container.name,
object.name, resp_chunk_size=None)
self.assertEqual(object.name, obj.name)
swift_api.get_object.assert_called_once_with(
container.name, object.name, resp_chunk_size=None)
def test_swift_get_object_with_data_chunked(self, mock_swiftclient):
container = self.containers.first()
object = self.objects.first()
swift_api = mock_swiftclient.return_value
swift_api.get_object.return_value = [object, object.data]
obj = api.swift.swift_get_object(
self.request, container.name, object.name)
self.assertEqual(object.name, obj.name)
swift_api.get_object.assert_called_once_with(
container.name, object.name, resp_chunk_size=api.swift.CHUNK_SIZE)
def test_swift_get_object_without_data(self, mock_swiftclient):
container = self.containers.first()
object = self.objects.first()
swift_api = mock_swiftclient.return_value
swift_api.head_object.return_value = object
obj = api.swift.swift_get_object(self.request,
container.name,
object.name,
with_data=False)
self.assertEqual(object.name, obj.name)
self.assertIsNone(obj.data)
swift_api.head_object.assert_called_once_with(container.name,
object.name)
def test_swift_create_pseudo_folder(self, mock_swiftclient):
container = self.containers.first()
folder = self.folder.first()
swift_api = mock_swiftclient.return_value
exc = self.exceptions.swift
swift_api.head_object.side_effect = exc
swift_api.put_object.return_value = folder
api.swift.swift_create_pseudo_folder(self.request,
container.name,
folder.name)
swift_api.head_object.assert_called_once_with(container.name,
folder.name)
swift_api.put_object.assert_called_once_with(container.name,
folder.name,
None,
headers={})
def test_swift_create_duplicate_folder(self, mock_swiftclient):
container = self.containers.first()
folder = self.folder.first()
swift_api = mock_swiftclient.return_value
swift_api.head_object.return_value = folder
with self.assertRaises(exceptions.AlreadyExists):
api.swift.swift_create_pseudo_folder(self.request,
container.name,
folder.name)
swift_api.head_object.assert_called_once_with(container.name,
folder.name)
def test_swift_upload_object(self, mock_swiftclient):
container = self.containers.first()
obj = self.objects.first()
fake_name = 'fake_object.jpg'
class FakeFile(object):
def __init__(self):
self.name = fake_name
self.data = obj.data
self.size = len(obj.data)
headers = {'X-Object-Meta-Orig-Filename': fake_name}
swift_api = mock_swiftclient.return_value
test_file = FakeFile()
swift_api.put_object.return_value = None
api.swift.swift_upload_object(self.request,
container.name,
obj.name,
test_file)
swift_api.put_object.assert_called_once_with(
container.name,
obj.name,
test.IsA(FakeFile),
content_length=test_file.size,
headers=headers)
def test_swift_upload_object_without_file(self, mock_swiftclient):
container = self.containers.first()
obj = self.objects.first()
swift_api = mock_swiftclient.return_value
swift_api.put_object.return_value = None
response = api.swift.swift_upload_object(self.request,
container.name,
obj.name,
None)
self.assertEqual(0, response['bytes'])
swift_api.put_object.assert_called_once_with(
container.name,
obj.name,
None,
content_length=0,
headers={})
def test_swift_object_exists(self, mock_swiftclient):
container = self.containers.first()
obj = self.objects.first()
swift_api = mock_swiftclient.return_value
swift_api.head_object.side_effect = [container, self.exceptions.swift]
args = self.request, container.name, obj.name
self.assertTrue(api.swift.swift_object_exists(*args))
# Again, for a "non-existent" object
self.assertFalse(api.swift.swift_object_exists(*args))
self.assertEqual(2, swift_api.head_object.call_count)
swift_api.head_object.assert_has_calls([
mock.call(container.name, obj.name),
mock.call(container.name, obj.name),
])