summaryrefslogtreecommitdiffstats
path: root/odl-pipeline/lib/test_environment
diff options
context:
space:
mode:
authorNikolas Hermanns <nikolas.hermanns@ericsson.com>2016-12-13 10:26:11 +0000
committerGerrit Code Review <gerrit@opnfv.org>2016-12-13 10:26:11 +0000
commit2dd53e9153a37a5a9f14a8772224dbdaa091bbfa (patch)
tree7e083f4959878d78f943c07cc32575c66a4ba05d /odl-pipeline/lib/test_environment
parent2445ba0a0325611e0b927ecc7ed4d361426e5fc4 (diff)
parent416f9cf177801cecfe39f5249d73e3a4a85f0bbd (diff)
Merge "adding odl-pipeline"
Diffstat (limited to 'odl-pipeline/lib/test_environment')
-rwxr-xr-xodl-pipeline/lib/test_environment/__init__.py0
-rwxr-xr-xodl-pipeline/lib/test_environment/test_environment.py131
2 files changed, 131 insertions, 0 deletions
diff --git a/odl-pipeline/lib/test_environment/__init__.py b/odl-pipeline/lib/test_environment/__init__.py
new file mode 100755
index 0000000..e69de29
--- /dev/null
+++ b/odl-pipeline/lib/test_environment/__init__.py
diff --git a/odl-pipeline/lib/test_environment/test_environment.py b/odl-pipeline/lib/test_environment/test_environment.py
new file mode 100755
index 0000000..481f38d
--- /dev/null
+++ b/odl-pipeline/lib/test_environment/test_environment.py
@@ -0,0 +1,131 @@
+#!/bin/python
+import os
+from utils.utils_log import LOG, for_all_methods, log_enter_exit
+from utils.service import Service
+from utils.processutils import execute
+from utils import utils_yaml
+from utils.shutil import shutil
+MAX_NODES = 5
+
+
+@for_all_methods(log_enter_exit)
+class TestEnvironment(Service):
+
+ NODE_NAME = 'baremetal'
+ TEMPLATES = '../templates'
+ BRIGES = ['admin', 'private', 'public', 'storage']
+
+ def run(self, sys_args, config):
+ self.BUILD_DIR = '../build/apex-%s' % sys_args.env_number
+ self.env = sys_args.env_number
+ self.cleanup()
+ if sys_args.cleanup:
+ return
+ if not sys_args.cloner_info or not sys_args.snapshot_disks:
+ LOG.error('--cloner-info, --snapshot-disks have to be given if not'
+ ' only cleanup.')
+ exit(1)
+ node_info = utils_yaml.read_dict_from_yaml(sys_args.cloner_info +
+ '/node.yaml')
+ nodes = node_info['servers']
+ number_of_nodes = len(nodes)
+ disk_home = self.BUILD_DIR + '/disks/'
+ shutil.mkdir_if_not_exsist(disk_home)
+ # Create Snapshots
+ for i in range(number_of_nodes):
+ disk_name = '%s%s.qcow2' % (self.NODE_NAME, i)
+ self.create_snapshot('%s/%s' % (sys_args.snapshot_disks,
+ disk_name),
+ '%s/%s' % (disk_home, disk_name))
+
+ # Create Bridges if not existing
+ for net in self.BRIGES:
+ bridge_name = '%s-%s' % (net, self.env)
+ if not self.check_if_br_exists(bridge_name):
+ LOG.info('Creating bridge %s' % bridge_name)
+ execute('ovs-vsctl add-br %s' % bridge_name, as_root=True)
+
+ # Create virtual Nodes
+ dom_template = self.TEMPLATES + '/nodes/baremetalX.xml'
+ dom_config = self.BUILD_DIR + '/nodes/baremetalX.xml'
+ shutil.mkdir_if_not_exsist(self.BUILD_DIR + '/nodes/')
+ LOG.info('Creating virtual Nodes')
+ for name, node in nodes.iteritems():
+ orig_node_name = node['vNode-name']
+ node_name = orig_node_name + '-' + self.env
+ LOG.info('Create node %s' % node_name)
+ type = node['type']
+ if type == 'compute':
+ cpu = 4
+ mem = 4
+ elif type == 'controller':
+ cpu = 8
+ mem = 10
+ else:
+ raise Exception('Unknown node type! %s' % type)
+ shutil.copy('to', dom_template, dom_config)
+ shutil.replace_string_in_file(dom_config, 'NaMe', node_name)
+ disk_full_path = os.path.abspath('%s/%s.qcow2' % (disk_home,
+ orig_node_name))
+ shutil.replace_string_in_file(dom_config, 'DiSk', disk_full_path)
+ shutil.replace_string_in_file(dom_config, 'vCpU', str(cpu))
+ shutil.replace_string_in_file(dom_config, 'MeMoRy', str(mem))
+ shutil.replace_string_in_file(dom_config, 'InDeX', self.env)
+
+ execute('virsh define ' + dom_config)
+ execute('virsh start ' + node_name)
+
+ cores_per_environment = 8
+ cores = '%s-%s' % (int(self.env) * 8, int(self.env) * 8 +
+ cores_per_environment - 1)
+ LOG.info('Pining vCPU of node %s to cores %s' % (node_name, cores))
+ for i in range(cpu):
+ execute('virsh vcpupin %(node)s %(nodes_cpu)s %(host_cpu)s' %
+ {'node': node_name,
+ 'nodes_cpu': i,
+ 'host_cpu': cores})
+
+ def check_if_br_exists(self, bridge):
+ _, (_, rc) = execute('ovs-vsctl br-exists %s' % bridge,
+ check_exit_code=[0, 2], as_root=True)
+ return True if rc == 0 else False
+
+ def create_snapshot(self, orig, path):
+ LOG.info('Creating snapshot of %s in %s' % (orig, path))
+ execute('qemu-img create -f qcow2 -b %s %s' % (orig, path),
+ as_root=True)
+
+ def cleanup(self):
+ for i in range(MAX_NODES):
+ rv, (_, rc) = execute('virsh destroy %(name)s%(i)s-%(env)s'
+ % {'i': i, 'env': self.env,
+ 'name': self.NODE_NAME},
+ check_exit_code=[0, 1])
+ if rc == 0:
+ LOG.info(rv)
+ rv, (_, rc) = execute('virsh undefine %(name)s%(i)s-%(env)s'
+ % {'i': i, 'env': self.env,
+ 'name': self.NODE_NAME},
+ check_exit_code=[0, 1])
+ if rc == 0:
+ LOG.info(rv)
+ execute('rm -rf ' + self.BUILD_DIR)
+
+ def create_cli_parser(self, parser):
+ parser.add_argument('--env-number', help="Number of the environment",
+ required=True)
+ parser.add_argument('--cloner-info', help="Path to the cloner-info",
+ required=False)
+ parser.add_argument('--snapshot-disks', help="Path to the snapshots",
+ required=False)
+ parser.add_argument('--cleanup', help="Only Cleanup",
+ required=False, action='store_true')
+ return parser
+
+
+def main():
+ main = TestEnvironment()
+ main.start()
+
+if __name__ == '__main__':
+ main()