test.py

   1 #!/usr/bin/python3
   2
   3
   4 import libvirt
   5
   6 import os
   7
   8 import configparser
   9
  10 from disk import disk
  11
  12 class log():
  13     def __init__(self, log_level):
  14         self.log_level = log_level
  15
  16     def debug(self, string):
  17         if self.log_level >= 4:
  18             print("DEBUG: {}".format(string))
  19
  20     def error(self, string):
  21         print("ERROR: {}".format(string))
  22
  23 class libvirt_con():
  24     def __init__(self, uri):
  25         self.log = log(4)
  26         self.uri = uri
  27         self.connection = None
  28
  29     def get_domain_from_name(self, name):
  30         dom = self.con.lookupByName(name)
  31
  32         if dom == None:
  33             raise BaseException
  34         return dom
  35
  36     @property
  37     def con(self):
  38         if self.connection == None:
  39             try:
  40                 self.connection = libvirt.open(self.uri)
  41             except BaseException as error:
  42                 self.log.error("Could not connect to: {}".format(self.uri))
  43
  44             self.log.debug("Connected to: {}".format(self.uri))
  45             return self.connection
  46
  47         return self.connection
  48
  49
  50
  51 class test():
  52     def __init__(self, path):
  53         self.log = log(4)
  54         try:
  55             self.path = os.path.abspath(path)
  56         except BaseException as e:
  57             self.log.error("Could not get absolute path")
  58
  59         self.log.debug(self.path)
  60
  61         self.settings_file = "{}/settings".format(self.path)
  62         if not os.path.isfile(self.settings_file):
  63             self.log.error("No such file: {}".format(self.settings_file))
  64
  65         self.recipe_file = "{}/recipe".format(self.path)
  66         if not os.path.isfile(self.recipe_file):
  67             self.log.error("No such file: {}".format(self.recipe_file))
  68
  69     def read_settings(self):
  70         self.config = configparser.ConfigParser()
  71         self.config.read(self.settings_file)
  72         self.name = self.config["DEFAULT"]["Name"]
  73         self.description = self.config["DEFAULT"]["Description"]
  74         self.copy_to = self.config["DEFAULT"]["Copy_to"]
  75         self.copy_from = self.config["DEFAULT"]["Copy_from"]
  76         self.copy_from = self.copy_from.split(",")
  77
  78         tmp = []
  79         for file in self.copy_from:
  80             file = file.strip()
  81             file = os.path.normpath(self.path + "/" + file)
  82             tmp.append(file)
  83
  84         self.copy_from = tmp
  85
  86         self.virtual_environ_name = self.config["VIRTUAL_ENVIRONMENT"]["Name"]
  87         self.virtual_environ_path = self.config["VIRTUAL_ENVIRONMENT"]["Path"]
  88         self.virtual_environ_path = os.path.normpath(self.path + "/" + self.virtual_environ_path)
  89
  90     def virtual_environ_setup(self):
  91         self.virtual_environ = virtual_environ(self.virtual_environ_path)
  92
  93         self.virtual_networks = self.virtual_environ.get_networks()
  94
  95         self.virtual_machines = self.virtual_environ.get_machines()
  96
  97     def virtual_environ_start(self):
  98         for name in self.virtual_environ.network_names:
  99             self.virtual_networks[name].define()
 100             self.virtual_networks[name].start()
 101
 102         for name in self.virtual_environ.machine_names:
 103             self.virtual_machines[name].define()
 104             self.virtual_machines[name].create_snapshot()
 105             self.virtual_machines[name].copy_in(self.copy_from, self.copy_to)
 106             self.virtual_machines[name].start()
 107
 108         self.log.debug("Try to login on all machines")
 109         for name in self.virtual_environ.machine_names:
 110             self.virtual_machines[name].login()
 111
 112     def load_recipe(self):
 113         try:
 114             self.recipe = recipe(self.recipe_file)
 115         except BaseException:
 116             self.log.error("Failed to load recipe")
 117             exit(1)
 118
 119     def run_recipe(self):
 120         for line in self.recipe.recipe:
 121             return_value = self.virtual_machines[line[0]].cmd(line[2])
 122             self.log.debug("Return value is: {}".format(return_value))
 123             if return_value != "0" and line[1] == "":
 124                 self.log.error("Failed to execute command '{}' on {}, return code: {}".format(line[2],line[0], return_value))
 125                 return False
 126             elif return_value == "0" and line[1] == "!":
 127                 self.log.error("Succeded to execute command '{}' on {}, return code: {}".format(line[2],line[0],return_value))
 128                 return False
 129             else:
 130                 self.log.debug("Command '{}' on {} returned with: {}".format(line[2],line[0],return_value))
 131
 132     def virtual_environ_stop(self):
 133         for name in self.virtual_environ.machine_names:
 134             self.virtual_machines[name].shutdown()
 135             self.virtual_machines[name].revert_snapshot()
 136             self.virtual_machines[name].undefine()
 137
 138         for name in self.virtual_environ.network_names:
 139             self.virtual_networks[name].undefine()
 140
 141
 142 if __name__ == "__main__":
 143     import argparse
 144
 145     parser = argparse.ArgumentParser()
 146
 147     parser.add_argument("-d", "--directory", dest="dir")
 148
 149     args = parser.parse_args()
 150
 151     currenttest = test(args.dir)
 152     currenttest.read_settings()
 153     currenttest.virtual_environ_setup()
 154     currenttest.load_recipe()
 155     try:
 156         currenttest.virtual_environ_start()
 157         currenttest.run_recipe()
 158     except BaseException as e:
 159         print(e)
 160     finally:
 161         currenttest.virtual_environ_stop()
 162