run-tests.py

#!/usr/bin/env python

from __future__ import absolute_import, print_function, unicode_literals

import subprocess
import sys
import time

import requests


def start_solr():
    solr_proc = subprocess.Popen("./start-test-solr.sh",
                                 stdout=open("test-solr.stdout.log", "wb"),
                                 stderr=open("test-solr.stderr.log", "wb"))

    solr_retries = 0

    while True:
        try:
            r = requests.get("http://localhost:8983/solr/core0/select/?q=startup")
            status_code = r.status_code
        except requests.RequestException:
            status_code = 0

        if status_code == 200:
            break
        elif solr_retries < 60:
            solr_retries += 1
            print("Waiting 10 seconds for Solr to start (retry #%d)" % solr_retries, file=sys.stderr)
            time.sleep(10)
        else:
            print("Solr took too long to start (#%d retries)" % solr_retries, file=sys.stderr)
            sys.exit(1)

    return solr_proc


def main():
    solr_proc = start_solr()

    if sys.version_info >= (3, 3):
        cmd = ['python', '-m', 'unittest', 'tests']
    elif sys.version_info >= (2, 7):
        cmd = ['python', '-m', 'unittest2', 'tests']
    else:
        cmd = ['unit2', 'discover', '-s', 'tests', '-p', '[a-z]*.py']

    try:
        subprocess.check_call(cmd)
    finally:
        solr_proc.terminate()
        solr_proc.wait()

if __name__ == "__main__":
    main()