3 # Copyright (C) 2014 Simo Sorce <simo@redhat.com>
5 # see file 'COPYING' for use and warranty information
7 # This program is free software; you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation, either version 3 of the License, or
10 # (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU General Public License for more details.
17 # You should have received a copy of the GNU General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
22 from ipsilon.util import plugin
36 (pathname, dummy) = os.path.split(inspect.getfile(Tests))
37 self.plugins = p.get_plugins(pathname, 'IpsilonTest')
41 parser = argparse.ArgumentParser(description='Ipsilon Tests Environment')
42 parser.add_argument('--path', default='%s/testdir' % os.getcwd(),
43 help="Directory in which tests are run")
44 parser.add_argument('--test', default='test1',
45 help="The test to run")
46 parser.add_argument('--wrappers', default='auto',
47 choices=['yes', 'no', 'auto'],
48 help="Run the tests with socket wrappers")
50 return vars(parser.parse_args())
53 def try_wrappers(base, wrappers):
57 pkgcfg = subprocess.Popen(['pkg-config', '--exists', 'socket_wrapper'])
59 if pkgcfg.returncode != 0:
60 if wrappers == 'auto':
63 raise ValueError('Socket Wrappers not available')
65 wrapdir = os.path.join(base, 'wrapdir')
68 wenv = {'LD_PRELOAD': 'libsocket_wrapper.so',
69 'SOCKET_WRAPPER_DIR': wrapdir,
70 'SOCKET_WRAPPER_DEFAULT_IFACE': '9'}
75 if __name__ == '__main__':
80 if args['test'] not in tests.plugins:
81 print >> sys.stderr, "Unknown test [%s]" % args['test']
83 test = tests.plugins[args['test']]
85 if not os.path.exists(args['path']):
86 os.makedirs(args['path'])
88 test.setup_base(args['path'], test)
90 env = try_wrappers(test.testdir, args['wrappers'])
91 env['PYTHONPATH'] = test.rootdir
94 test.setup_servers(env)
99 except Exception, e: # pylint: disable=broad-except
100 print >> sys.stderr, "Error: %s" % repr(e)
101 traceback.print_exc(None, sys.stderr)