blob: e3386afba8b709101aa73321bf84396d2fe2c535 (
plain)
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
|
import os
import os.path
import sys
import tempfile
import shutil
ipd = tempfile.mkdtemp(prefix = 'codecheck-aux')
try:
# add a fake ipaddr module, we don't want to
# deal with the real one (just test our code)
f = open(os.path.join(ipd, 'ipaddr.py'), 'w')
f.write("""
class IPAddress(object):
pass
class IPNetwork(list):
pass
""")
f.close()
sys.path.append(ipd)
fl = os.listdir(os.path.dirname(sys.argv[0]) or '.')
fl.sort()
for f in fl:
if f[-3:] != '.py' or f[0] == '_':
continue
m = f[:-3]
sys.stdout.write('importing %s ...' % m)
__import__(m)
print(' OK.')
def sys_argv_set(a):
sys.argv = sys.argv[:1] + a
gsyncd = sys.modules['gsyncd']
for a in [['--help'], ['--version'], ['--canonicalize-escape-url', '/foo']]:
print('>>> invoking program with args: %s' % ' '.join(a))
pid = os.fork()
if not pid:
sys_argv_set(a)
gsyncd.main()
_, r = os.waitpid(pid, 0)
if r:
raise RuntimeError('invocation failed')
finally:
shutil.rmtree(ipd)
|
