vmcelery: make vmdriver importable with LIBVIRT_TEST option

vmcelery.py

@@ -5,13 +5,20 @@ from os import getenv
 
 
 from argparse import ArgumentParser
 from argparse import ArgumentParser
 
 
-parser = ArgumentParser()
-parser.add_argument("-n", "--hostname", dest="hostname",
+
+def to_bool(value):
+    return value.lower() in ("true", "yes", "y", "t")
+
+if to_bool(getenv("LIBVIRT_TEST", "False")):
+    HOSTNAME = "vmdriver.test"
+else:
+    parser = ArgumentParser()
+    parser.add_argument("-n", "--hostname", dest="hostname",
                         help="Define the full queue name with"
                         help="Define the full queue name with"
                         "with priority", metavar="hostname.queue.priority")
                         "with priority", metavar="hostname.queue.priority")
-(args, unknwon_args) = parser.parse_known_args()
-HOSTNAME = vars(args).pop("hostname")
-if HOSTNAME is None:
+    (args, unknwon_args) = parser.parse_known_args()
+    HOSTNAME = vars(args).pop("hostname")
+    if HOSTNAME is None:
         raise Exception("You must define hostname as -n <hostname> or "
         raise Exception("You must define hostname as -n <hostname> or "
                         "--hostname=<hostname>.\n"
                         "--hostname=<hostname>.\n"
                         "Hostname format must be hostname.module.priority.")
                         "Hostname format must be hostname.module.priority.")
@@ -20,9 +27,6 @@ AMQP_URI = getenv('AMQP_URI')
 CACHE_URI = getenv('CACHE_URI')
 CACHE_URI = getenv('CACHE_URI')
 
 
 
 
-def to_bool(value):
-    return value.lower() in ("true", "yes", "y", "t")
-
 # Global configuration parameters declaration
 # Global configuration parameters declaration
 lib_connection = None
 lib_connection = None
 native_ovs = False
 native_ovs = False