diff options
Diffstat (limited to 'horizon/horizon/dashboards/nova/images/tests.py')
| -rw-r--r-- | horizon/horizon/dashboards/nova/images/tests.py | 370 |
1 files changed, 370 insertions, 0 deletions
diff --git a/horizon/horizon/dashboards/nova/images/tests.py b/horizon/horizon/dashboards/nova/images/tests.py new file mode 100644 index 00000000..24e2f5bc --- /dev/null +++ b/horizon/horizon/dashboards/nova/images/tests.py @@ -0,0 +1,370 @@ +# vim: tabstop=4 shiftwidth=4 softtabstop=4 + +# Copyright 2011 United States Government as represented by the +# Administrator of the National Aeronautics and Space Administration. +# All Rights Reserved. +# +# Copyright 2011 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 django import http +from django.contrib import messages +from django.core.urlresolvers import reverse +from glance.common import exception as glance_exception +from openstackx.api import exceptions as api_exceptions +from mox import IgnoreArg, IsA + +from horizon import api +from horizon import test + + +IMAGES_INDEX_URL = reverse('horizon:nova:images:index') + + +class FakeQuota: + ram = 100 + + +class ImageViewTests(test.BaseViewTests): + def setUp(self): + super(ImageViewTests, self).setUp() + image_dict = {'name': 'visibleImage', + 'container_format': 'novaImage'} + self.visibleImage = api.Image(image_dict) + + image_dict = {'name': 'invisibleImage', + 'container_format': 'aki'} + self.invisibleImage = api.Image(image_dict) + + self.images = (self.visibleImage, self.invisibleImage) + + flavor = self.mox.CreateMock(api.Flavor) + flavor.id = 1 + flavor.name = 'm1.massive' + flavor.vcpus = 1000 + flavor.disk = 1024 + flavor.ram = 10000 + self.flavors = (flavor,) + + keypair = self.mox.CreateMock(api.KeyPair) + keypair.name = 'keyName' + self.keypairs = (keypair,) + + security_group = self.mox.CreateMock(api.SecurityGroup) + security_group.name = 'default' + self.security_groups = (security_group,) + + def test_index(self): + self.mox.StubOutWithMock(api, 'image_list_detailed') + api.image_list_detailed(IsA(http.HttpRequest)).AndReturn(self.images) + + self.mox.ReplayAll() + + res = self.client.get(IMAGES_INDEX_URL) + + self.assertTemplateUsed(res, 'nova/images/index.html') + + self.assertIn('images', res.context) + images = res.context['images'] + self.assertEqual(len(images), 1) + self.assertEqual(images[0].name, 'visibleImage') + + self.mox.VerifyAll() + + def test_index_no_images(self): + self.mox.StubOutWithMock(api, 'image_list_detailed') + api.image_list_detailed(IsA(http.HttpRequest)).AndReturn([]) + + self.mox.StubOutWithMock(messages, 'info') + messages.info(IsA(http.HttpRequest), IsA(basestring)) + + self.mox.ReplayAll() + + res = self.client.get(IMAGES_INDEX_URL) + + self.assertTemplateUsed(res, 'nova/images/index.html') + + self.mox.VerifyAll() + + def test_index_client_conn_error(self): + self.mox.StubOutWithMock(api, 'image_list_detailed') + exception = glance_exception.ClientConnectionError('clientConnError') + api.image_list_detailed(IsA(http.HttpRequest)).AndRaise(exception) + + self.mox.StubOutWithMock(messages, 'error') + messages.error(IsA(http.HttpRequest), IsA(basestring)) + + self.mox.ReplayAll() + + res = self.client.get(IMAGES_INDEX_URL) + + self.assertTemplateUsed(res, + 'nova/images/index.html') + + self.mox.VerifyAll() + + def test_index_glance_error(self): + self.mox.StubOutWithMock(api, 'image_list_detailed') + exception = glance_exception.Error('glanceError') + api.image_list_detailed(IsA(http.HttpRequest)).AndRaise(exception) + + self.mox.StubOutWithMock(messages, 'error') + messages.error(IsA(http.HttpRequest), IsA(basestring)) + + self.mox.ReplayAll() + + res = self.client.get(IMAGES_INDEX_URL) + + self.assertTemplateUsed(res, 'nova/images/index.html') + + self.mox.VerifyAll() + + def test_launch_get(self): + IMAGE_ID = '1' + + self.mox.StubOutWithMock(api, 'image_get') + api.image_get(IsA(http.HttpRequest), + IMAGE_ID).AndReturn(self.visibleImage) + + self.mox.StubOutWithMock(api, 'tenant_quota_get') + api.tenant_quota_get(IsA(http.HttpRequest), + self.TEST_TENANT).AndReturn(FakeQuota) + + self.mox.StubOutWithMock(api, 'flavor_list') + api.flavor_list(IsA(http.HttpRequest)).AndReturn(self.flavors) + + self.mox.StubOutWithMock(api, 'keypair_list') + api.keypair_list(IsA(http.HttpRequest)).AndReturn(self.keypairs) + + self.mox.StubOutWithMock(api, 'security_group_list') + api.security_group_list(IsA(http.HttpRequest)).AndReturn( + self.security_groups) + + self.mox.ReplayAll() + + res = self.client.get(reverse('horizon:nova:images:launch', + args=[IMAGE_ID])) + + self.assertTemplateUsed(res, 'nova/images/launch.html') + + image = res.context['image'] + self.assertEqual(image.name, self.visibleImage.name) + + form = res.context['form'] + + form_flavorfield = form.fields['flavor'] + self.assertIn('m1.massive', form_flavorfield.choices[0][1]) + + form_keyfield = form.fields['key_name'] + self.assertEqual(form_keyfield.choices[0][0], + self.keypairs[0].name) + + self.mox.VerifyAll() + + def test_launch_post(self): + FLAVOR_ID = self.flavors[0].id + IMAGE_ID = '1' + KEY_NAME = self.keypairs[0].name + SERVER_NAME = 'serverName' + USER_DATA = 'userData' + + form_data = {'method': 'LaunchForm', + 'flavor': FLAVOR_ID, + 'image_id': IMAGE_ID, + 'key_name': KEY_NAME, + 'name': SERVER_NAME, + 'user_data': USER_DATA, + 'tenant_id': self.TEST_TENANT, + 'security_groups': 'default', + } + + self.mox.StubOutWithMock(api, 'image_get') + api.image_get(IsA(http.HttpRequest), + IMAGE_ID).AndReturn(self.visibleImage) + + self.mox.StubOutWithMock(api, 'tenant_quota_get') + api.tenant_quota_get(IsA(http.HttpRequest), + self.TEST_TENANT).AndReturn(FakeQuota) + + self.mox.StubOutWithMock(api, 'flavor_list') + api.flavor_list(IsA(http.HttpRequest)).AndReturn(self.flavors) + + self.mox.StubOutWithMock(api, 'keypair_list') + api.keypair_list(IsA(http.HttpRequest)).AndReturn(self.keypairs) + + self.mox.StubOutWithMock(api, 'security_group_list') + api.security_group_list(IsA(http.HttpRequest)).AndReturn( + self.security_groups) + + # called again by the form + api.image_get(IsA(http.HttpRequest), + IMAGE_ID).AndReturn(self.visibleImage) + + self.mox.StubOutWithMock(api, 'flavor_get') + api.flavor_get(IsA(http.HttpRequest), + IsA(unicode)).AndReturn(self.flavors[0]) + + self.mox.StubOutWithMock(api, 'server_create') + + api.server_create(IsA(http.HttpRequest), SERVER_NAME, + self.visibleImage, self.flavors[0], + KEY_NAME, USER_DATA, [self.security_groups[0].name]) + + self.mox.StubOutWithMock(messages, 'success') + messages.success(IsA(http.HttpRequest), IsA(basestring)) + + self.mox.ReplayAll() + + res = self.client.post(reverse('horizon:nova:images:launch', + args=[IMAGE_ID]), + form_data) + + self.assertRedirectsNoFollow(res, + reverse('horizon:nova:instances:index')) + + self.mox.VerifyAll() + + def test_launch_flavorlist_error(self): + IMAGE_ID = '1' + + self.mox.StubOutWithMock(api, 'image_get') + api.image_get(IsA(http.HttpRequest), + IMAGE_ID).AndReturn(self.visibleImage) + + self.mox.StubOutWithMock(api, 'tenant_quota_get') + api.tenant_quota_get(IsA(http.HttpRequest), + self.TEST_TENANT).AndReturn(FakeQuota) + + exception = api_exceptions.ApiException('apiException') + self.mox.StubOutWithMock(api, 'flavor_list') + api.flavor_list(IsA(http.HttpRequest)).AndRaise(exception) + + self.mox.StubOutWithMock(api, 'keypair_list') + api.keypair_list(IsA(http.HttpRequest)).AndReturn(self.keypairs) + + self.mox.StubOutWithMock(api, 'security_group_list') + api.security_group_list(IsA(http.HttpRequest)).AndReturn( + self.security_groups) + + self.mox.ReplayAll() + + res = self.client.get(reverse('horizon:nova:images:launch', + args=[IMAGE_ID])) + + self.assertTemplateUsed(res, 'nova/images/launch.html') + + form = res.context['form'] + + form_flavorfield = form.fields['flavor'] + self.assertIn('m1.tiny', form_flavorfield.choices[0][1]) + + self.mox.VerifyAll() + + def test_launch_keypairlist_error(self): + IMAGE_ID = '2' + + self.mox.StubOutWithMock(api, 'image_get') + api.image_get(IsA(http.HttpRequest), + IMAGE_ID).AndReturn(self.visibleImage) + + self.mox.StubOutWithMock(api, 'tenant_quota_get') + api.tenant_quota_get(IsA(http.HttpRequest), + self.TEST_TENANT).AndReturn(FakeQuota) + + self.mox.StubOutWithMock(api, 'flavor_list') + api.flavor_list(IsA(http.HttpRequest)).AndReturn(self.flavors) + + exception = api_exceptions.ApiException('apiException') + self.mox.StubOutWithMock(api, 'keypair_list') + api.keypair_list(IsA(http.HttpRequest)).AndRaise(exception) + + self.mox.StubOutWithMock(api, 'security_group_list') + api.security_group_list(IsA(http.HttpRequest)).AndReturn( + self.security_groups) + + self.mox.ReplayAll() + + res = self.client.get(reverse('horizon:nova:images:launch', + args=[IMAGE_ID])) + + self.assertTemplateUsed(res, 'nova/images/launch.html') + + form = res.context['form'] + + form_keyfield = form.fields['key_name'] + self.assertEqual(len(form_keyfield.choices), 0) + + self.mox.VerifyAll() + + def test_launch_form_apiexception(self): + FLAVOR_ID = self.flavors[0].id + IMAGE_ID = '1' + KEY_NAME = self.keypairs[0].name + SERVER_NAME = 'serverName' + USER_DATA = 'userData' + + form_data = {'method': 'LaunchForm', + 'flavor': FLAVOR_ID, + 'image_id': IMAGE_ID, + 'key_name': KEY_NAME, + 'name': SERVER_NAME, + 'tenant_id': self.TEST_TENANT, + 'user_data': USER_DATA, + 'security_groups': 'default', + } + + self.mox.StubOutWithMock(api, 'image_get') + api.image_get(IgnoreArg(), + IMAGE_ID).AndReturn(self.visibleImage) + + self.mox.StubOutWithMock(api, 'tenant_quota_get') + api.tenant_quota_get(IsA(http.HttpRequest), + self.TEST_TENANT).AndReturn(FakeQuota) + + self.mox.StubOutWithMock(api, 'flavor_list') + api.flavor_list(IgnoreArg()).AndReturn(self.flavors) + + self.mox.StubOutWithMock(api, 'keypair_list') + api.keypair_list(IgnoreArg()).AndReturn(self.keypairs) + + self.mox.StubOutWithMock(api, 'security_group_list') + api.security_group_list(IsA(http.HttpRequest)).AndReturn( + self.security_groups) + + # called again by the form + api.image_get(IgnoreArg(), + IMAGE_ID).AndReturn(self.visibleImage) + + self.mox.StubOutWithMock(api, 'flavor_get') + api.flavor_get(IgnoreArg(), + IsA(unicode)).AndReturn(self.flavors[0]) + + self.mox.StubOutWithMock(api, 'server_create') + + exception = api_exceptions.ApiException('apiException') + api.server_create(IsA(http.HttpRequest), SERVER_NAME, + self.visibleImage, self.flavors[0], + KEY_NAME, USER_DATA, + self.security_groups).AndRaise(exception) + + self.mox.StubOutWithMock(messages, 'error') + messages.error(IsA(http.HttpRequest), IsA(basestring)) + + self.mox.ReplayAll() + url = reverse('horizon:nova:images:launch', args=[IMAGE_ID]) + res = self.client.post(url, form_data) + + self.assertTemplateUsed(res, 'nova/images/launch.html') + + self.mox.VerifyAll() |