#!/usr/bin/env python
# Copyright (C) 2005, Giovanni Bajo
# Based on previous work under copyright (c) 2001, 2002 McMillan Enterprises, Inc.
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA

# This program will execute any file with name test*<digit>.py. If your test
# need an aditional dependency name it test*<digit><letter>.py to be ignored
# by this program but be recognizable by any one as a dependency of that
# particual test.

import os, sys, glob, string
import shutil
try:
    here=os.path.dirname(__file__)
except NameError:
    here=os.path.dirname(sys.argv[0])
PYTHON = sys.executable
if sys.platform[:3] == 'win':
    if string.find(PYTHON, ' ') > -1:
        PYTHON='"%s"' % PYTHON

def clean():
    distdirs = glob.glob(os.path.join(here, 'disttest*'))
    for dir in distdirs:
        try:
            shutil.rmtree(dir)
        except OSError, e:
            print e
    builddirs = glob.glob(os.path.join(here, 'buildtest*'))
    for dir in builddirs:
        try:
            shutil.rmtree(dir)
        except OSError, e:
            print e
    wfiles = glob.glob(os.path.join(here, 'warn*.txt'))
    for file in wfiles:
        try:
            os.remove(file)
        except OSError, e:
            print e

def runtests():
    global here
    sources = glob.glob(os.path.join(here, 'test*[0-9].py'))
    path = os.environ["PATH"]
    for src in sources:
        print
        print "################## EXECUTING TEST %s ################################" % src
        print
        test = os.path.splitext(os.path.basename(src))[0]
        os.system('%s ../Build.py %s' % (PYTHON, test+".spec"))
        # Run the test in a clean environment to make sure they're really self-contained
        del os.environ["PATH"]
        os.system('dist%s%s%s' % (test, os.sep, test))
        os.environ["PATH"] = path
        print "################## FINISHING TEST %s  ################################" % src

if __name__ == '__main__':
    if len(sys.argv) == 1:
        clean()
        runtests()
    if '--clean' in sys.argv:
        clean()
    if '--run' in sys.argv:
        runtests()
    raw_input("Press any key to exit")