From 0291266fb61dd398147fa2ae1a37366dd2fa869e Mon Sep 17 00:00:00 2001 From: Johannes Erdfelt Date: Mon, 1 Oct 2012 23:00:19 +0000 Subject: Add TestCase.stub_module to make stubbing modules easier Previously various test cases would modify sys.modules directly, duplicating code unnecessarily. Change-Id: I09d333d6daf527e34db0262b28edcb0323cb2855 --- nova/tests/network/test_manager.py | 4 +--- nova/tests/xenapi/stubs.py | 13 +------------ 2 files changed, 2 insertions(+), 15 deletions(-) (limited to 'nova/tests') diff --git a/nova/tests/network/test_manager.py b/nova/tests/network/test_manager.py index 2f597547c..c24bee2d8 100644 --- a/nova/tests/network/test_manager.py +++ b/nova/tests/network/test_manager.py @@ -1846,8 +1846,7 @@ class LdapDNSTestCase(test.TestCase): def setUp(self): super(LdapDNSTestCase, self).setUp() - self.saved_ldap = sys.modules.get('ldap') - sys.modules['ldap'] = fake_ldap + self.stub_module('ldap', fake_ldap) self.driver = FakeLdapDNS() self.driver.create_domain(domain1) @@ -1856,7 +1855,6 @@ class LdapDNSTestCase(test.TestCase): def tearDown(self): self.driver.delete_domain(domain1) self.driver.delete_domain(domain2) - sys.modules['ldap'] = self.saved_ldap super(LdapDNSTestCase, self).tearDown() def test_ldap_dns_domains(self): diff --git a/nova/tests/xenapi/stubs.py b/nova/tests/xenapi/stubs.py index 35aa8e4aa..cde17a7af 100644 --- a/nova/tests/xenapi/stubs.py +++ b/nova/tests/xenapi/stubs.py @@ -18,7 +18,6 @@ import contextlib import pickle import random -import sys from nova.openstack.common import jsonutils from nova import test @@ -355,16 +354,6 @@ class XenAPITestBase(test.TestCase): def setUp(self): super(XenAPITestBase, self).setUp() - self.orig_XenAPI = sys.modules.get('XenAPI') - sys.modules['XenAPI'] = fake + self.stub_module('XenAPI', fake) fake.reset() - - def tearDown(self): - if self.orig_XenAPI is not None: - sys.modules['XenAPI'] = self.orig_XenAPI - self.orig_XenAPI = None - else: - sys.modules.pop('XenAPI') - - super(XenAPITestBase, self).tearDown() -- cgit