summaryrefslogtreecommitdiffstats
path: root/geo-replication/syncdaemon/__codecheck.py
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)