Allows test_virt_drivers to work when run alone

* Properly sets fake_resize using a stub in test_libvirt
 * Adds a stub for test_virt_drivers
 * Fixes typo and misuse of flags in setUp and TearDown
 * Fixes bug 925891

Change-Id: I015e942305f6a08941168ca0ccadb2382db064d7
This commit is contained in:
Vishvananda Ishaya
2012-02-02 23:14:29 -08:00
parent 4064930eb3
commit 390b4ecc71
2 changed files with 12 additions and 10 deletions

View File

@@ -35,7 +35,6 @@ from nova import utils
from nova.api.ec2 import cloud from nova.api.ec2 import cloud
from nova.compute import power_state from nova.compute import power_state
from nova.compute import vm_states from nova.compute import vm_states
from nova.virt.disk import api as disk
from nova.virt import images from nova.virt import images
from nova.virt import driver from nova.virt import driver
from nova.virt.libvirt import connection from nova.virt.libvirt import connection
@@ -216,12 +215,11 @@ class CacheConcurrencyTestCase(test.TestCase):
self.stubs.Set(os.path, 'exists', fake_exists) self.stubs.Set(os.path, 'exists', fake_exists)
self.stubs.Set(utils, 'execute', fake_execute) self.stubs.Set(utils, 'execute', fake_execute)
self.stubs.Set(connection.disk, 'extend', fake_extend)
connection.libvirt_utils = fake_libvirt_utils connection.libvirt_utils = fake_libvirt_utils
connection.disk.extend = fake_extend
def tearDown(self): def tearDown(self):
connection.libvirt_utils = libvirt_utils connection.libvirt_utils = libvirt_utils
connection.disk.extend = disk.extend
super(CacheConcurrencyTestCase, self).tearDown() super(CacheConcurrencyTestCase, self).tearDown()
def test_same_fname_concurrency(self): def test_same_fname_concurrency(self):
@@ -301,11 +299,10 @@ class LibvirtConnTestCase(test.TestCase):
def fake_extend(image, size): def fake_extend(image, size):
pass pass
connection.disk.extend = fake_extend self.stubs.Set(connection.disk, 'extend', fake_extend)
def tearDown(self): def tearDown(self):
connection.libvirt_utils = libvirt_utils connection.libvirt_utils = libvirt_utils
connection.disk.extend = disk.extend
super(LibvirtConnTestCase, self).tearDown() super(LibvirtConnTestCase, self).tearDown()
test_instance = {'memory_kb': '1024000', test_instance = {'memory_kb': '1024000',

View File

@@ -444,13 +444,17 @@ class LibvirtConnTestCase(_VirtDriverTestCase):
self.driver_module = nova.virt.libvirt.connection self.driver_module = nova.virt.libvirt.connection
FLAGS.firewall_driver = nova.virt.libvirt.firewall.drivers[0] FLAGS.firewall_driver = nova.virt.libvirt.firewall.drivers[0]
super(LibvirtConnTestCase, self).setUp() super(LibvirtConnTestCase, self).setUp()
FLAGS.rescue_image_id = "2" self.flags(rescue_image_id="2",
FLAGS.rescue_kernel_id = "3" rescue_kernel_id="3",
FLAGS.rescue_ramdisk_id = None rescue_ramdisk_id=None)
def fake_extend(image, size):
pass
self.stubs.Set(nova.virt.libvirt.connection.disk,
'extend', fake_extend)
def tearDown(self): def tearDown(self):
super(LibvirtConnTestCase, self).tearDown()
# Restore libvirt # Restore libvirt
import nova.virt.libvirt.connection import nova.virt.libvirt.connection
import nova.virt.libvirt.firewall import nova.virt.libvirt.firewall
@@ -459,3 +463,4 @@ class LibvirtConnTestCase(_VirtDriverTestCase):
nova.virt.libvirt.connection.libvirt = self.saved_libvirt nova.virt.libvirt.connection.libvirt = self.saved_libvirt
nova.virt.libvirt.connection.libvirt_utils = self.saved_libvirt nova.virt.libvirt.connection.libvirt_utils = self.saved_libvirt
nova.virt.libvirt.firewall.libvirt = self.saved_libvirt nova.virt.libvirt.firewall.libvirt = self.saved_libvirt
super(LibvirtConnTestCase, self).tearDown()