summaryrefslogtreecommitdiffstats
path: root/tests/integration/saml2/__init__.py
blob: 50fcba0e6e23b9b186938a78b2e94a6e2667052f (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
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
import sys
import os
import signal
import subprocess
import time
import twill
import urllib2
import os.path
import re

CONFIG_FILE = os.path.expanduser('~/.config/lasso_integration.conf')
CONFIG = dict()

if os.path.exists(CONFIG_FILE):
    lines = open(CONFIG_FILE).read().splitlines()
    i = 1
    for line in lines:
        try:
            m = re.match('(\w*) = (.*)', line)
            CONFIG[m.groups()[0]] = m.groups()[1]
        except:
            print "Line", i, " of configuration file", CONFIG_FILE, "is invalid:", line
        i +=1

# Combine default and configuration file
AUTHENTIC_SRCDIR = CONFIG.get('AUTHENTIC_SRCDIR') or '/usr/local/src/authentic'
AUTHENTICCTL = CONFIG.get('AUTHENTICCTL') or '/usr/sbin/authenticctl.py'
AUTHENTIC_DATA_DIR = CONFIG.get('AUTHENTIC_DATA_DIR') or '/usr/share/authentic/'
LCSCTL = CONFIG.get('LCSCTL') or '/usr/sbin/lcsctl.py'
LCS_DATADIR = CONFIG.get('LCS_DATADIR') or '/usr/share/lcs/'
LASSO_BUILDDIR = os.environ.get('LASSO_BUILDDIR') or \
    CONFIG.get('LASSO_BUILDDIR') or \
    os.path.realpath(os.path.join(os.path.dirname(__FILE__), "..", "..", ".."))

os.environ['LD_LIBRARY_PATH'] = os.path.join(LASSO_BUILDDIR, "lasso", ".libs") + ":" + \
        os.environ.get('LD_LIBRARY_PATH', '')
os.environ['PYTHONPATH'] = os.path.join(LASSO_BUILDDIR, "bindings", "python") + \
        ":" + os.path.join(LASSO_BUILDDIR, "bindings", "python", ".libs") + ":" + \
        os.environ.get('PYTHONPATH', '')

try:
    from config import *
except ImportError:
    pass

pids = []

def waitforport(port, start):
    while True:
        if time.time() - start > 90:
            raise Exception('Servers did not start in 90 seconds!!')
        time.sleep(5)
        try:
            urllib2.urlopen('http://localhost:%s' % port)
        except urllib2.URLError:
            continue
        else:
            break

def setup():
    if not os.path.exists(AUTHENTIC_SRCDIR):
        print >> sys.stderr, 'Authentic source dir (%s) does not exist' % AUTHENTIC_SRCDIR
        print >> sys.stderr, 'Create it or edit tests/config.py to match your local installation'
        sys.exit(1)

    silent = os.environ.get('NO_SILENT') is None
    twill.commands.reset_browser()
    twill.set_output(file('/dev/null', 'w'))
    base = []
    if os.environ.get('VALGRIND') is '1' and os.path.exists('/usr/bin/valgrind'):
        base = ['./valgrind-wrapper.sh', 'python']

    os.mkdir('/tmp/.tests')
    authentic_command = base + [AUTHENTICCTL, 'start',
            '--app-dir', '/tmp/.tests/authentictests',
            '--data-dir', AUTHENTIC_DATADIR,
            '--extra', os.path.join(AUTHENTIC_SRCDIR, 'extra', 'conformance'),
            '--port', '10001', '--http']
    if silent:
        authentic_command.append('--silent')
    sp = subprocess.Popen(authentic_command)
    pids.append(sp.pid)
    lcs_command = base + [LCSCTL, 'start',
            '--app-dir', '/tmp/.tests/lcstests',
            '--data-dir', LCS_DATADIR,
            '--port', '10002', '--http']
    if silent:
        lcs_command.append('--silent')
    sp = subprocess.Popen(lcs_command)
    pids.append(sp.pid)

    # Wait for the daemons to load themselves
    starttime = time.time()
    waitforport(10001, starttime)
    waitforport(10002, starttime)


def teardown():
    for pid in pids:
        try:
            # valgrind seems to prefer SIGINT to SIGTERM
            os.kill(pid, signal.SIGINT)
        except OSError:
            print >> sys.stderr, 'failed to kill pid %s' % pid
    os.system('rm -rf /tmp/.tests/')