09ecdc55a0
Fix the HeatTestCase import issue as required by H302, a following patch of https://review.openstack.org/#/c/129846/ Change-Id: Iab9a508682e9fe2932b7d00b191c8a4cdea4738b
195 lines
6.9 KiB
Python
195 lines
6.9 KiB
Python
#
|
|
# 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 glanceclient import exc as glance_exceptions
|
|
import mock
|
|
import six
|
|
|
|
from heat.common import exception
|
|
from heat.common import template_format
|
|
from heat.engine import parser
|
|
from heat.engine import resource
|
|
from heat.engine.resources import glance_image as gi
|
|
from heat.tests import common
|
|
from heat.tests import utils
|
|
|
|
image_template = '''
|
|
heat_template_version: 2013-05-23
|
|
description: This template to define a glance image.
|
|
resources:
|
|
my_image:
|
|
type: OS::Glance::Image
|
|
properties:
|
|
name: cirros_image
|
|
id: 41f0e60c-ebb4-4375-a2b4-845ae8b9c995
|
|
disk_format: qcow2
|
|
container_format: bare
|
|
is_public: True
|
|
min_disk: 10
|
|
min_ram: 512
|
|
protected: False
|
|
location: https://launchpad.net/cirros/cirros-0.3.0-x86_64-disk.img
|
|
'''
|
|
|
|
image_template_validate = '''
|
|
heat_template_version: 2013-05-23
|
|
description: This template to define a glance image.
|
|
resources:
|
|
image:
|
|
type: OS::Glance::Image
|
|
properties:
|
|
name: image_validate
|
|
disk_format: qcow2
|
|
container_format: bare
|
|
location: https://launchpad.net/cirros/cirros-0.3.0-x86_64-disk.img
|
|
'''
|
|
|
|
|
|
class GlanceImageTest(common.HeatTestCase):
|
|
def setUp(self):
|
|
super(GlanceImageTest, self).setUp()
|
|
|
|
utils.setup_dummy_db()
|
|
self.ctx = utils.dummy_context()
|
|
|
|
# For unit testing purpose. Register resource provider
|
|
# explicitly.
|
|
resource._register_class("OS::Glance::Image", gi.GlanceImage)
|
|
tpl = template_format.parse(image_template)
|
|
self.stack = parser.Stack(
|
|
self.ctx, 'glance_image_test_stack',
|
|
parser.Template(tpl)
|
|
)
|
|
|
|
self.my_image = self.stack['my_image']
|
|
glance = mock.MagicMock()
|
|
self.glanceclient = mock.MagicMock()
|
|
self.my_image.glance = glance
|
|
glance.return_value = self.glanceclient
|
|
self.images = self.glanceclient.images
|
|
|
|
def _test_validate(self, resource, error_msg):
|
|
exc = self.assertRaises(exception.StackValidationFailed,
|
|
resource.validate)
|
|
self.assertIn(error_msg, six.text_type(exc))
|
|
|
|
def test_resource_mapping(self):
|
|
mapping = gi.resource_mapping()
|
|
self.assertEqual(1, len(mapping))
|
|
self.assertEqual(gi.GlanceImage, mapping['OS::Glance::Image'])
|
|
self.assertIsInstance(self.my_image, gi.GlanceImage)
|
|
|
|
def test_invalid_min_disk(self):
|
|
# invalid 'min_disk'
|
|
tpl = template_format.parse(image_template_validate)
|
|
stack = parser.Stack(
|
|
self.ctx, 'glance_image_stack_validate',
|
|
parser.Template(tpl)
|
|
)
|
|
image = stack['image']
|
|
image.t['Properties']['min_disk'] = -1
|
|
error_msg = 'min_disk -1 is out of range (min: 0, max: None)'
|
|
self._test_validate(image, error_msg)
|
|
|
|
def test_invalid_min_ram(self):
|
|
# invalid 'min_ram'
|
|
tpl = template_format.parse(image_template_validate)
|
|
stack = parser.Stack(
|
|
self.ctx, 'glance_image_stack_validate',
|
|
parser.Template(tpl)
|
|
)
|
|
image = stack['image']
|
|
image.t['Properties']['min_ram'] = -1
|
|
error_msg = 'min_ram -1 is out of range (min: 0, max: None)'
|
|
self._test_validate(image, error_msg)
|
|
|
|
def test_miss_disk_format(self):
|
|
# miss disk_format
|
|
tpl = template_format.parse(image_template_validate)
|
|
stack = parser.Stack(
|
|
self.ctx, 'glance_image_stack_validate',
|
|
parser.Template(tpl)
|
|
)
|
|
image = stack['image']
|
|
image.t['Properties'].pop('disk_format')
|
|
error_msg = 'Property disk_format not assigned'
|
|
self._test_validate(image, error_msg)
|
|
|
|
def test_invalid_disk_format(self):
|
|
# invalid disk_format
|
|
tpl = template_format.parse(image_template_validate)
|
|
stack = parser.Stack(
|
|
self.ctx, 'glance_image_stack_validate',
|
|
parser.Template(tpl)
|
|
)
|
|
image = stack['image']
|
|
image.t['Properties']['disk_format'] = 'incorrect_format'
|
|
error_msg = ('disk_format "incorrect_format" is not an allowed value '
|
|
'[ami, ari, aki, vhd, vmdk, raw, qcow2, vdi, iso]')
|
|
self._test_validate(image, error_msg)
|
|
|
|
def test_miss_container_format(self):
|
|
# miss container_format
|
|
tpl = template_format.parse(image_template_validate)
|
|
stack = parser.Stack(
|
|
self.ctx, 'glance_image_stack_validate',
|
|
parser.Template(tpl)
|
|
)
|
|
image = stack['image']
|
|
image.t['Properties'].pop('container_format')
|
|
error_msg = 'Property container_format not assigned'
|
|
self._test_validate(image, error_msg)
|
|
|
|
def test_invalid_container_format(self):
|
|
# invalid container_format
|
|
tpl = template_format.parse(image_template_validate)
|
|
stack = parser.Stack(
|
|
self.ctx, 'glance_image_stack_validate',
|
|
parser.Template(tpl)
|
|
)
|
|
image = stack['image']
|
|
image.t['Properties']['container_format'] = 'incorrect_format'
|
|
error_msg = ('container_format "incorrect_format" is not an '
|
|
'allowed value [ami, ari, aki, bare, ova, ovf]')
|
|
self._test_validate(image, error_msg)
|
|
|
|
def test_miss_location(self):
|
|
# miss location
|
|
tpl = template_format.parse(image_template_validate)
|
|
stack = parser.Stack(
|
|
self.ctx, 'glance_image_stack_validate',
|
|
parser.Template(tpl)
|
|
)
|
|
image = stack['image']
|
|
image.t['Properties'].pop('location')
|
|
error_msg = 'Property location not assigned'
|
|
self._test_validate(image, error_msg)
|
|
|
|
def test_image_handle_create(self):
|
|
value = mock.MagicMock()
|
|
image_id = '41f0e60c-ebb4-4375-a2b4-845ae8b9c995'
|
|
value.id = image_id
|
|
self.images.create.return_value = value
|
|
self.my_image.handle_create()
|
|
self.assertEqual(image_id, self.my_image.resource_id)
|
|
|
|
def test_image_handle_delete(self):
|
|
self.resource_id = None
|
|
self.assertIsNone(self.my_image.handle_delete())
|
|
image_id = '41f0e60c-ebb4-4375-a2b4-845ae8b9c995'
|
|
self.my_image.resource_id = image_id
|
|
self.images.delete.return_value = None
|
|
self.assertIsNone(self.my_image.handle_delete())
|
|
self.images.delete.side_effect = glance_exceptions.HTTPNotFound(404)
|
|
self.assertIsNone(self.my_image.handle_delete())
|