a5580f4636
Make the fake image service able to store image data in memory. This is handy for unit testing, but probably not very useful otherwise (images are pretty big, so holding them in memory is not a very good idea). Change-Id: I2d07baf24cab544e19521b9486feee3272d0407e
147 lines
5.6 KiB
Python
147 lines
5.6 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2011 OpenStack LLC
|
|
# Author: Soren Hansen
|
|
#
|
|
# 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 datetime
|
|
import StringIO
|
|
|
|
from nova import context
|
|
from nova import exception
|
|
from nova import test
|
|
import nova.image
|
|
|
|
|
|
class _ImageTestCase(test.TestCase):
|
|
def setUp(self):
|
|
super(_ImageTestCase, self).setUp()
|
|
self.context = context.get_admin_context()
|
|
|
|
def test_index(self):
|
|
res = self.image_service.index(self.context)
|
|
for image in res:
|
|
self.assertEquals(set(image.keys()), set(['id', 'name']))
|
|
|
|
def test_detail(self):
|
|
res = self.image_service.detail(self.context)
|
|
for image in res:
|
|
keys = set(image.keys())
|
|
self.assertEquals(keys, set(['id', 'name', 'created_at',
|
|
'updated_at', 'deleted_at', 'deleted',
|
|
'status', 'is_public', 'properties',
|
|
'disk_format', 'container_format']))
|
|
self.assertTrue(isinstance(image['created_at'], datetime.datetime))
|
|
self.assertTrue(isinstance(image['updated_at'], datetime.datetime))
|
|
|
|
if not (isinstance(image['deleted_at'], datetime.datetime) or
|
|
image['deleted_at'] is None):
|
|
self.fail('image\'s "deleted_at" attribute was neither a '
|
|
'datetime object nor None')
|
|
|
|
def check_is_bool(image, key):
|
|
val = image.get('deleted')
|
|
if not isinstance(val, bool):
|
|
self.fail('image\'s "%s" attribute wasn\'t '
|
|
'a bool: %r' % (key, val))
|
|
|
|
check_is_bool(image, 'deleted')
|
|
check_is_bool(image, 'is_public')
|
|
|
|
def test_index_and_detail_have_same_results(self):
|
|
index = self.image_service.index(self.context)
|
|
detail = self.image_service.detail(self.context)
|
|
index_set = set([(i['id'], i['name']) for i in index])
|
|
detail_set = set([(i['id'], i['name']) for i in detail])
|
|
self.assertEqual(index_set, detail_set)
|
|
|
|
def test_show_raises_imagenotfound_for_invalid_id(self):
|
|
self.assertRaises(exception.ImageNotFound,
|
|
self.image_service.show,
|
|
self.context,
|
|
'this image does not exist')
|
|
|
|
def test_show_by_name(self):
|
|
self.assertRaises(exception.ImageNotFound,
|
|
self.image_service.show_by_name,
|
|
self.context,
|
|
'this image does not exist')
|
|
|
|
def test_create_adds_id(self):
|
|
index = self.image_service.index(self.context)
|
|
image_count = len(index)
|
|
|
|
self.image_service.create(self.context, {})
|
|
|
|
index = self.image_service.index(self.context)
|
|
self.assertEquals(len(index), image_count + 1)
|
|
|
|
self.assertTrue(index[0]['id'])
|
|
|
|
def test_create_keeps_id(self):
|
|
self.image_service.create(self.context, {'id': '34'})
|
|
self.image_service.show(self.context, '34')
|
|
|
|
def test_create_rejects_duplicate_ids(self):
|
|
self.image_service.create(self.context, {'id': '34'})
|
|
self.assertRaises(exception.Duplicate,
|
|
self.image_service.create,
|
|
self.context,
|
|
{'id': '34'})
|
|
|
|
# Make sure there's still one left
|
|
self.image_service.show(self.context, '34')
|
|
|
|
def test_update(self):
|
|
self.image_service.create(self.context,
|
|
{'id': '34', 'foo': 'bar'})
|
|
|
|
self.image_service.update(self.context, '34',
|
|
{'id': '34', 'foo': 'baz'})
|
|
|
|
img = self.image_service.show(self.context, '34')
|
|
self.assertEquals(img['foo'], 'baz')
|
|
|
|
def test_delete(self):
|
|
self.image_service.create(self.context, {'id': '34', 'foo': 'bar'})
|
|
self.image_service.delete(self.context, '34')
|
|
self.assertRaises(exception.NotFound,
|
|
self.image_service.show,
|
|
self.context,
|
|
'34')
|
|
|
|
def test_delete_all(self):
|
|
self.image_service.create(self.context, {'id': '32', 'foo': 'bar'})
|
|
self.image_service.create(self.context, {'id': '33', 'foo': 'bar'})
|
|
self.image_service.create(self.context, {'id': '34', 'foo': 'bar'})
|
|
self.image_service.delete_all()
|
|
index = self.image_service.index(self.context)
|
|
self.assertEquals(len(index), 0)
|
|
|
|
def test_create_then_get(self):
|
|
blob = 'some data'
|
|
s1 = StringIO.StringIO(blob)
|
|
self.image_service.create(self.context,
|
|
{'id': '32', 'foo': 'bar'},
|
|
data=s1)
|
|
s2 = StringIO.StringIO()
|
|
self.image_service.get(self.context, '32', data=s2)
|
|
self.assertEquals(s2.getvalue(), blob, 'Did not get blob back intact')
|
|
|
|
|
|
class FakeImageTestCase(_ImageTestCase):
|
|
def setUp(self):
|
|
super(FakeImageTestCase, self).setUp()
|
|
self.image_service = nova.image.fake.FakeImageService()
|