#!/usr/bin/env python -u
import commands
import os
import random
import sys
import time


restart = '-s' in sys.argv
if restart:
    sys.argv.remove('-s')

try:
    nreps = int(sys.argv[1])
except IndexError:
    nreps = 4

if restart:
    os.system('cd /Library/PostgreSQL/9.1/ && sudo -u postgres pg_ctl restart -D data')

for i in range(nreps):
    time.sleep(2 * random.random())
    output = commands.getoutput('swaddle env ./nosetests test_participant.py')
    if 'FAILED' in output:
        assert 'two_dollars' in output, output
        sys.stdout.write('X')
    else:
        assert 'OK' in output, output
        sys.stdout.write('.')

print