1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#!/usr/bin/python3
# Copyright (C) 2014,2015,2016 - GSS-Proxy contributors; see COPYING for the license.
import argparse
import importlib
import signal
import traceback
from testlib import *
def parse_args():
parser = argparse.ArgumentParser(description='GSS-Proxy Tests Environment')
parser.add_argument('--path', default='%s/testdir' % os.getcwd(),
help="Directory in which tests are run")
return vars(parser.parse_args())
if __name__ == '__main__':
args = parse_args()
testdir = args['path']
if os.path.exists(testdir):
shutil.rmtree(testdir)
os.makedirs(testdir)
processes = dict()
try:
wrapenv = setup_wrappers(testdir)
ldapproc, ldapenv = setup_ldap(testdir, wrapenv)
processes["LDAP(%d)" % ldapproc.pid] = ldapproc
kdcproc, kdcenv = setup_kdc(testdir, wrapenv)
processes['KDC(%d)' % kdcproc.pid] = kdcproc
keysenv = setup_keys(testdir, kdcenv)
gssapienv = setup_gssapi_env(testdir, kdcenv)
gssproxylog = os.path.join(testdir, 'gssproxy.log')
logfile = open(gssproxylog, "a")
gssproxyenv = keysenv
gssproxyenv['KRB5_TRACE'] = os.path.join(testdir, 'gssproxy.trace')
gproc, gpsocket = setup_gssproxy(testdir, logfile, gssproxyenv)
time.sleep(5) #Give time to gssproxy to fully start up
processes['GSS-Proxy(%d)' % gproc.pid] = gproc
gssapienv['GSSPROXY_SOCKET'] = gpsocket
basicconf = {'svc_name': "host@%s" % WRAP_HOSTNAME,
'keytab': os.path.join(testdir, SVC_KTNAME)}
basicconf["gpid"] = gproc.pid
basicconf["keysenv"] = keysenv
testnum = 0
testfiles = [f for f in os.listdir("tests") \
if f.endswith(".py") and f.startswith("t_")]
testfiles.sort()
print("Tests to be run: " + ", ".join(testfiles))
for f in testfiles:
fmod = f[:-len(".py")]
t = importlib.__import__(fmod)
basicconf['prefix'] = '%02d' % testnum
logfile = os.path.join(testdir, "%02d_%s.log" % (testnum, fmod))
basicconf['logfile'] = open(logfile, 'a')
t.run(testdir, gssapienv, basicconf)
testnum += 1
except Exception:
traceback.print_exc()
finally:
for name in processes:
print("Killing %s" % name)
os.killpg(processes[name].pid, signal.SIGTERM)
|
