# vim: set ts=2 sw=4 tw=99 ft=python et:

 import os

 import re

 import sys

 import time

 import os.path

 from test import Test

 from subprocess import *

 from threading import Thread

 from display import Interactive

 def FindTests(tests, root, folder):

   folders = []

   for name in os.listdir(os.path.join(root, folder)):

     path = os.path.join(root, folder, name)

     if os.path.isdir(path):

       folders.append(name)

       continue

     out = re.match("^([^\.]+)\.out$", name)

     err = re.match("^([^\.]+)\.err$", name)

     if out:

         name = os.path.join(folder, out.group(1))

         tests.append(Test(root, name, 'out'))

     elif err:

         name = os.path.join(folder, err.group(1))

         tests.append(Test(root, name, 'err'))

   for name in folders:

     path = os.path.join(folder, name)

     FindTests(tests, root, path)

 def IsTestRoot(path, name):

   return os.path.isdir(path)

 def RunTest_(argv, options, L):

   if sys.platform != 'win32':

     options['close_fds'] = True

   p = Popen(argv, stdin=PIPE, stdout=PIPE, stderr=PIPE, **options)

   L[0] = p

   out, err = p.communicate()

   L[1] = (out, err, p.returncode)

 def RunTest(test, options, shell):

   argv = [shell, test.cmd()]

   L = [None, None]

   timeout = False

   th = Thread(target = RunTest_, args = (argv, options, L))

   th.start()

   th.join(10.0)

   while th.isAlive():

     if L[0] is not None:

       try:

         L[0].kill()

         time.sleep(.1)

         timeout = True

       except e:

         pass

   th.join()

   (out, err, code) = L[1]

   return test.check(out, err, code, timeout)

 def RunTests(display, tests, shell):

   display.begin(tests)

   ok = True

   try:

     for test in tests:

       options = { }

       result = RunTest(test, options, shell)

       ok = ok and result.ok

       display.result(test, result)

   except KeyboardInterrupt:

     display.cancel()

     ok = False

   return ok

 def main(argv):

   root = os.path.dirname(os.path.abspath(__file__))

   from optparse import OptionParser

   op = OptionParser(usage='%prog [options] <shell>')

   (options, args) = op.parse_args(argv)

   if len(args) != 1:

     op.error('missing <shell> argument')

   shell = os.path.normpath(args[0])

   tests = []

   for name in os.listdir(root):

     path = os.path.join(root, name)

     if IsTestRoot(path, name):

       FindTests(tests, root, name)

   if len(tests) == 0:

     op.error('no test cases found')

   display = Interactive()

   ok = RunTests(display, tests, shell)

   if ok:

     sys.exit(0)

   else:

     sys.exit(1)

 if __name__ == '__main__':

   main(sys.argv[1:])