from nitsi.logger import init_logging, Log_Formatter
from nitsi.recipe import RecipeExeption
-from nitsi.test import TestException, test
+import nitsi.test
logger = logging.getLogger("nitsi")
# here we run a test
try:
- currenttest = test(args.dir, log_dir)
+ currenttest = nitsi.test.Test(args.dir, log_dir)
currenttest.read_settings()
currenttest.virtual_environ_setup()
currenttest.load_recipe()
try:
currenttest.virtual_environ_start()
currenttest.run_recipe()
- except TestException as e:
+ except nitsi.test.TestException as e:
logger.exception(e)
return 1
except BaseException as e:
logger = logging.getLogger("nitsi.disk")
-class disk():
+class Disk():
def __init__(self, disk):
self.log = logger.getChild(os.path.basename(disk))
self.log.debug("Initiated a disk class for {}".format(disk))
logger = logging.getLogger("nitsi.machine")
-class machine():
+class Machine():
def __init__(self, libvirt_con, vm_xml_file, snapshot_xml_file, image, root_uid, username, password):
self.log = logger.getChild(os.path.basename(vm_xml_file))
self.con = libvirt_con
self.log.error("No such file: {}".format(self.image))
self.root_uid = root_uid
- self.disk = disk.disk(image)
+ self.disk = disk.Disk(image)
self.username = username
self.password = password
return elem.text
def check_is_booted_up(self):
- serial_con = serial_connection.serial_connection(self.get_serial_device())
+ serial_con = serial_connection.Serial_connection(self.get_serial_device())
serial_con.write("\n")
# This will block till the domain is booted up
def login(self, log_file, log_start_time=None, longest_machine_name=10):
try:
- self.serial_con = serial_connection.serial_connection(self.get_serial_device(),
+ self.serial_con = serial_connection.Serial_connection(self.get_serial_device(),
username=self.username,
log_file=log_file,
log_start_time=log_start_time,
logger = logging.getLogger("nitsi.network")
# # A class which define and undefine a virtual network based on an xml file
-class network():
+class Network():
def __init__(self, libvirt_con, network_xml_file):
self.log = logger.getChild(os.path.basename(os.path.dirname(network_xml_file)))
self.con = libvirt_con
# Should read the test, check if the syntax are valid
# and return tuples with the ( host, command ) structure
-class recipe():
+class Recipe():
def __init__(self, path, circle=[], machines=[]):
self.recipe_file = path
try:
self.log.error("Detect import loop!")
raise RecipeExeption("Detect import loop!")
self.circle.append(path)
- recipe_to_include = recipe(path, circle=self.circle)
+ recipe_to_include = Recipe(path, circle=self.circle)
if machine == "include":
self._recipe.extend(recipe_to_include.recipe)
log = logging.getLogger("nitsi.serial")
-class serial_connection():
+class Serial_connection():
def __init__(self, device, username=None, log_file=None, name=None, log_start_time=None, longest_machine_name=10):
self.buffer = b""
self.back_at_prompt_pattern = None
def __init__(self, message):
self.message = message
-class test():
+class Test():
def __init__(self, path, log_path):
try:
self.path = os.path.abspath(path)
self.virtual_environ_path = os.path.normpath(self.path + "/" + self.virtual_environ_path)
def virtual_environ_setup(self):
- self.virtual_environ = virtual_environ.virtual_environ(self.virtual_environ_path)
+ self.virtual_environ = virtual_environ.Virtual_environ(self.virtual_environ_path)
self.virtual_networks = self.virtual_environ.get_networks()
def load_recipe(self):
self.log.info("Going to load the recipe")
try:
- self.recipe = recipe.recipe(self.recipe_file, machines=self.virtual_environ.machine_names)
+ self.recipe = recipe.Recipe(self.recipe_file, machines=self.virtual_environ.machine_names)
for line in self.recipe.recipe:
self.log.debug(line)
# Should return all vms and networks in a list
# and should provide the path to the necessary xml files
-class virtual_environ():
+class Virtual_environ():
def __init__(self, path):
self.log = logger.getChild(os.path.basename(os.path.abspath(path)))
try:
networks = {}
for _network in self.networks:
self.log.debug(_network)
- networks.setdefault(_network, network.network(self.con, os.path.normpath(self.path + "/" + self.config[_network]["xml_file"])))
+ networks.setdefault(_network, network.Network(self.con, os.path.normpath(self.path + "/" + self.config[_network]["xml_file"])))
return networks
def get_machines(self):
machines = {}
for _machine in self.machines:
self.log.debug(_machine)
- machines.setdefault(_machine, machine.machine(
+ machines.setdefault(_machine, machine.Machine(
self.con,
os.path.normpath(self.path + "/" + self.config[_machine]["xml_file"]),
os.path.normpath(self.path + "/" + self.config[_machine]["snapshot_xml_file"]),