8de8238167
Another side effect is that it fixes a bug in host_test_filter.py where there was a tearDown method in the class that didn't call its parent. Ie, test.TestCase.tearDown() was not being called. host_test_filter.py's tearDown has been removed as a part of this cleanup, so the parent will be called now. This happens to fix an unrelated pep8 issue in trunk, as well, so that my tests pass cleanly.
1010 lines
41 KiB
Python
1010 lines
41 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright (c) 2010 Citrix Systems, 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.
|
|
|
|
"""Test suite for XenAPI."""
|
|
|
|
import eventlet
|
|
import functools
|
|
import json
|
|
import os
|
|
import re
|
|
import stubout
|
|
import ast
|
|
|
|
from nova import db
|
|
from nova import context
|
|
from nova import flags
|
|
from nova import log as logging
|
|
from nova import test
|
|
from nova import utils
|
|
from nova.compute import instance_types
|
|
from nova.compute import power_state
|
|
from nova import exception
|
|
from nova.virt import xenapi_conn
|
|
from nova.virt.xenapi import fake as xenapi_fake
|
|
from nova.virt.xenapi import volume_utils
|
|
from nova.virt.xenapi import vmops
|
|
from nova.virt.xenapi import vm_utils
|
|
from nova.tests.db import fakes as db_fakes
|
|
from nova.tests.xenapi import stubs
|
|
from nova.tests.glance import stubs as glance_stubs
|
|
from nova.tests import fake_utils
|
|
|
|
LOG = logging.getLogger('nova.tests.test_xenapi')
|
|
|
|
FLAGS = flags.FLAGS
|
|
|
|
|
|
def stub_vm_utils_with_vdi_attached_here(function, should_return=True):
|
|
"""
|
|
vm_utils.with_vdi_attached_here needs to be stubbed out because it
|
|
calls down to the filesystem to attach a vdi. This provides a
|
|
decorator to handle that.
|
|
"""
|
|
@functools.wraps(function)
|
|
def decorated_function(self, *args, **kwargs):
|
|
orig_with_vdi_attached_here = vm_utils.with_vdi_attached_here
|
|
vm_utils.with_vdi_attached_here = lambda *x: should_return
|
|
function(self, *args, **kwargs)
|
|
vm_utils.with_vdi_attached_here = orig_with_vdi_attached_here
|
|
return decorated_function
|
|
|
|
|
|
class XenAPIVolumeTestCase(test.TestCase):
|
|
"""Unit tests for Volume operations."""
|
|
def setUp(self):
|
|
super(XenAPIVolumeTestCase, self).setUp()
|
|
self.stubs = stubout.StubOutForTesting()
|
|
self.user_id = 'fake'
|
|
self.project_id = 'fake'
|
|
self.context = context.RequestContext(self.user_id, self.project_id)
|
|
self.flags(target_host='127.0.0.1',
|
|
xenapi_connection_url='test_url',
|
|
xenapi_connection_password='test_pass')
|
|
db_fakes.stub_out_db_instance_api(self.stubs)
|
|
stubs.stub_out_get_target(self.stubs)
|
|
xenapi_fake.reset()
|
|
self.values = {'id': 1,
|
|
'project_id': self.user_id,
|
|
'user_id': 'fake',
|
|
'image_ref': 1,
|
|
'kernel_id': 2,
|
|
'ramdisk_id': 3,
|
|
'instance_type_id': '3', # m1.large
|
|
'os_type': 'linux',
|
|
'architecture': 'x86-64'}
|
|
|
|
def _create_volume(self, size='0'):
|
|
"""Create a volume object."""
|
|
vol = {}
|
|
vol['size'] = size
|
|
vol['user_id'] = 'fake'
|
|
vol['project_id'] = 'fake'
|
|
vol['host'] = 'localhost'
|
|
vol['availability_zone'] = FLAGS.storage_availability_zone
|
|
vol['status'] = "creating"
|
|
vol['attach_status'] = "detached"
|
|
return db.volume_create(self.context, vol)
|
|
|
|
def test_create_iscsi_storage(self):
|
|
"""This shows how to test helper classes' methods."""
|
|
stubs.stubout_session(self.stubs, stubs.FakeSessionForVolumeTests)
|
|
session = xenapi_conn.XenAPISession('test_url', 'root', 'test_pass')
|
|
helper = volume_utils.VolumeHelper
|
|
helper.XenAPI = session.get_imported_xenapi()
|
|
vol = self._create_volume()
|
|
info = helper.parse_volume_info(vol['id'], '/dev/sdc')
|
|
label = 'SR-%s' % vol['id']
|
|
description = 'Test-SR'
|
|
sr_ref = helper.create_iscsi_storage(session, info, label, description)
|
|
srs = xenapi_fake.get_all('SR')
|
|
self.assertEqual(sr_ref, srs[0])
|
|
db.volume_destroy(context.get_admin_context(), vol['id'])
|
|
|
|
def test_parse_volume_info_raise_exception(self):
|
|
"""This shows how to test helper classes' methods."""
|
|
stubs.stubout_session(self.stubs, stubs.FakeSessionForVolumeTests)
|
|
session = xenapi_conn.XenAPISession('test_url', 'root', 'test_pass')
|
|
helper = volume_utils.VolumeHelper
|
|
helper.XenAPI = session.get_imported_xenapi()
|
|
vol = self._create_volume()
|
|
# oops, wrong mount point!
|
|
self.assertRaises(volume_utils.StorageError,
|
|
helper.parse_volume_info,
|
|
vol['id'],
|
|
'/dev/sd')
|
|
db.volume_destroy(context.get_admin_context(), vol['id'])
|
|
|
|
def test_attach_volume(self):
|
|
"""This shows how to test Ops classes' methods."""
|
|
stubs.stubout_session(self.stubs, stubs.FakeSessionForVolumeTests)
|
|
conn = xenapi_conn.get_connection(False)
|
|
volume = self._create_volume()
|
|
instance = db.instance_create(self.context, self.values)
|
|
vm = xenapi_fake.create_vm(instance.name, 'Running')
|
|
result = conn.attach_volume(instance.name, volume['id'], '/dev/sdc')
|
|
|
|
def check():
|
|
# check that the VM has a VBD attached to it
|
|
# Get XenAPI record for VBD
|
|
vbds = xenapi_fake.get_all('VBD')
|
|
vbd = xenapi_fake.get_record('VBD', vbds[0])
|
|
vm_ref = vbd['VM']
|
|
self.assertEqual(vm_ref, vm)
|
|
|
|
check()
|
|
|
|
def test_attach_volume_raise_exception(self):
|
|
"""This shows how to test when exceptions are raised."""
|
|
stubs.stubout_session(self.stubs,
|
|
stubs.FakeSessionForVolumeFailedTests)
|
|
conn = xenapi_conn.get_connection(False)
|
|
volume = self._create_volume()
|
|
instance = db.instance_create(self.context, self.values)
|
|
xenapi_fake.create_vm(instance.name, 'Running')
|
|
self.assertRaises(Exception,
|
|
conn.attach_volume,
|
|
instance.name,
|
|
volume['id'],
|
|
'/dev/sdc')
|
|
|
|
def tearDown(self):
|
|
super(XenAPIVolumeTestCase, self).tearDown()
|
|
self.stubs.UnsetAll()
|
|
|
|
|
|
def reset_network(*args):
|
|
pass
|
|
|
|
|
|
def _find_rescue_vbd_ref(*args):
|
|
pass
|
|
|
|
|
|
class XenAPIVMTestCase(test.TestCase):
|
|
"""Unit tests for VM operations."""
|
|
def setUp(self):
|
|
super(XenAPIVMTestCase, self).setUp()
|
|
self.network = utils.import_object(FLAGS.network_manager)
|
|
self.stubs = stubout.StubOutForTesting()
|
|
self.flags(xenapi_connection_url='test_url',
|
|
xenapi_connection_password='test_pass',
|
|
instance_name_template='%d')
|
|
xenapi_fake.reset()
|
|
xenapi_fake.create_local_srs()
|
|
xenapi_fake.create_local_pifs()
|
|
db_fakes.stub_out_db_instance_api(self.stubs)
|
|
xenapi_fake.create_network('fake', FLAGS.flat_network_bridge)
|
|
stubs.stubout_session(self.stubs, stubs.FakeSessionForVMTests)
|
|
stubs.stubout_get_this_vm_uuid(self.stubs)
|
|
stubs.stubout_stream_disk(self.stubs)
|
|
stubs.stubout_is_vdi_pv(self.stubs)
|
|
self.stubs.Set(vmops.VMOps, 'reset_network', reset_network)
|
|
self.stubs.Set(vmops.VMOps, '_find_rescue_vbd_ref',
|
|
_find_rescue_vbd_ref)
|
|
stubs.stub_out_vm_methods(self.stubs)
|
|
glance_stubs.stubout_glance_client(self.stubs)
|
|
fake_utils.stub_out_utils_execute(self.stubs)
|
|
self.user_id = 'fake'
|
|
self.project_id = 'fake'
|
|
self.context = context.RequestContext(self.user_id, self.project_id)
|
|
self.conn = xenapi_conn.get_connection(False)
|
|
|
|
def test_parallel_builds(self):
|
|
stubs.stubout_loopingcall_delay(self.stubs)
|
|
|
|
def _do_build(id, proj, user, *args):
|
|
values = {
|
|
'id': id,
|
|
'project_id': proj,
|
|
'user_id': user,
|
|
'image_ref': 1,
|
|
'kernel_id': 2,
|
|
'ramdisk_id': 3,
|
|
'instance_type_id': '3', # m1.large
|
|
'os_type': 'linux',
|
|
'architecture': 'x86-64'}
|
|
network_info = [({'bridge': 'fa0', 'id': 0, 'injected': False},
|
|
{'broadcast': '192.168.0.255',
|
|
'dns': ['192.168.0.1'],
|
|
'gateway': '192.168.0.1',
|
|
'gateway6': 'dead:beef::1',
|
|
'ip6s': [{'enabled': '1',
|
|
'ip': 'dead:beef::dcad:beff:feef:0',
|
|
'netmask': '64'}],
|
|
'ips': [{'enabled': '1',
|
|
'ip': '192.168.0.100',
|
|
'netmask': '255.255.255.0'}],
|
|
'label': 'fake',
|
|
'mac': 'DE:AD:BE:EF:00:00',
|
|
'rxtx_cap': 3})]
|
|
instance = db.instance_create(self.context, values)
|
|
self.conn.spawn(self.context, instance, network_info)
|
|
|
|
gt1 = eventlet.spawn(_do_build, 1, self.project_id, self.user_id)
|
|
gt2 = eventlet.spawn(_do_build, 2, self.project_id, self.user_id)
|
|
gt1.wait()
|
|
gt2.wait()
|
|
|
|
def test_list_instances_0(self):
|
|
instances = self.conn.list_instances()
|
|
self.assertEquals(instances, [])
|
|
|
|
def test_get_diagnostics(self):
|
|
instance = self._create_instance()
|
|
self.conn.get_diagnostics(instance)
|
|
|
|
def test_instance_snapshot_fails_with_no_primary_vdi(self):
|
|
def create_bad_vbd(vm_ref, vdi_ref):
|
|
vbd_rec = {'VM': vm_ref,
|
|
'VDI': vdi_ref,
|
|
'userdevice': 'fake',
|
|
'currently_attached': False}
|
|
vbd_ref = xenapi_fake._create_object('VBD', vbd_rec)
|
|
xenapi_fake.after_VBD_create(vbd_ref, vbd_rec)
|
|
return vbd_ref
|
|
|
|
self.stubs.Set(xenapi_fake, 'create_vbd', create_bad_vbd)
|
|
stubs.stubout_instance_snapshot(self.stubs)
|
|
instance = self._create_instance()
|
|
|
|
name = "MySnapshot"
|
|
self.assertRaises(exception.Error, self.conn.snapshot,
|
|
self.context, instance, name)
|
|
|
|
def test_instance_snapshot(self):
|
|
stubs.stubout_instance_snapshot(self.stubs)
|
|
instance = self._create_instance()
|
|
|
|
name = "MySnapshot"
|
|
template_vm_ref = self.conn.snapshot(self.context, instance, name)
|
|
|
|
def ensure_vm_was_torn_down():
|
|
vm_labels = []
|
|
for vm_ref in xenapi_fake.get_all('VM'):
|
|
vm_rec = xenapi_fake.get_record('VM', vm_ref)
|
|
if not vm_rec["is_control_domain"]:
|
|
vm_labels.append(vm_rec["name_label"])
|
|
|
|
self.assertEquals(vm_labels, ['1'])
|
|
|
|
def ensure_vbd_was_torn_down():
|
|
vbd_labels = []
|
|
for vbd_ref in xenapi_fake.get_all('VBD'):
|
|
vbd_rec = xenapi_fake.get_record('VBD', vbd_ref)
|
|
vbd_labels.append(vbd_rec["vm_name_label"])
|
|
|
|
self.assertEquals(vbd_labels, ['1'])
|
|
|
|
def ensure_vdi_was_torn_down():
|
|
for vdi_ref in xenapi_fake.get_all('VDI'):
|
|
vdi_rec = xenapi_fake.get_record('VDI', vdi_ref)
|
|
name_label = vdi_rec["name_label"]
|
|
self.assert_(not name_label.endswith('snapshot'))
|
|
|
|
def check():
|
|
ensure_vm_was_torn_down()
|
|
ensure_vbd_was_torn_down()
|
|
ensure_vdi_was_torn_down()
|
|
|
|
check()
|
|
|
|
def create_vm_record(self, conn, os_type, instance_id=1):
|
|
instances = conn.list_instances()
|
|
self.assertEquals(instances, [str(instance_id)])
|
|
|
|
# Get Nova record for VM
|
|
vm_info = conn.get_info(instance_id)
|
|
# Get XenAPI record for VM
|
|
vms = [rec for ref, rec
|
|
in xenapi_fake.get_all_records('VM').iteritems()
|
|
if not rec['is_control_domain']]
|
|
vm = vms[0]
|
|
self.vm_info = vm_info
|
|
self.vm = vm
|
|
|
|
def check_vm_record(self, conn, check_injection=False):
|
|
# Check that m1.large above turned into the right thing.
|
|
instance_type = db.instance_type_get_by_name(conn, 'm1.large')
|
|
mem_kib = long(instance_type['memory_mb']) << 10
|
|
mem_bytes = str(mem_kib << 10)
|
|
vcpus = instance_type['vcpus']
|
|
self.assertEquals(self.vm_info['max_mem'], mem_kib)
|
|
self.assertEquals(self.vm_info['mem'], mem_kib)
|
|
self.assertEquals(self.vm['memory_static_max'], mem_bytes)
|
|
self.assertEquals(self.vm['memory_dynamic_max'], mem_bytes)
|
|
self.assertEquals(self.vm['memory_dynamic_min'], mem_bytes)
|
|
self.assertEquals(self.vm['VCPUs_max'], str(vcpus))
|
|
self.assertEquals(self.vm['VCPUs_at_startup'], str(vcpus))
|
|
|
|
# Check that the VM is running according to Nova
|
|
self.assertEquals(self.vm_info['state'], power_state.RUNNING)
|
|
|
|
# Check that the VM is running according to XenAPI.
|
|
self.assertEquals(self.vm['power_state'], 'Running')
|
|
|
|
if check_injection:
|
|
xenstore_data = self.vm['xenstore_data']
|
|
key = 'vm-data/networking/DEADBEEF0000'
|
|
xenstore_value = xenstore_data[key]
|
|
tcpip_data = ast.literal_eval(xenstore_value)
|
|
self.assertEquals(tcpip_data,
|
|
{'broadcast': '192.168.0.255',
|
|
'dns': ['192.168.0.1'],
|
|
'gateway': '192.168.0.1',
|
|
'gateway6': 'dead:beef::1',
|
|
'ip6s': [{'enabled': '1',
|
|
'ip': 'dead:beef::dcad:beff:feef:0',
|
|
'netmask': '64'}],
|
|
'ips': [{'enabled': '1',
|
|
'ip': '192.168.0.100',
|
|
'netmask': '255.255.255.0'}],
|
|
'label': 'fake',
|
|
'mac': 'DE:AD:BE:EF:00:00'})
|
|
|
|
def check_vm_params_for_windows(self):
|
|
self.assertEquals(self.vm['platform']['nx'], 'true')
|
|
self.assertEquals(self.vm['HVM_boot_params'], {'order': 'dc'})
|
|
self.assertEquals(self.vm['HVM_boot_policy'], 'BIOS order')
|
|
|
|
# check that these are not set
|
|
self.assertEquals(self.vm['PV_args'], '')
|
|
self.assertEquals(self.vm['PV_bootloader'], '')
|
|
self.assertEquals(self.vm['PV_kernel'], '')
|
|
self.assertEquals(self.vm['PV_ramdisk'], '')
|
|
|
|
def check_vm_params_for_linux(self):
|
|
self.assertEquals(self.vm['platform']['nx'], 'false')
|
|
self.assertEquals(self.vm['PV_args'], '')
|
|
self.assertEquals(self.vm['PV_bootloader'], 'pygrub')
|
|
|
|
# check that these are not set
|
|
self.assertEquals(self.vm['PV_kernel'], '')
|
|
self.assertEquals(self.vm['PV_ramdisk'], '')
|
|
self.assertEquals(self.vm['HVM_boot_params'], {})
|
|
self.assertEquals(self.vm['HVM_boot_policy'], '')
|
|
|
|
def check_vm_params_for_linux_with_external_kernel(self):
|
|
self.assertEquals(self.vm['platform']['nx'], 'false')
|
|
self.assertEquals(self.vm['PV_args'], 'root=/dev/xvda1')
|
|
self.assertNotEquals(self.vm['PV_kernel'], '')
|
|
self.assertNotEquals(self.vm['PV_ramdisk'], '')
|
|
|
|
# check that these are not set
|
|
self.assertEquals(self.vm['HVM_boot_params'], {})
|
|
self.assertEquals(self.vm['HVM_boot_policy'], '')
|
|
|
|
def _list_vdis(self):
|
|
url = FLAGS.xenapi_connection_url
|
|
username = FLAGS.xenapi_connection_username
|
|
password = FLAGS.xenapi_connection_password
|
|
session = xenapi_conn.XenAPISession(url, username, password)
|
|
return session.call_xenapi('VDI.get_all')
|
|
|
|
def _check_vdis(self, start_list, end_list):
|
|
for vdi_ref in end_list:
|
|
if not vdi_ref in start_list:
|
|
self.fail('Found unexpected VDI:%s' % vdi_ref)
|
|
|
|
def _test_spawn(self, image_ref, kernel_id, ramdisk_id,
|
|
instance_type_id="3", os_type="linux",
|
|
architecture="x86-64", instance_id=1,
|
|
check_injection=False,
|
|
create_record=True, empty_dns=False):
|
|
stubs.stubout_loopingcall_start(self.stubs)
|
|
if create_record:
|
|
values = {'id': instance_id,
|
|
'project_id': self.project_id,
|
|
'user_id': self.user_id,
|
|
'image_ref': image_ref,
|
|
'kernel_id': kernel_id,
|
|
'ramdisk_id': ramdisk_id,
|
|
'instance_type_id': instance_type_id,
|
|
'os_type': os_type,
|
|
'architecture': architecture}
|
|
instance = db.instance_create(self.context, values)
|
|
else:
|
|
instance = db.instance_get(self.context, instance_id)
|
|
network_info = [({'bridge': 'fa0', 'id': 0, 'injected': True},
|
|
{'broadcast': '192.168.0.255',
|
|
'dns': ['192.168.0.1'],
|
|
'gateway': '192.168.0.1',
|
|
'gateway6': 'dead:beef::1',
|
|
'ip6s': [{'enabled': '1',
|
|
'ip': 'dead:beef::dcad:beff:feef:0',
|
|
'netmask': '64'}],
|
|
'ips': [{'enabled': '1',
|
|
'ip': '192.168.0.100',
|
|
'netmask': '255.255.255.0'}],
|
|
'label': 'fake',
|
|
'mac': 'DE:AD:BE:EF:00:00',
|
|
'rxtx_cap': 3})]
|
|
if empty_dns:
|
|
network_info[0][1]['dns'] = []
|
|
|
|
self.conn.spawn(self.context, instance, network_info)
|
|
self.create_vm_record(self.conn, os_type, instance_id)
|
|
self.check_vm_record(self.conn, check_injection)
|
|
self.assertTrue(instance.os_type)
|
|
self.assertTrue(instance.architecture)
|
|
|
|
def test_spawn_empty_dns(self):
|
|
""""Test spawning with an empty dns list"""
|
|
self._test_spawn(glance_stubs.FakeGlance.IMAGE_VHD, None, None,
|
|
os_type="linux", architecture="x86-64",
|
|
empty_dns=True)
|
|
self.check_vm_params_for_linux()
|
|
|
|
def test_spawn_not_enough_memory(self):
|
|
self.assertRaises(Exception,
|
|
self._test_spawn,
|
|
1, 2, 3, "4") # m1.xlarge
|
|
|
|
def test_spawn_fail_cleanup_1(self):
|
|
"""Simulates an error while downloading an image.
|
|
|
|
Verifies that VDIs created are properly cleaned up.
|
|
|
|
"""
|
|
vdi_recs_start = self._list_vdis()
|
|
stubs.stubout_fetch_image_glance_disk(self.stubs)
|
|
self.assertRaises(xenapi_fake.Failure,
|
|
self._test_spawn, 1, 2, 3)
|
|
# No additional VDI should be found.
|
|
vdi_recs_end = self._list_vdis()
|
|
self._check_vdis(vdi_recs_start, vdi_recs_end)
|
|
|
|
def test_spawn_fail_cleanup_2(self):
|
|
"""Simulates an error while creating VM record.
|
|
|
|
It verifies that VDIs created are properly cleaned up.
|
|
|
|
"""
|
|
vdi_recs_start = self._list_vdis()
|
|
stubs.stubout_create_vm(self.stubs)
|
|
self.assertRaises(xenapi_fake.Failure,
|
|
self._test_spawn, 1, 2, 3)
|
|
# No additional VDI should be found.
|
|
vdi_recs_end = self._list_vdis()
|
|
self._check_vdis(vdi_recs_start, vdi_recs_end)
|
|
|
|
@stub_vm_utils_with_vdi_attached_here
|
|
def test_spawn_raw_glance(self):
|
|
self._test_spawn(glance_stubs.FakeGlance.IMAGE_RAW, None, None)
|
|
self.check_vm_params_for_linux()
|
|
|
|
def test_spawn_vhd_glance_linux(self):
|
|
self._test_spawn(glance_stubs.FakeGlance.IMAGE_VHD, None, None,
|
|
os_type="linux", architecture="x86-64")
|
|
self.check_vm_params_for_linux()
|
|
|
|
def test_spawn_vhd_glance_swapdisk(self):
|
|
# Change the default host_call_plugin to one that'll return
|
|
# a swap disk
|
|
orig_func = stubs.FakeSessionForVMTests.host_call_plugin
|
|
|
|
stubs.FakeSessionForVMTests.host_call_plugin = \
|
|
stubs.FakeSessionForVMTests.host_call_plugin_swap
|
|
|
|
try:
|
|
# We'll steal the above glance linux test
|
|
self.test_spawn_vhd_glance_linux()
|
|
finally:
|
|
# Make sure to put this back
|
|
stubs.FakeSessionForVMTests.host_call_plugin = orig_func
|
|
|
|
# We should have 2 VBDs.
|
|
self.assertEqual(len(self.vm['VBDs']), 2)
|
|
# Now test that we have 1.
|
|
self.tearDown()
|
|
self.setUp()
|
|
self.test_spawn_vhd_glance_linux()
|
|
self.assertEqual(len(self.vm['VBDs']), 1)
|
|
|
|
def test_spawn_vhd_glance_windows(self):
|
|
self._test_spawn(glance_stubs.FakeGlance.IMAGE_VHD, None, None,
|
|
os_type="windows", architecture="i386")
|
|
self.check_vm_params_for_windows()
|
|
|
|
def test_spawn_glance(self):
|
|
self._test_spawn(glance_stubs.FakeGlance.IMAGE_MACHINE,
|
|
glance_stubs.FakeGlance.IMAGE_KERNEL,
|
|
glance_stubs.FakeGlance.IMAGE_RAMDISK)
|
|
self.check_vm_params_for_linux_with_external_kernel()
|
|
|
|
def test_spawn_netinject_file(self):
|
|
db_fakes.stub_out_db_instance_api(self.stubs, injected=True)
|
|
|
|
self._tee_executed = False
|
|
|
|
def _tee_handler(cmd, **kwargs):
|
|
input = kwargs.get('process_input', None)
|
|
self.assertNotEqual(input, None)
|
|
config = [line.strip() for line in input.split("\n")]
|
|
# Find the start of eth0 configuration and check it
|
|
index = config.index('auto eth0')
|
|
self.assertEquals(config[index + 1:index + 8], [
|
|
'iface eth0 inet static',
|
|
'address 192.168.0.100',
|
|
'netmask 255.255.255.0',
|
|
'broadcast 192.168.0.255',
|
|
'gateway 192.168.0.1',
|
|
'dns-nameservers 192.168.0.1',
|
|
''])
|
|
self._tee_executed = True
|
|
return '', ''
|
|
|
|
fake_utils.fake_execute_set_repliers([
|
|
# Capture the sudo tee .../etc/network/interfaces command
|
|
(r'(sudo\s+)?tee.*interfaces', _tee_handler),
|
|
])
|
|
self._test_spawn(glance_stubs.FakeGlance.IMAGE_MACHINE,
|
|
glance_stubs.FakeGlance.IMAGE_KERNEL,
|
|
glance_stubs.FakeGlance.IMAGE_RAMDISK,
|
|
check_injection=True)
|
|
self.assertTrue(self._tee_executed)
|
|
|
|
def test_spawn_netinject_xenstore(self):
|
|
db_fakes.stub_out_db_instance_api(self.stubs, injected=True)
|
|
|
|
self._tee_executed = False
|
|
|
|
def _mount_handler(cmd, *ignore_args, **ignore_kwargs):
|
|
# When mounting, create real files under the mountpoint to simulate
|
|
# files in the mounted filesystem
|
|
|
|
# mount point will be the last item of the command list
|
|
self._tmpdir = cmd[len(cmd) - 1]
|
|
LOG.debug(_('Creating files in %s to simulate guest agent' %
|
|
self._tmpdir))
|
|
os.makedirs(os.path.join(self._tmpdir, 'usr', 'sbin'))
|
|
# Touch the file using open
|
|
open(os.path.join(self._tmpdir, 'usr', 'sbin',
|
|
'xe-update-networking'), 'w').close()
|
|
return '', ''
|
|
|
|
def _umount_handler(cmd, *ignore_args, **ignore_kwargs):
|
|
# Umount would normall make files in the m,ounted filesystem
|
|
# disappear, so do that here
|
|
LOG.debug(_('Removing simulated guest agent files in %s' %
|
|
self._tmpdir))
|
|
os.remove(os.path.join(self._tmpdir, 'usr', 'sbin',
|
|
'xe-update-networking'))
|
|
os.rmdir(os.path.join(self._tmpdir, 'usr', 'sbin'))
|
|
os.rmdir(os.path.join(self._tmpdir, 'usr'))
|
|
return '', ''
|
|
|
|
def _tee_handler(cmd, *ignore_args, **ignore_kwargs):
|
|
self._tee_executed = True
|
|
return '', ''
|
|
|
|
fake_utils.fake_execute_set_repliers([
|
|
(r'(sudo\s+)?mount', _mount_handler),
|
|
(r'(sudo\s+)?umount', _umount_handler),
|
|
(r'(sudo\s+)?tee.*interfaces', _tee_handler)])
|
|
self._test_spawn(1, 2, 3, check_injection=True)
|
|
|
|
# tee must not run in this case, where an injection-capable
|
|
# guest agent is detected
|
|
self.assertFalse(self._tee_executed)
|
|
|
|
def test_spawn_vlanmanager(self):
|
|
self.flags(image_service='nova.image.glance.GlanceImageService',
|
|
network_manager='nova.network.manager.VlanManager',
|
|
vlan_interface='fake0')
|
|
|
|
def dummy(*args, **kwargs):
|
|
pass
|
|
|
|
self.stubs.Set(vmops.VMOps, 'create_vifs', dummy)
|
|
# Reset network table
|
|
xenapi_fake.reset_table('network')
|
|
# Instance id = 2 will use vlan network (see db/fakes.py)
|
|
ctxt = self.context.elevated()
|
|
instance = self._create_instance(2, False)
|
|
networks = self.network.db.network_get_all(ctxt)
|
|
for network in networks:
|
|
self.network.set_network_host(ctxt, network)
|
|
|
|
self.network.allocate_for_instance(ctxt,
|
|
instance_id=2,
|
|
host=FLAGS.host,
|
|
vpn=None,
|
|
instance_type_id=1,
|
|
project_id=self.project_id)
|
|
self._test_spawn(glance_stubs.FakeGlance.IMAGE_MACHINE,
|
|
glance_stubs.FakeGlance.IMAGE_KERNEL,
|
|
glance_stubs.FakeGlance.IMAGE_RAMDISK,
|
|
instance_id=2,
|
|
create_record=False)
|
|
# TODO(salvatore-orlando): a complete test here would require
|
|
# a check for making sure the bridge for the VM's VIF is
|
|
# consistent with bridge specified in nova db
|
|
|
|
def test_spawn_with_network_qos(self):
|
|
self._create_instance()
|
|
for vif_ref in xenapi_fake.get_all('VIF'):
|
|
vif_rec = xenapi_fake.get_record('VIF', vif_ref)
|
|
self.assertEquals(vif_rec['qos_algorithm_type'], 'ratelimit')
|
|
self.assertEquals(vif_rec['qos_algorithm_params']['kbps'],
|
|
str(3 * 1024))
|
|
|
|
def test_rescue(self):
|
|
self.flags(flat_injected=False)
|
|
instance = self._create_instance()
|
|
conn = xenapi_conn.get_connection(False)
|
|
conn.rescue(self.context, instance, None, [])
|
|
|
|
def test_unrescue(self):
|
|
instance = self._create_instance()
|
|
conn = xenapi_conn.get_connection(False)
|
|
# Ensure that it will not unrescue a non-rescued instance.
|
|
self.assertRaises(Exception, conn.unrescue, instance, None)
|
|
|
|
def _create_instance(self, instance_id=1, spawn=True):
|
|
"""Creates and spawns a test instance."""
|
|
stubs.stubout_loopingcall_start(self.stubs)
|
|
values = {
|
|
'id': instance_id,
|
|
'project_id': self.project_id,
|
|
'user_id': self.user_id,
|
|
'image_ref': 1,
|
|
'kernel_id': 2,
|
|
'ramdisk_id': 3,
|
|
'instance_type_id': '3', # m1.large
|
|
'os_type': 'linux',
|
|
'architecture': 'x86-64'}
|
|
instance = db.instance_create(self.context, values)
|
|
network_info = [({'bridge': 'fa0', 'id': 0, 'injected': False},
|
|
{'broadcast': '192.168.0.255',
|
|
'dns': ['192.168.0.1'],
|
|
'gateway': '192.168.0.1',
|
|
'gateway6': 'dead:beef::1',
|
|
'ip6s': [{'enabled': '1',
|
|
'ip': 'dead:beef::dcad:beff:feef:0',
|
|
'netmask': '64'}],
|
|
'ips': [{'enabled': '1',
|
|
'ip': '192.168.0.100',
|
|
'netmask': '255.255.255.0'}],
|
|
'label': 'fake',
|
|
'mac': 'DE:AD:BE:EF:00:00',
|
|
'rxtx_cap': 3})]
|
|
if spawn:
|
|
self.conn.spawn(self.context, instance, network_info)
|
|
return instance
|
|
|
|
|
|
class XenAPIDiffieHellmanTestCase(test.TestCase):
|
|
"""Unit tests for Diffie-Hellman code."""
|
|
def setUp(self):
|
|
super(XenAPIDiffieHellmanTestCase, self).setUp()
|
|
self.alice = vmops.SimpleDH()
|
|
self.bob = vmops.SimpleDH()
|
|
|
|
def test_shared(self):
|
|
alice_pub = self.alice.get_public()
|
|
bob_pub = self.bob.get_public()
|
|
alice_shared = self.alice.compute_shared(bob_pub)
|
|
bob_shared = self.bob.compute_shared(alice_pub)
|
|
self.assertEquals(alice_shared, bob_shared)
|
|
|
|
def _test_encryption(self, message):
|
|
enc = self.alice.encrypt(message)
|
|
self.assertFalse(enc.endswith('\n'))
|
|
dec = self.bob.decrypt(enc)
|
|
self.assertEquals(dec, message)
|
|
|
|
def test_encrypt_simple_message(self):
|
|
self._test_encryption('This is a simple message.')
|
|
|
|
def test_encrypt_message_with_newlines_at_end(self):
|
|
self._test_encryption('This message has a newline at the end.\n')
|
|
|
|
def test_encrypt_many_newlines_at_end(self):
|
|
self._test_encryption('Message with lotsa newlines.\n\n\n')
|
|
|
|
def test_encrypt_newlines_inside_message(self):
|
|
self._test_encryption('Message\nwith\ninterior\nnewlines.')
|
|
|
|
def test_encrypt_with_leading_newlines(self):
|
|
self._test_encryption('\n\nMessage with leading newlines.')
|
|
|
|
def test_encrypt_really_long_message(self):
|
|
self._test_encryption(''.join(['abcd' for i in xrange(1024)]))
|
|
|
|
def tearDown(self):
|
|
super(XenAPIDiffieHellmanTestCase, self).tearDown()
|
|
|
|
|
|
class XenAPIMigrateInstance(test.TestCase):
|
|
"""Unit test for verifying migration-related actions."""
|
|
|
|
def setUp(self):
|
|
super(XenAPIMigrateInstance, self).setUp()
|
|
self.stubs = stubout.StubOutForTesting()
|
|
self.flags(target_host='127.0.0.1',
|
|
xenapi_connection_url='test_url',
|
|
xenapi_connection_password='test_pass')
|
|
db_fakes.stub_out_db_instance_api(self.stubs)
|
|
stubs.stub_out_get_target(self.stubs)
|
|
xenapi_fake.reset()
|
|
xenapi_fake.create_network('fake', FLAGS.flat_network_bridge)
|
|
self.user_id = 'fake'
|
|
self.project_id = 'fake'
|
|
self.context = context.RequestContext(self.user_id, self.project_id)
|
|
self.values = {'id': 1,
|
|
'project_id': self.project_id,
|
|
'user_id': self.user_id,
|
|
'image_ref': 1,
|
|
'kernel_id': None,
|
|
'ramdisk_id': None,
|
|
'local_gb': 5,
|
|
'instance_type_id': '3', # m1.large
|
|
'os_type': 'linux',
|
|
'architecture': 'x86-64'}
|
|
|
|
fake_utils.stub_out_utils_execute(self.stubs)
|
|
stubs.stub_out_migration_methods(self.stubs)
|
|
stubs.stubout_get_this_vm_uuid(self.stubs)
|
|
glance_stubs.stubout_glance_client(self.stubs)
|
|
|
|
def test_migrate_disk_and_power_off(self):
|
|
instance = db.instance_create(self.context, self.values)
|
|
stubs.stubout_session(self.stubs, stubs.FakeSessionForMigrationTests)
|
|
conn = xenapi_conn.get_connection(False)
|
|
conn.migrate_disk_and_power_off(instance, '127.0.0.1')
|
|
|
|
def test_finish_migrate(self):
|
|
instance = db.instance_create(self.context, self.values)
|
|
self.called = False
|
|
self.fake_vm_start_called = False
|
|
|
|
def fake_vm_start(*args, **kwargs):
|
|
self.fake_vm_start_called = True
|
|
|
|
def fake_vdi_resize(*args, **kwargs):
|
|
self.called = True
|
|
|
|
self.stubs.Set(stubs.FakeSessionForMigrationTests,
|
|
"VDI_resize_online", fake_vdi_resize)
|
|
self.stubs.Set(vmops.VMOps, '_start', fake_vm_start)
|
|
|
|
stubs.stubout_session(self.stubs, stubs.FakeSessionForMigrationTests)
|
|
stubs.stubout_loopingcall_start(self.stubs)
|
|
conn = xenapi_conn.get_connection(False)
|
|
network_info = [({'bridge': 'fa0', 'id': 0, 'injected': False},
|
|
{'broadcast': '192.168.0.255',
|
|
'dns': ['192.168.0.1'],
|
|
'gateway': '192.168.0.1',
|
|
'gateway6': 'dead:beef::1',
|
|
'ip6s': [{'enabled': '1',
|
|
'ip': 'dead:beef::dcad:beff:feef:0',
|
|
'netmask': '64'}],
|
|
'ips': [{'enabled': '1',
|
|
'ip': '192.168.0.100',
|
|
'netmask': '255.255.255.0'}],
|
|
'label': 'fake',
|
|
'mac': 'DE:AD:BE:EF:00:00',
|
|
'rxtx_cap': 3})]
|
|
conn.finish_migration(self.context, instance,
|
|
dict(base_copy='hurr', cow='durr'),
|
|
network_info, resize_instance=True)
|
|
self.assertEqual(self.called, True)
|
|
self.assertEqual(self.fake_vm_start_called, True)
|
|
|
|
def test_finish_migrate_no_local_storage(self):
|
|
tiny_type_id = \
|
|
instance_types.get_instance_type_by_name('m1.tiny')['id']
|
|
self.values.update({'instance_type_id': tiny_type_id, 'local_gb': 0})
|
|
instance = db.instance_create(self.context, self.values)
|
|
|
|
def fake_vdi_resize(*args, **kwargs):
|
|
raise Exception("This shouldn't be called")
|
|
|
|
self.stubs.Set(stubs.FakeSessionForMigrationTests,
|
|
"VDI_resize_online", fake_vdi_resize)
|
|
stubs.stubout_session(self.stubs, stubs.FakeSessionForMigrationTests)
|
|
stubs.stubout_loopingcall_start(self.stubs)
|
|
conn = xenapi_conn.get_connection(False)
|
|
network_info = [({'bridge': 'fa0', 'id': 0, 'injected': False},
|
|
{'broadcast': '192.168.0.255',
|
|
'dns': ['192.168.0.1'],
|
|
'gateway': '192.168.0.1',
|
|
'gateway6': 'dead:beef::1',
|
|
'ip6s': [{'enabled': '1',
|
|
'ip': 'dead:beef::dcad:beff:feef:0',
|
|
'netmask': '64'}],
|
|
'ips': [{'enabled': '1',
|
|
'ip': '192.168.0.100',
|
|
'netmask': '255.255.255.0'}],
|
|
'label': 'fake',
|
|
'mac': 'DE:AD:BE:EF:00:00',
|
|
'rxtx_cap': 3})]
|
|
conn.finish_migration(self.context, instance,
|
|
dict(base_copy='hurr', cow='durr'),
|
|
network_info, resize_instance=True)
|
|
|
|
def test_finish_migrate_no_resize_vdi(self):
|
|
instance = db.instance_create(self.context, self.values)
|
|
|
|
def fake_vdi_resize(*args, **kwargs):
|
|
raise Exception("This shouldn't be called")
|
|
|
|
self.stubs.Set(stubs.FakeSessionForMigrationTests,
|
|
"VDI_resize_online", fake_vdi_resize)
|
|
stubs.stubout_session(self.stubs, stubs.FakeSessionForMigrationTests)
|
|
stubs.stubout_loopingcall_start(self.stubs)
|
|
conn = xenapi_conn.get_connection(False)
|
|
network_info = [({'bridge': 'fa0', 'id': 0, 'injected': False},
|
|
{'broadcast': '192.168.0.255',
|
|
'dns': ['192.168.0.1'],
|
|
'gateway': '192.168.0.1',
|
|
'gateway6': 'dead:beef::1',
|
|
'ip6s': [{'enabled': '1',
|
|
'ip': 'dead:beef::dcad:beff:feef:0',
|
|
'netmask': '64'}],
|
|
'ips': [{'enabled': '1',
|
|
'ip': '192.168.0.100',
|
|
'netmask': '255.255.255.0'}],
|
|
'label': 'fake',
|
|
'mac': 'DE:AD:BE:EF:00:00',
|
|
'rxtx_cap': 3})]
|
|
|
|
# Resize instance would be determined by the compute call
|
|
conn.finish_migration(self.context, instance,
|
|
dict(base_copy='hurr', cow='durr'),
|
|
network_info, resize_instance=False)
|
|
|
|
|
|
class XenAPIImageTypeTestCase(test.TestCase):
|
|
"""Test ImageType class."""
|
|
|
|
def test_to_string(self):
|
|
"""Can convert from type id to type string."""
|
|
self.assertEquals(
|
|
vm_utils.ImageType.to_string(vm_utils.ImageType.KERNEL),
|
|
vm_utils.ImageType.KERNEL_STR)
|
|
|
|
def test_from_string(self):
|
|
"""Can convert from string to type id."""
|
|
self.assertEquals(
|
|
vm_utils.ImageType.from_string(vm_utils.ImageType.KERNEL_STR),
|
|
vm_utils.ImageType.KERNEL)
|
|
|
|
|
|
class XenAPIDetermineDiskImageTestCase(test.TestCase):
|
|
"""Unit tests for code that detects the ImageType."""
|
|
def setUp(self):
|
|
super(XenAPIDetermineDiskImageTestCase, self).setUp()
|
|
glance_stubs.stubout_glance_client(self.stubs)
|
|
|
|
class FakeInstance(object):
|
|
pass
|
|
|
|
self.fake_instance = FakeInstance()
|
|
self.fake_instance.id = 42
|
|
self.fake_instance.os_type = 'linux'
|
|
self.fake_instance.architecture = 'x86-64'
|
|
|
|
def assert_disk_type(self, disk_type):
|
|
dt = vm_utils.VMHelper.determine_disk_image_type(
|
|
self.fake_instance)
|
|
self.assertEqual(disk_type, dt)
|
|
|
|
def test_instance_disk(self):
|
|
"""If a kernel is specified, the image type is DISK (aka machine)."""
|
|
self.fake_instance.image_ref = glance_stubs.FakeGlance.IMAGE_MACHINE
|
|
self.fake_instance.kernel_id = glance_stubs.FakeGlance.IMAGE_KERNEL
|
|
self.assert_disk_type(vm_utils.ImageType.DISK)
|
|
|
|
def test_instance_disk_raw(self):
|
|
"""
|
|
If the kernel isn't specified, and we're not using Glance, then
|
|
DISK_RAW is assumed.
|
|
"""
|
|
self.fake_instance.image_ref = glance_stubs.FakeGlance.IMAGE_RAW
|
|
self.fake_instance.kernel_id = None
|
|
self.assert_disk_type(vm_utils.ImageType.DISK_RAW)
|
|
|
|
def test_glance_disk_raw(self):
|
|
"""
|
|
If we're using Glance, then defer to the image_type field, which in
|
|
this case will be 'raw'.
|
|
"""
|
|
self.fake_instance.image_ref = glance_stubs.FakeGlance.IMAGE_RAW
|
|
self.fake_instance.kernel_id = None
|
|
self.assert_disk_type(vm_utils.ImageType.DISK_RAW)
|
|
|
|
def test_glance_disk_vhd(self):
|
|
"""
|
|
If we're using Glance, then defer to the image_type field, which in
|
|
this case will be 'vhd'.
|
|
"""
|
|
self.fake_instance.image_ref = glance_stubs.FakeGlance.IMAGE_VHD
|
|
self.fake_instance.kernel_id = None
|
|
self.assert_disk_type(vm_utils.ImageType.DISK_VHD)
|
|
|
|
|
|
class CompareVersionTestCase(test.TestCase):
|
|
def test_less_than(self):
|
|
"""Test that cmp_version compares a as less than b"""
|
|
self.assertTrue(vmops.cmp_version('1.2.3.4', '1.2.3.5') < 0)
|
|
|
|
def test_greater_than(self):
|
|
"""Test that cmp_version compares a as greater than b"""
|
|
self.assertTrue(vmops.cmp_version('1.2.3.5', '1.2.3.4') > 0)
|
|
|
|
def test_equal(self):
|
|
"""Test that cmp_version compares a as equal to b"""
|
|
self.assertTrue(vmops.cmp_version('1.2.3.4', '1.2.3.4') == 0)
|
|
|
|
def test_non_lexical(self):
|
|
"""Test that cmp_version compares non-lexically"""
|
|
self.assertTrue(vmops.cmp_version('1.2.3.10', '1.2.3.4') > 0)
|
|
|
|
def test_length(self):
|
|
"""Test that cmp_version compares by length as last resort"""
|
|
self.assertTrue(vmops.cmp_version('1.2.3', '1.2.3.4') < 0)
|
|
|
|
|
|
class FakeXenApi(object):
|
|
"""Fake XenApi for testing HostState."""
|
|
|
|
class FakeSR(object):
|
|
def get_record(self, ref):
|
|
return {'virtual_allocation': 10000,
|
|
'physical_utilisation': 20000}
|
|
|
|
SR = FakeSR()
|
|
|
|
|
|
class FakeSession(object):
|
|
"""Fake Session class for HostState testing."""
|
|
|
|
def async_call_plugin(self, *args):
|
|
return None
|
|
|
|
def wait_for_task(self, *args):
|
|
vm = {'total': 10,
|
|
'overhead': 20,
|
|
'free': 30,
|
|
'free-computed': 40}
|
|
return json.dumps({'host_memory': vm})
|
|
|
|
def get_xenapi(self):
|
|
return FakeXenApi()
|
|
|
|
|
|
class HostStateTestCase(test.TestCase):
|
|
"""Tests HostState, which holds metrics from XenServer that get
|
|
reported back to the Schedulers."""
|
|
|
|
def _fake_safe_find_sr(self, session):
|
|
"""None SR ref since we're ignoring it in FakeSR."""
|
|
return None
|
|
|
|
def test_host_state(self):
|
|
self.stubs = stubout.StubOutForTesting()
|
|
self.stubs.Set(vm_utils, 'safe_find_sr', self._fake_safe_find_sr)
|
|
host_state = xenapi_conn.HostState(FakeSession())
|
|
stats = host_state._stats
|
|
self.assertEquals(stats['disk_total'], 10000)
|
|
self.assertEquals(stats['disk_used'], 20000)
|
|
self.assertEquals(stats['host_memory_total'], 10)
|
|
self.assertEquals(stats['host_memory_overhead'], 20)
|
|
self.assertEquals(stats['host_memory_free'], 30)
|
|
self.assertEquals(stats['host_memory_free_computed'], 40)
|