Change how libvirt firewall drivers work to have meaningful flags.

1 files changed, 6 insertions, 1 deletions

diff --git a/nova/tests/test_virt.py b/nova/tests/test_virt.py
index 0b9b847a0..1760b73ab 100644
--- a/nova/tests/test_virt.py
+++ b/nova/tests/test_virt.py
@@ -221,7 +221,12 @@ class IptablesFirewallTestCase(test.TestCase):
         self.project = self.manager.create_project('fake', 'fake', 'fake')
         self.context = context.RequestContext('fake', 'fake')
         self.network = utils.import_object(FLAGS.network_manager)
-        self.fw = libvirt_conn.IptablesFirewallDriver()
+
+        class Mock(object):
+            pass
+        self.fake_libvirt_connection = Mock()
+        self.fw = libvirt_conn.IptablesFirewallDriver(
+                      get_connection=lambda: self.fake_libvirt_connection)
 
     def tearDown(self):
         self.manager.delete_project(self.project)