Reorder imports
[functest.git] / testcases / Controllers / ODL / OpenDaylightTesting.py
1 #!/usr/bin/python
2
3 import argparse
4 import fileinput
5 import os
6 import re
7 import shutil
8 import sys
9
10 from robot import run
11
12 import functest.utils.functest_logger as ft_logger
13
14
15 class ODLTestCases:
16
17     repos = "/home/opnfv/repos/"
18     odl_test_repo = repos + "odl_test/"
19     neutron_suite_dir = odl_test_repo + "csit/suites/openstack/neutron/"
20     basic_suite_dir = odl_test_repo + "csit/suites/integration/basic/"
21     logger = ft_logger.Logger("opendaylight").getLogger()
22
23     @classmethod
24     def copy_opnf_testcases(cls):
25         opnfv_testcases_dir = (os.path.dirname(os.path.abspath(__file__)) +
26                                "/custom_tests/neutron/")
27         files = [opnfv_testcases_dir + "001__reachability.robot",
28                  opnfv_testcases_dir + "040__delete_ports.robot",
29                  opnfv_testcases_dir + "050__delete_subnets.robot",
30                  opnfv_testcases_dir + "060__delete_networks.robot"]
31         for f in files:
32             try:
33                 shutil.copy(f, cls.neutron_suite_dir)
34             except IOError as e:
35                 cls.logger.error(
36                     "Cannot copy OPNFV's testcases to ODL directory", e)
37                 return False
38         return True
39
40     @classmethod
41     def set_robotframework_vars(cls, odlusername="admin", odlpassword="admin"):
42         odl_variables_files = cls.odl_test_repo + 'csit/variables/Variables.py'
43         try:
44             cls.logger.debug(cls.neutron_suite_dir + '__init__.robot')
45             for line in fileinput.input(odl_variables_files,
46                                         inplace=True):
47                 print re.sub("AUTH = .*",
48                              ("AUTH = [u'" + odlusername + "', u'" +
49                               odlpassword + "']"),
50                              line.rstrip())
51             return True
52         except Exception as e:
53             cls.logger.error("Cannot set ODL creds", e)
54             return False
55
56     @classmethod
57     def run(cls, **kwargs):
58         dirs = [cls.basic_suite_dir, cls.neutron_suite_dir]
59         try:
60             odlusername = kwargs['odlusername']
61             odlpassword = kwargs['odlpassword']
62             variables = ['KEYSTONE:' + kwargs['keystoneip'],
63                          'NEUTRON:' + kwargs['neutronip'],
64                          'OSUSERNAME:"' + kwargs['osusername'] + '"',
65                          'OSTENANTNAME:"' + kwargs['ostenantname'] + '"',
66                          'OSPASSWORD:"' + kwargs['ospassword'] + '"',
67                          'ODL_SYSTEM_IP:' + kwargs['odlip'],
68                          'PORT:' + kwargs['odlwebport'],
69                          'RESTCONFPORT:' + kwargs['odlrestconfport']]
70         except KeyError as e:
71             cls.logger.error("Cannot run ODL testcases. Please check", e)
72             return False
73         res_dir = '/home/opnfv/functest/results/odl/'
74         if (cls.copy_opnf_testcases() and
75                 cls.set_robotframework_vars(odlusername, odlpassword)):
76             try:
77                 os.makedirs(res_dir)
78             except OSError:
79                 pass
80             stdout_file = res_dir + 'stdout.txt'
81             with open(stdout_file, 'w') as stdout:
82                 result = run(*dirs, variable=variables,
83                              output=res_dir + 'output.xml',
84                              log=res_dir + 'log.html',
85                              report=res_dir + 'report.html',
86                              stdout=stdout)
87
88             with open(stdout_file, 'r') as stdout:
89                 cls.logger.info("\n" + stdout.read())
90
91             return result
92         else:
93             return False
94
95
96 if __name__ == '__main__':
97     parser = argparse.ArgumentParser()
98     parser.add_argument('-k', '--keystoneip',
99                         help='Keystone IP',
100                         default='127.0.0.1')
101     parser.add_argument('-n', '--neutronip',
102                         help='Neutron IP',
103                         default='127.0.0.1')
104     parser.add_argument('-a', '--osusername',
105                         help='Username for OpenStack',
106                         default='admin')
107     parser.add_argument('-b', '--ostenantname',
108                         help='Tenantname for OpenStack',
109                         default='admin')
110     parser.add_argument('-c', '--ospassword',
111                         help='Password for OpenStack',
112                         default='admin')
113     parser.add_argument('-o', '--odlip',
114                         help='OpenDaylight IP',
115                         default='127.0.0.1')
116     parser.add_argument('-w', '--odlwebport',
117                         help='OpenDaylight Web Portal Port',
118                         default='8080')
119     parser.add_argument('-r', '--odlrestconfport',
120                         help='OpenDaylight RESTConf Port',
121                         default='8181')
122     parser.add_argument('-d', '--odlusername',
123                         help='Username for ODL',
124                         default='admin')
125     parser.add_argument('-e', '--odlpassword',
126                         help='Password for ODL',
127                         default='admin')
128     args = vars(parser.parse_args())
129     sys.exit(ODLTestCases.run(**args))