Update example test settings

[nitsi.git] / network.py

diff --git a/network.py b/network.py
index 223930becc20006436949bd7698f5f514db4164d..0e76274a55d1c5884e25fa789ecd23ca40f96f91 100644 (file)
--- a/network.py
+++ b/network.py
@@ -2,9 +2,9 @@
 
 # # A class which define and undefine a virtual network based on an xml file
 class network():
-    def __init__(self, network_xml_file):
+    def __init__(self, libvirt_con, network_xml_file):
         self.log = log(4)
-        self.con = libvirt_con("qemu:///system")
+        self.con = libvirt_con
         try:
             with open(network_xml_file) as fobj:
                 self.network_xml = fobj.read()
@@ -12,7 +12,7 @@ class network():
             self.log.error("No such file: {}".format(vm_xml_file))
 
     def define(self):
-        self.network = self.con.con.networkDefineXML(self.network_xml)
+        self.network = self.con.networkDefineXML(self.network_xml)
 
         if network == None:
             self.log.error("Failed to define virtual network")