88bb1af1d5
Clean up imports due to F401 and F403 checks. F401 'module' imported but unused F403 unable to detect undefined names with wildcard import Change-Id: I487edb157de1a6babc7ad8a3fb65f195e476c490
645 lines
26 KiB
Python
645 lines
26 KiB
Python
# Copyright 2012 OpenStack Foundation.
|
|
# Copyright 2013 IBM Corp.
|
|
# 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 oslo.config import cfg
|
|
|
|
from glance.common import crypt
|
|
from glance.common import exception
|
|
import glance.context
|
|
import glance.db
|
|
from glance.openstack.common import uuidutils
|
|
import glance.tests.unit.utils as unit_test_utils
|
|
import glance.tests.utils as test_utils
|
|
|
|
|
|
CONF = cfg.CONF
|
|
CONF.import_opt('metadata_encryption_key', 'glance.common.config')
|
|
|
|
UUID1 = 'c80a1a6c-bd1f-41c5-90ee-81afedb1d58d'
|
|
UUID2 = 'a85abd86-55b3-4d5b-b0b4-5d0a6e6042fc'
|
|
UUID3 = '971ec09a-8067-4bc8-a91f-ae3557f1c4c7'
|
|
UUID4 = '6bbe7cc2-eae7-4c0f-b50d-a7160b0c6a86'
|
|
|
|
TENANT1 = '6838eb7b-6ded-434a-882c-b344c77fe8df'
|
|
TENANT2 = '2c014f32-55eb-467d-8fcb-4bd706012f81'
|
|
TENANT3 = '5a3e60e8-cfa9-4a9e-a90a-62b42cea92b8'
|
|
TENANT4 = 'c6c87f25-8a94-47ed-8c83-053c25f42df4'
|
|
|
|
USER1 = '54492ba0-f4df-4e4e-be62-27f4d76b29cf'
|
|
|
|
UUID1_LOCATION = 'file:///path/to/image'
|
|
UUID1_LOCATION_METADATA = {'key': 'value'}
|
|
UUID3_LOCATION = 'http://somehost.com/place'
|
|
|
|
CHECKSUM = '93264c3edf5972c9f1cb309543d38a5c'
|
|
CHCKSUM1 = '43264c3edf4972c9f1cb309543d38a55'
|
|
|
|
|
|
def _db_fixture(id, **kwargs):
|
|
obj = {
|
|
'id': id,
|
|
'name': None,
|
|
'is_public': False,
|
|
'properties': {},
|
|
'checksum': None,
|
|
'owner': None,
|
|
'status': 'queued',
|
|
'tags': [],
|
|
'size': None,
|
|
'locations': [],
|
|
'protected': False,
|
|
'disk_format': None,
|
|
'container_format': None,
|
|
'deleted': False,
|
|
'min_ram': None,
|
|
'min_disk': None,
|
|
}
|
|
obj.update(kwargs)
|
|
return obj
|
|
|
|
|
|
def _db_image_member_fixture(image_id, member_id, **kwargs):
|
|
obj = {
|
|
'image_id': image_id,
|
|
'member': member_id,
|
|
}
|
|
obj.update(kwargs)
|
|
return obj
|
|
|
|
|
|
def _db_task_fixture(task_id, type, status, **kwargs):
|
|
obj = {
|
|
'id': task_id,
|
|
'type': type,
|
|
'status': status,
|
|
'input': None,
|
|
'result': None,
|
|
'owner': None,
|
|
'message': None,
|
|
'deleted': False,
|
|
}
|
|
obj.update(kwargs)
|
|
return obj
|
|
|
|
|
|
class TestImageRepo(test_utils.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestImageRepo, self).setUp()
|
|
self.db = unit_test_utils.FakeDB()
|
|
self.db.reset()
|
|
self.context = glance.context.RequestContext(
|
|
user=USER1, tenant=TENANT1)
|
|
self.image_repo = glance.db.ImageRepo(self.context, self.db)
|
|
self.image_factory = glance.domain.ImageFactory()
|
|
self._create_images()
|
|
self._create_image_members()
|
|
|
|
def _create_images(self):
|
|
self.db.reset()
|
|
self.images = [
|
|
_db_fixture(UUID1, owner=TENANT1, checksum=CHECKSUM,
|
|
name='1', size=256,
|
|
is_public=True, status='active',
|
|
locations=[{'url': UUID1_LOCATION,
|
|
'metadata': UUID1_LOCATION_METADATA}]),
|
|
_db_fixture(UUID2, owner=TENANT1, checksum=CHCKSUM1,
|
|
name='2', size=512, is_public=False),
|
|
_db_fixture(UUID3, owner=TENANT3, checksum=CHCKSUM1,
|
|
name='3', size=1024, is_public=True,
|
|
locations=[{'url': UUID3_LOCATION,
|
|
'metadata': {}}]),
|
|
_db_fixture(UUID4, owner=TENANT4, name='4', size=2048),
|
|
]
|
|
[self.db.image_create(None, image) for image in self.images]
|
|
|
|
self.db.image_tag_set_all(None, UUID1, ['ping', 'pong'])
|
|
|
|
def _create_image_members(self):
|
|
self.image_members = [
|
|
_db_image_member_fixture(UUID2, TENANT2),
|
|
_db_image_member_fixture(UUID2, TENANT3, status='accepted'),
|
|
]
|
|
[self.db.image_member_create(None, image_member)
|
|
for image_member in self.image_members]
|
|
|
|
def test_get(self):
|
|
image = self.image_repo.get(UUID1)
|
|
self.assertEqual(image.image_id, UUID1)
|
|
self.assertEqual(image.name, '1')
|
|
self.assertEqual(image.tags, set(['ping', 'pong']))
|
|
self.assertEqual(image.visibility, 'public')
|
|
self.assertEqual(image.status, 'active')
|
|
self.assertEqual(image.size, 256)
|
|
self.assertEqual(image.owner, TENANT1)
|
|
|
|
def test_location_value(self):
|
|
image = self.image_repo.get(UUID3)
|
|
self.assertEqual(image.locations[0]['url'], UUID3_LOCATION)
|
|
|
|
def test_location_data_value(self):
|
|
image = self.image_repo.get(UUID1)
|
|
self.assertEqual(image.locations[0]['url'], UUID1_LOCATION)
|
|
self.assertEqual(image.locations[0]['metadata'],
|
|
UUID1_LOCATION_METADATA)
|
|
|
|
def test_location_data_exists(self):
|
|
image = self.image_repo.get(UUID2)
|
|
self.assertEqual(image.locations, [])
|
|
|
|
def test_get_not_found(self):
|
|
fake_uuid = uuidutils.generate_uuid()
|
|
exc = self.assertRaises(exception.NotFound, self.image_repo.get,
|
|
fake_uuid)
|
|
self.assertTrue(fake_uuid in unicode(exc))
|
|
|
|
def test_get_forbidden(self):
|
|
self.assertRaises(exception.NotFound, self.image_repo.get, UUID4)
|
|
|
|
def test_list(self):
|
|
images = self.image_repo.list()
|
|
image_ids = set([i.image_id for i in images])
|
|
self.assertEqual(set([UUID1, UUID2, UUID3]), image_ids)
|
|
|
|
def _do_test_list_status(self, status, expected):
|
|
self.context = glance.context.RequestContext(
|
|
user=USER1, tenant=TENANT3)
|
|
self.image_repo = glance.db.ImageRepo(self.context, self.db)
|
|
images = self.image_repo.list(member_status=status)
|
|
self.assertEqual(expected, len(images))
|
|
|
|
def test_list_status(self):
|
|
self._do_test_list_status(None, 3)
|
|
|
|
def test_list_status_pending(self):
|
|
self._do_test_list_status('pending', 2)
|
|
|
|
def test_list_status_rejected(self):
|
|
self._do_test_list_status('rejected', 2)
|
|
|
|
def test_list_status_all(self):
|
|
self._do_test_list_status('all', 3)
|
|
|
|
def test_list_with_marker(self):
|
|
full_images = self.image_repo.list()
|
|
full_ids = [i.image_id for i in full_images]
|
|
marked_images = self.image_repo.list(marker=full_ids[0])
|
|
actual_ids = [i.image_id for i in marked_images]
|
|
self.assertEqual(actual_ids, full_ids[1:])
|
|
|
|
def test_list_with_last_marker(self):
|
|
images = self.image_repo.list()
|
|
marked_images = self.image_repo.list(marker=images[-1].image_id)
|
|
self.assertEqual(len(marked_images), 0)
|
|
|
|
def test_limited_list(self):
|
|
limited_images = self.image_repo.list(limit=2)
|
|
self.assertEqual(len(limited_images), 2)
|
|
|
|
def test_list_with_marker_and_limit(self):
|
|
full_images = self.image_repo.list()
|
|
full_ids = [i.image_id for i in full_images]
|
|
marked_images = self.image_repo.list(marker=full_ids[0], limit=1)
|
|
actual_ids = [i.image_id for i in marked_images]
|
|
self.assertEqual(actual_ids, full_ids[1:2])
|
|
|
|
def test_list_private_images(self):
|
|
filters = {'visibility': 'private'}
|
|
images = self.image_repo.list(filters=filters)
|
|
image_ids = set([i.image_id for i in images])
|
|
self.assertEqual(set([UUID2]), image_ids)
|
|
|
|
def test_list_with_checksum_filter_single_image(self):
|
|
filters = {'checksum': CHECKSUM}
|
|
images = self.image_repo.list(filters=filters)
|
|
image_ids = list([i.image_id for i in images])
|
|
self.assertEqual(1, len(image_ids))
|
|
self.assertEqual([UUID1], image_ids)
|
|
|
|
def test_list_with_checksum_filter_multiple_images(self):
|
|
filters = {'checksum': CHCKSUM1}
|
|
images = self.image_repo.list(filters=filters)
|
|
image_ids = list([i.image_id for i in images])
|
|
self.assertEqual(2, len(image_ids))
|
|
self.assertEqual([UUID3, UUID2], image_ids)
|
|
|
|
def test_list_with_wrong_checksum(self):
|
|
WRONG_CHKSUM = 'd2fd42f979e1ed1aafadc7eb9354bff839c858cd'
|
|
filters = {'checksum': WRONG_CHKSUM}
|
|
images = self.image_repo.list(filters=filters)
|
|
self.assertEqual(0, len(images))
|
|
|
|
def test_list_with_tags_filter_single_tag(self):
|
|
filters = {'tags': ['ping']}
|
|
images = self.image_repo.list(filters=filters)
|
|
image_ids = list([i.image_id for i in images])
|
|
self.assertEqual(1, len(image_ids))
|
|
self.assertEqual([UUID1], image_ids)
|
|
|
|
def test_list_with_tags_filter_multiple_tags(self):
|
|
filters = {'tags': ['ping', 'pong']}
|
|
images = self.image_repo.list(filters=filters)
|
|
image_ids = list([i.image_id for i in images])
|
|
self.assertEqual(1, len(image_ids))
|
|
self.assertEqual([UUID1], image_ids)
|
|
|
|
def test_list_with_tags_filter_multiple_tags_and_nonexistent(self):
|
|
filters = {'tags': ['ping', 'fake']}
|
|
images = self.image_repo.list(filters=filters)
|
|
image_ids = list([i.image_id for i in images])
|
|
self.assertEqual(0, len(image_ids))
|
|
|
|
def test_list_with_wrong_tags(self):
|
|
filters = {'tags': ['fake']}
|
|
images = self.image_repo.list(filters=filters)
|
|
self.assertEqual(0, len(images))
|
|
|
|
def test_list_public_images(self):
|
|
filters = {'visibility': 'public'}
|
|
images = self.image_repo.list(filters=filters)
|
|
image_ids = set([i.image_id for i in images])
|
|
self.assertEqual(set([UUID1, UUID3]), image_ids)
|
|
|
|
def test_sorted_list(self):
|
|
images = self.image_repo.list(sort_key='size', sort_dir='asc')
|
|
image_ids = [i.image_id for i in images]
|
|
self.assertEqual([UUID1, UUID2, UUID3], image_ids)
|
|
|
|
def test_add_image(self):
|
|
image = self.image_factory.new_image(name='added image')
|
|
self.assertEqual(image.updated_at, image.created_at)
|
|
self.image_repo.add(image)
|
|
retreived_image = self.image_repo.get(image.image_id)
|
|
self.assertEqual(retreived_image.name, 'added image')
|
|
self.assertEqual(retreived_image.updated_at, image.updated_at)
|
|
|
|
def test_save_image(self):
|
|
image = self.image_repo.get(UUID1)
|
|
original_update_time = image.updated_at
|
|
image.name = 'foo'
|
|
image.tags = ['king', 'kong']
|
|
self.image_repo.save(image)
|
|
current_update_time = image.updated_at
|
|
self.assertTrue(current_update_time > original_update_time)
|
|
image = self.image_repo.get(UUID1)
|
|
self.assertEqual(image.name, 'foo')
|
|
self.assertEqual(image.tags, set(['king', 'kong']))
|
|
self.assertEqual(image.updated_at, current_update_time)
|
|
|
|
def test_save_image_not_found(self):
|
|
fake_uuid = uuidutils.generate_uuid()
|
|
image = self.image_repo.get(UUID1)
|
|
image.image_id = fake_uuid
|
|
exc = self.assertRaises(exception.NotFound, self.image_repo.save,
|
|
image)
|
|
self.assertTrue(fake_uuid in unicode(exc))
|
|
|
|
def test_remove_image(self):
|
|
image = self.image_repo.get(UUID1)
|
|
previous_update_time = image.updated_at
|
|
self.image_repo.remove(image)
|
|
self.assertTrue(image.updated_at > previous_update_time)
|
|
self.assertRaises(exception.NotFound, self.image_repo.get, UUID1)
|
|
|
|
def test_remove_image_not_found(self):
|
|
fake_uuid = uuidutils.generate_uuid()
|
|
image = self.image_repo.get(UUID1)
|
|
image.image_id = fake_uuid
|
|
exc = self.assertRaises(exception.NotFound, self.image_repo.remove,
|
|
image)
|
|
self.assertTrue(fake_uuid in unicode(exc))
|
|
|
|
|
|
class TestEncryptedLocations(test_utils.BaseTestCase):
|
|
def setUp(self):
|
|
super(TestEncryptedLocations, self).setUp()
|
|
self.db = unit_test_utils.FakeDB()
|
|
self.db.reset()
|
|
self.context = glance.context.RequestContext(
|
|
user=USER1, tenant=TENANT1)
|
|
self.image_repo = glance.db.ImageRepo(self.context, self.db)
|
|
self.image_factory = glance.domain.ImageFactory()
|
|
self.crypt_key = '0123456789abcdef'
|
|
self.config(metadata_encryption_key=self.crypt_key)
|
|
self.foo_bar_location = [{'url': 'foo', 'metadata': {}},
|
|
{'url': 'bar', 'metadata': {}}]
|
|
|
|
def test_encrypt_locations_on_add(self):
|
|
image = self.image_factory.new_image(UUID1)
|
|
image.locations = self.foo_bar_location
|
|
self.image_repo.add(image)
|
|
db_data = self.db.image_get(self.context, UUID1)
|
|
self.assertNotEqual(db_data['locations'], ['foo', 'bar'])
|
|
decrypted_locations = [crypt.urlsafe_decrypt(self.crypt_key, l['url'])
|
|
for l in db_data['locations']]
|
|
self.assertEqual(decrypted_locations,
|
|
[l['url'] for l in self.foo_bar_location])
|
|
|
|
def test_encrypt_locations_on_save(self):
|
|
image = self.image_factory.new_image(UUID1)
|
|
self.image_repo.add(image)
|
|
image.locations = self.foo_bar_location
|
|
self.image_repo.save(image)
|
|
db_data = self.db.image_get(self.context, UUID1)
|
|
self.assertNotEqual(db_data['locations'], ['foo', 'bar'])
|
|
decrypted_locations = [crypt.urlsafe_decrypt(self.crypt_key, l['url'])
|
|
for l in db_data['locations']]
|
|
self.assertEqual(decrypted_locations,
|
|
[l['url'] for l in self.foo_bar_location])
|
|
|
|
def test_decrypt_locations_on_get(self):
|
|
url_loc = ['ping', 'pong']
|
|
orig_locations = [{'url': l, 'metadata': {}} for l in url_loc]
|
|
encrypted_locs = [crypt.urlsafe_encrypt(self.crypt_key, l)
|
|
for l in url_loc]
|
|
encrypted_locations = [{'url': l, 'metadata': {}}
|
|
for l in encrypted_locs]
|
|
self.assertNotEqual(encrypted_locations, orig_locations)
|
|
db_data = _db_fixture(UUID1, owner=TENANT1,
|
|
locations=encrypted_locations)
|
|
self.db.image_create(None, db_data)
|
|
image = self.image_repo.get(UUID1)
|
|
self.assertEqual(image.locations, orig_locations)
|
|
|
|
def test_decrypt_locations_on_list(self):
|
|
url_loc = ['ping', 'pong']
|
|
orig_locations = [{'url': l, 'metadata': {}} for l in url_loc]
|
|
encrypted_locs = [crypt.urlsafe_encrypt(self.crypt_key, l)
|
|
for l in url_loc]
|
|
encrypted_locations = [{'url': l, 'metadata': {}}
|
|
for l in encrypted_locs]
|
|
self.assertNotEqual(encrypted_locations, orig_locations)
|
|
db_data = _db_fixture(UUID1, owner=TENANT1,
|
|
locations=encrypted_locations)
|
|
self.db.image_create(None, db_data)
|
|
image = self.image_repo.list()[0]
|
|
self.assertEqual(image.locations, orig_locations)
|
|
|
|
|
|
class TestImageMemberRepo(test_utils.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestImageMemberRepo, self).setUp()
|
|
self.db = unit_test_utils.FakeDB()
|
|
self.db.reset()
|
|
self.context = glance.context.RequestContext(
|
|
user=USER1, tenant=TENANT1)
|
|
self.image_repo = glance.db.ImageRepo(self.context, self.db)
|
|
self.image_member_factory = glance.domain.ImageMemberFactory()
|
|
self._create_images()
|
|
self._create_image_members()
|
|
image = self.image_repo.get(UUID1)
|
|
self.image_member_repo = glance.db.ImageMemberRepo(self.context,
|
|
self.db, image)
|
|
|
|
def _create_images(self):
|
|
self.images = [
|
|
_db_fixture(UUID1, owner=TENANT1, name='1', size=256,
|
|
status='active'),
|
|
_db_fixture(UUID2, owner=TENANT1, name='2',
|
|
size=512, is_public=False),
|
|
]
|
|
[self.db.image_create(None, image) for image in self.images]
|
|
|
|
self.db.image_tag_set_all(None, UUID1, ['ping', 'pong'])
|
|
|
|
def _create_image_members(self):
|
|
self.image_members = [
|
|
_db_image_member_fixture(UUID1, TENANT2),
|
|
_db_image_member_fixture(UUID1, TENANT3),
|
|
]
|
|
[self.db.image_member_create(None, image_member)
|
|
for image_member in self.image_members]
|
|
|
|
def test_list(self):
|
|
image_members = self.image_member_repo.list()
|
|
image_member_ids = set([i.member_id for i in image_members])
|
|
self.assertEqual(set([TENANT2, TENANT3]), image_member_ids)
|
|
|
|
def test_list_no_members(self):
|
|
image = self.image_repo.get(UUID2)
|
|
self.image_member_repo_uuid2 = glance.db.ImageMemberRepo(
|
|
self.context, self.db, image)
|
|
image_members = self.image_member_repo_uuid2.list()
|
|
image_member_ids = set([i.member_id for i in image_members])
|
|
self.assertEqual(set([]), image_member_ids)
|
|
|
|
def test_save_image_member(self):
|
|
image_member = self.image_member_repo.get(TENANT2)
|
|
image_member.status = 'accepted'
|
|
image_member_updated = self.image_member_repo.save(image_member)
|
|
self.assertEqual(image_member.id, image_member_updated.id)
|
|
self.assertEqual(image_member_updated.status, 'accepted')
|
|
|
|
def test_add_image_member(self):
|
|
image = self.image_repo.get(UUID1)
|
|
image_member = self.image_member_factory.new_image_member(image,
|
|
TENANT4)
|
|
self.assertTrue(image_member.id is None)
|
|
retreived_image_member = self.image_member_repo.add(image_member)
|
|
self.assertEqual(retreived_image_member.id, image_member.id)
|
|
self.assertEqual(retreived_image_member.image_id,
|
|
image_member.image_id)
|
|
self.assertEqual(retreived_image_member.member_id,
|
|
image_member.member_id)
|
|
self.assertEqual(retreived_image_member.status,
|
|
'pending')
|
|
|
|
def test_add_duplicate_image_member(self):
|
|
image = self.image_repo.get(UUID1)
|
|
image_member = self.image_member_factory.new_image_member(image,
|
|
TENANT4)
|
|
self.assertTrue(image_member.id is None)
|
|
retreived_image_member = self.image_member_repo.add(image_member)
|
|
self.assertEqual(retreived_image_member.id, image_member.id)
|
|
self.assertEqual(retreived_image_member.image_id,
|
|
image_member.image_id)
|
|
self.assertEqual(retreived_image_member.member_id,
|
|
image_member.member_id)
|
|
self.assertEqual(retreived_image_member.status,
|
|
'pending')
|
|
|
|
self.assertRaises(exception.Duplicate, self.image_member_repo.add,
|
|
image_member)
|
|
|
|
def test_get_image_member(self):
|
|
image = self.image_repo.get(UUID1)
|
|
image_member = self.image_member_factory.new_image_member(image,
|
|
TENANT4)
|
|
self.assertIsNone(image_member.id)
|
|
self.image_member_repo.add(image_member)
|
|
|
|
member = self.image_member_repo.get(image_member.member_id)
|
|
|
|
self.assertEqual(member.id, image_member.id)
|
|
self.assertEqual(member.image_id, image_member.image_id)
|
|
self.assertEqual(member.member_id, image_member.member_id)
|
|
self.assertEqual(member.status, 'pending')
|
|
|
|
def test_get_nonexistent_image_member(self):
|
|
fake_image_member_id = 'fake'
|
|
self.assertRaises(exception.NotFound, self.image_member_repo.get,
|
|
fake_image_member_id)
|
|
|
|
def test_remove_image_member(self):
|
|
image_member = self.image_member_repo.get(TENANT2)
|
|
self.image_member_repo.remove(image_member)
|
|
self.assertRaises(exception.NotFound, self.image_member_repo.get,
|
|
TENANT2)
|
|
|
|
def test_remove_image_member_does_not_exist(self):
|
|
fake_uuid = uuidutils.generate_uuid()
|
|
image = self.image_repo.get(UUID2)
|
|
fake_member = glance.domain.ImageMemberFactory()\
|
|
.new_image_member(image, TENANT4)
|
|
fake_member.id = fake_uuid
|
|
exc = self.assertRaises(exception.NotFound,
|
|
self.image_member_repo.remove,
|
|
fake_member)
|
|
self.assertTrue(fake_uuid in unicode(exc))
|
|
|
|
|
|
class TestTaskRepo(test_utils.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestTaskRepo, self).setUp()
|
|
self.db = unit_test_utils.FakeDB()
|
|
self.db.reset()
|
|
self.context = glance.context.RequestContext(user=USER1,
|
|
tenant=TENANT1)
|
|
self.task_repo = glance.db.TaskRepo(self.context, self.db)
|
|
self.task_factory = glance.domain.TaskFactory()
|
|
self.fake_task_input = ('{"import_from": '
|
|
'"swift://cloud.foo/account/mycontainer/path"'
|
|
',"image_from_format": "qcow2"}')
|
|
self._create_tasks()
|
|
|
|
def _create_tasks(self):
|
|
self.db.reset()
|
|
self.tasks = [
|
|
_db_task_fixture(UUID1, type='import', status='pending',
|
|
input=self.fake_task_input,
|
|
result='',
|
|
owner=TENANT1,
|
|
message='',
|
|
),
|
|
_db_task_fixture(UUID2, type='import', status='processing',
|
|
input=self.fake_task_input,
|
|
result='',
|
|
owner=TENANT1,
|
|
message='',
|
|
),
|
|
_db_task_fixture(UUID3, type='import', status='failure',
|
|
input=self.fake_task_input,
|
|
result='',
|
|
owner=TENANT1,
|
|
message='',
|
|
),
|
|
_db_task_fixture(UUID4, type='import', status='success',
|
|
input=self.fake_task_input,
|
|
result='',
|
|
owner=TENANT2,
|
|
message='',
|
|
),
|
|
]
|
|
[self.db.task_create(None, task) for task in self.tasks]
|
|
|
|
def test_get(self):
|
|
task = self.task_repo.get(UUID1)
|
|
self.assertEqual(task.task_id, UUID1)
|
|
self.assertEqual(task.type, 'import')
|
|
self.assertEqual(task.status, 'pending')
|
|
self.assertEqual(task.input, self.fake_task_input)
|
|
self.assertEqual(task.result, '')
|
|
self.assertEqual(task.owner, TENANT1)
|
|
|
|
def test_get_not_found(self):
|
|
self.assertRaises(exception.NotFound, self.task_repo.get,
|
|
uuidutils.generate_uuid())
|
|
|
|
def test_get_forbidden(self):
|
|
self.assertRaises(exception.NotFound, self.task_repo.get, UUID4)
|
|
|
|
def test_list(self):
|
|
tasks = self.task_repo.list()
|
|
task_ids = set([i.task_id for i in tasks])
|
|
self.assertEqual(set([UUID1, UUID2, UUID3]), task_ids)
|
|
|
|
def test_list_with_type(self):
|
|
filters = {'type': 'import'}
|
|
tasks = self.task_repo.list(filters=filters)
|
|
task_ids = set([i.task_id for i in tasks])
|
|
self.assertEqual(set([UUID1, UUID2, UUID3]), task_ids)
|
|
|
|
def test_list_with_status(self):
|
|
filters = {'status': 'failure'}
|
|
tasks = self.task_repo.list(filters=filters)
|
|
task_ids = set([i.task_id for i in tasks])
|
|
self.assertEqual(set([UUID3]), task_ids)
|
|
|
|
def test_list_with_marker(self):
|
|
full_tasks = self.task_repo.list()
|
|
full_ids = [i.task_id for i in full_tasks]
|
|
marked_tasks = self.task_repo.list(marker=full_ids[0])
|
|
actual_ids = [i.task_id for i in marked_tasks]
|
|
self.assertEqual(actual_ids, full_ids[1:])
|
|
|
|
def test_list_with_last_marker(self):
|
|
tasks = self.task_repo.list()
|
|
marked_tasks = self.task_repo.list(marker=tasks[-1].task_id)
|
|
self.assertEqual(len(marked_tasks), 0)
|
|
|
|
def test_limited_list(self):
|
|
limited_tasks = self.task_repo.list(limit=2)
|
|
self.assertEqual(len(limited_tasks), 2)
|
|
|
|
def test_list_with_marker_and_limit(self):
|
|
full_tasks = self.task_repo.list()
|
|
full_ids = [i.task_id for i in full_tasks]
|
|
marked_tasks = self.task_repo.list(marker=full_ids[0], limit=1)
|
|
actual_ids = [i.task_id for i in marked_tasks]
|
|
self.assertEqual(actual_ids, full_ids[1:2])
|
|
|
|
def test_sorted_list(self):
|
|
tasks = self.task_repo.list(sort_key='status', sort_dir='desc')
|
|
task_ids = [i.task_id for i in tasks]
|
|
self.assertEqual([UUID2, UUID1, UUID3], task_ids)
|
|
|
|
def test_add_task(self):
|
|
task_type = 'import'
|
|
task = self.task_factory.new_task(task_type, self.fake_task_input,
|
|
None)
|
|
self.assertEqual(task.updated_at, task.created_at)
|
|
self.task_repo.add(task)
|
|
retrieved_task = self.task_repo.get(task.task_id)
|
|
self.assertEqual(retrieved_task.updated_at, task.updated_at)
|
|
|
|
def test_save_task(self):
|
|
task = self.task_repo.get(UUID1)
|
|
original_update_time = task.updated_at
|
|
self.task_repo.save(task)
|
|
current_update_time = task.updated_at
|
|
self.assertTrue(current_update_time > original_update_time)
|
|
task = self.task_repo.get(UUID1)
|
|
self.assertEqual(task.updated_at, current_update_time)
|
|
|
|
def test_remove_task(self):
|
|
task = self.task_repo.get(UUID1)
|
|
self.task_repo.remove(task)
|
|
self.assertRaises(exception.NotFound,
|
|
self.task_repo.get,
|
|
task.task_id)
|