# 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, with_data=True) 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), ])