def test_launch(self):
instance = boto.ec2.instance.Instance()
instance.id = TEST_INSTANCE_ID
instance.image_id = TEST_IMAGE_ID
reservation = boto.ec2.instance.Reservation()
reservation.instances = [instance]
self.mox.StubOutWithMock(forms, 'get_key_pair_choices')
self.mox.StubOutWithMock(forms, 'get_instance_type_choices')
self.mox.StubOutWithMock(self.project, 'run_instances')
forms.get_key_pair_choices(self.project).AndReturn(
self.create_key_pair_choices([TEST_KEY]))
forms.get_instance_type_choices().AndReturn(
self.create_instance_type_choices())
self.project.run_instances(TEST_IMAGE_ID,
addressing_type=mox.IgnoreArg(),
key_name=TEST_KEY,
user_data='',
instance_type='m1.medium',
min_count='1',
max_count='1').AndReturn(reservation)
self.mox.ReplayAll()
url = reverse('nova_images_launch', args=[TEST_PROJECT, TEST_IMAGE_ID])
data = {'key_name': TEST_KEY,
'count': '1',
'size': 'm1.medium',
'user_data': ''}
res = self.client.post(url, data)
self.assertRedirectsNoFollow(res, reverse('nova_instances',
args=[TEST_PROJECT]))
self.mox.VerifyAll()
def test_index(self):
self.mox.StubOutWithMock(self.project, 'get_images')
self.mox.StubOutWithMock(forms, 'get_key_pair_choices')
self.mox.StubOutWithMock(forms, 'get_instance_type_choices')
self.project.get_images().AndReturn([])
forms.get_key_pair_choices(self.project).AndReturn([])
forms.get_instance_type_choices().AndReturn([])
self.mox.ReplayAll()
res = self.client.get(reverse('nova_images', args=[TEST_PROJECT]))
self.assertEqual(res.status_code, 200)
self.assertTemplateUsed(res, 'django_nova/images/index.html')
self.assertEqual(len(res.context['image_lists']), 3)
self.mox.VerifyAll()
def test_launch_form(self):
self.mox.StubOutWithMock(self.project, 'get_image')
self.mox.StubOutWithMock(forms, 'get_key_pair_choices')
self.mox.StubOutWithMock(forms, 'get_instance_type_choices')
self.project.get_image(TEST_IMAGE_ID).AndReturn(self.ami)
forms.get_key_pair_choices(self.project).AndReturn([])
forms.get_instance_type_choices().AndReturn([])
self.mox.ReplayAll()
args = [TEST_PROJECT, TEST_IMAGE_ID]
res = self.client.get(reverse('nova_images_launch', args=args))
self.assertEqual(res.status_code, 200)
self.assertTemplateUsed(res, 'django_nova/images/launch.html')
self.assertEqual(res.context['ami'].id, TEST_IMAGE_ID)
self.mox.VerifyAll()
def test_detail(self):
self.mox.StubOutWithMock(self.project, 'get_images')
self.mox.StubOutWithMock(self.project, 'get_image')
self.mox.StubOutWithMock(shortcuts, 'get_user_image_permissions')
self.mox.StubOutWithMock(forms, 'get_key_pair_choices')
self.mox.StubOutWithMock(forms, 'get_instance_type_choices')
self.project.get_images().AndReturn([self.ami])
self.project.get_image(TEST_IMAGE_ID).AndReturn(self.ami)
forms.get_key_pair_choices(self.project).AndReturn(
self.create_key_pair_choices([TEST_KEY]))
forms.get_instance_type_choices().AndReturn(
self.create_instance_type_choices())
self.mox.ReplayAll()
res = self.client.get(reverse('nova_images_detail',
args=[TEST_PROJECT, TEST_IMAGE_ID]))
self.assertEqual(res.status_code, 200)
self.assertTemplateUsed(res, 'django_nova/images/index.html')
self.assertEqual(res.context['ami'].id, TEST_IMAGE_ID)
self.mox.VerifyAll()
