Added image metadata tests, fixed minor bug in servers service with metadata

Change-Id: Ia77280ce393619c7dc3f700a7fa4c45305a2defb
This commit is contained in:
Daryl Walleck
2011-11-22 22:28:33 -06:00
parent 3e3f1259e0
commit 416af9271d
4 changed files with 180 additions and 12 deletions

View File

@@ -70,11 +70,15 @@ class ImagesClient(object):
"""Deletes the provided image"""
return self.client.delete("images/%s" % str(image_id))
def wait_for_image_exists(self, image_id):
def wait_for_image_resp_code(self, image_id, code):
"""
Waits until the HTTP response code for the request matches the
expected value
"""
resp, body = self.client.get("images/%s" % str(image_id))
start = int(time.time())
while resp.status != 200:
while resp.status != code:
time.sleep(self.build_interval)
resp, body = self.client.get("images/%s" % str(image_id))
@@ -97,39 +101,45 @@ class ImagesClient(object):
raise exceptions.BuildErrorException
def list_image_metadata(self, image_id):
"""Lists all metadata items for an image"""
resp, body = self.client.get("images/%s/metadata" % str(image_id))
body = json.loads(body)
return resp, body
return resp, body['metadata']
def set_image_metadata(self, image_id, meta):
"""Sets the metadata for an image"""
post_body = json.dumps({'metadata': meta})
resp, body = self.client.put('images/%s/metadata' %
str(image_id), post_body, self.headers)
body = json.loads(body)
return resp, body
return resp, body['metadata']
def update_image_metadata(self, image_id, meta):
"""Updates the metadata for an image"""
post_body = json.dumps({'metadata': meta})
resp, body = self.client.post('images/%s/metadata' %
str(image_id), post_body, self.headers)
body = json.loads(body)
return resp, body
return resp, body['metadata']
def get_image_metadata_item(self, image_id, key):
"""Returns the value for a specific image metadata key"""
resp, body = self.client.get("images/%s/metadata/%s" %
(str(image_id), key))
body = json.loads(body)
return resp, body
return resp, body['meta']
def set_image_metadata_item(self, image_id, key, meta):
"""Sets the value for a specific image metadata key"""
post_body = json.dumps({'meta': meta})
resp, body = self.client.put('images/%s/metdata/%s' %
(str(image_id), key),
post_body, self.headers)
resp, body = self.client.put('images/%s/metadata/%s' %
(str(image_id), key), post_body,
self.headers)
body = json.loads(body)
return resp, body
return resp, body['meta']
def delete_image_metadata_item(self, image_id, key):
"""Deletes a single image metadata key/value pair"""
resp, body = self.client.delete("images/%s/metadata/%s" %
(str(image_id), key))
return resp, body

View File

@@ -285,7 +285,7 @@ class ServersClient(object):
def set_server_metadata_item(self, server_id, key, meta):
post_body = json.dumps({'meta': meta})
resp, body = self.client.put('servers/%s/metdata/%s' %
resp, body = self.client.put('servers/%s/metadata/%s' %
(str(server_id), key),
post_body, self.headers)
body = json.loads(body)

View File

@@ -0,0 +1,158 @@
from nose.plugins.attrib import attr
from storm import openstack
from storm.common.utils.data_utils import rand_name
import storm.config
import unittest2 as unittest
class ImagesMetadataTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.os = openstack.Manager()
cls.servers_client = cls.os.servers_client
cls.client = cls.os.images_client
cls.config = storm.config.StormConfig()
cls.image_ref = cls.config.env.image_ref
cls.flavor_ref = cls.config.env.flavor_ref
cls.ssh_timeout = cls.config.nova.ssh_timeout
name = rand_name('server')
resp, cls.server = cls.servers_client.create_server(name,
cls.image_ref,
cls.flavor_ref)
#Wait for the server to become active
cls.servers_client.wait_for_server_status(cls.server['id'], 'ACTIVE')
#Create an image from the server
name = rand_name('image')
cls.meta = {'key1': 'value1', 'key2': 'value2'}
resp, body = cls.client.create_image(cls.server['id'], name, cls.meta)
image_ref = resp['location']
temp = image_ref.rsplit('/')
image_id = temp[len(temp) - 1]
cls.client.wait_for_image_resp_code(image_id, 200)
cls.client.wait_for_image_status(image_id, 'ACTIVE')
resp, cls.image = cls.client.get_image(image_id)
@classmethod
def tearDownClass(cls):
cls.servers_client.delete_server(cls.server['id'])
cls.client.delete_image(cls.image['id'])
def _parse_image_id(self, image_ref):
temp = image_ref.rsplit('/')
return len(temp) - 1
def test_list_image_metadata(self):
"""All metadata key/value pairs for an image should be returned"""
resp, metadata = self.client.list_image_metadata(self.image['id'])
self.assertEqual('value1', metadata['key1'])
self.assertEqual('value2', metadata['key2'])
def test_set_image_metadata(self):
"""The metadata for the image should match the new values"""
meta = {'meta1': 'data1'}
name = rand_name('server')
resp, server = self.servers_client.create_server(name, self.image_ref,
self.flavor_ref)
self.servers_client.wait_for_server_status(server['id'], 'ACTIVE')
name = rand_name('image')
resp, body = self.client.create_image(server['id'], name, meta)
image_id = self._parse_image_id(resp['location'])
self.client.wait_for_image_resp_code(image_id, 200)
self.client.wait_for_image_status(image_id, 'ACTIVE')
resp, image = self.client.get_image(image_id)
meta = {'meta2': 'data2', 'meta3': 'data3'}
resp, body = self.client.set_image_metadata(image['id'], meta)
resp, metadata = self.client.list_image_metadata(image['id'])
self.assertEqual('data2', metadata['meta2'])
self.assertEqual('data3', metadata['meta3'])
self.assertTrue('meta1' not in metadata)
self.servers_client.delete_server(server['id'])
self.client.delete_image(image['id'])
def test_update_image_metadata(self):
"""The metadata for the image should match the updated values"""
meta = {'key1': 'value1', 'key2': 'value2'}
name = rand_name('server')
resp, server = self.servers_client.create_server(name, self.image_ref,
self.flavor_ref)
self.servers_client.wait_for_server_status(server['id'], 'ACTIVE')
name = rand_name('image')
resp, body = self.client.create_image(server['id'], name, meta)
image_id = self._parse_image_id(resp['location'])
self.client.wait_for_image_resp_code(image_id, 200)
self.client.wait_for_image_status(image_id, 'ACTIVE')
resp, image = self.client.get_image(image_id)
meta = {'key1': 'alt1', 'key2': 'alt2'}
resp, metadata = self.client.update_image_metadata(image['id'], meta)
resp, metadata = self.client.list_image_metadata(image['id'])
self.assertEqual('alt1', metadata['key1'])
self.assertEqual('alt2', metadata['key2'])
self.servers_client.delete_server(server['id'])
self.client.delete_image(image['id'])
def test_get_image_metadata_item(self):
"""The value for a specic metadata key should be returned"""
resp, meta = self.client.get_image_metadata_item(self.image['id'],
'key2')
self.assertTrue('value2', meta['key2'])
def test_set_image_metadata_item(self):
"""
The value provided for the given meta item should be set for the image
"""
meta = {'nova': 'server'}
name = rand_name('server')
resp, server = self.servers_client.create_server(name, self.image_ref,
self.flavor_ref)
self.servers_client.wait_for_server_status(server['id'], 'ACTIVE')
name = rand_name('image')
resp, body = self.client.create_image(server['id'], name, meta)
image_id = self._parse_image_id(resp['location'])
self.client.wait_for_image_resp_code(image_id, 200)
self.client.wait_for_image_status(image_id, 'ACTIVE')
resp, image = self.client.get_image(image_id)
meta = {'nova': 'alt'}
resp, body = self.client.set_image_metadata_item(image['id'],
'nova', meta)
resp, metadata = self.client.list_image_metadata(image['id'])
self.assertEqual('alt', metadata['nova'])
self.servers_client.delete_server(server['id'])
self.client.delete_image(image['id'])
def test_delete_image_metadata_item(self):
"""The metadata value/key pair should be deleted from the image"""
meta = {'delkey': 'delvalue'}
name = rand_name('server')
resp, server = self.servers_client.create_server(name, self.image_ref,
self.flavor_ref)
self.servers_client.wait_for_server_status(server['id'], 'ACTIVE')
name = rand_name('image')
resp, body = self.client.create_image(server['id'], name, meta)
image_id = self._parse_image_id(resp['location'])
self.client.wait_for_image_resp_code(image_id, 200)
self.client.wait_for_image_status(image_id, 'ACTIVE')
resp, image = self.client.get_image(image_id)
resp, body = self.client.delete_image_metadata_item(image['id'],
'delkey')
resp, metadata = self.client.list_image_metadata(image['id'])
self.assertTrue('delkey' not in metadata)
self.servers_client.delete_server(server['id'])
self.client.delete_image(image['id'])

View File

@@ -35,7 +35,7 @@ class ImagesTest(unittest.TestCase):
name = rand_name('image')
resp, body = self.client.create_image(server['id'], name)
image_id = self._parse_image_id(resp['location'])
self.client.wait_for_image_exists(image_id)
self.client.wait_for_image_resp_code(image_id, 200)
self.client.wait_for_image_status(image_id, 'ACTIVE')
#Verify the image was created correctly