root / trunk / testall.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
##
## runtests.py -- Runs tests to verify the gdhost code works
## Started on  Wed Sep 14 23:49:35 2005 Christian Höltje
## 
## Copyright (C) 2005 Christian Höltje
## 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
##

import os, sys
import unittest
sys.path.insert(0,os.path.realpath(os.curdir))

def importdir( dir, limit=3, strip=None, verbose=0 ):
    global counter
    if verbose: print "importdir( %s, %d, %s, %s )" % (dir, limit,
                                                       strip, verbose)
    if limit <= 0:
        return []
    dir = os.path.abspath( dir )
    if strip is None:
        strip = len(os.path.abspath( os.curdir ) + os.sep)
    filelist = os.listdir( dir )
    filelist.sort()
    output = []
    for file in filelist:
        path = os.path.join( dir, file )[strip:]
        if os.path.isdir( path ):
            if not (
                file.startswith('.') 
            ):
               output.extend( importdir( path, limit - 1, strip, verbose ) )
        elif os.path.isfile( path ) and \
             file.endswith('.py') and \
             not (
            file.startswith( '_' ) or \
            file.startswith( 'testall.py' ) or \
            file.find( '#' ) >= 0
            ):
            imp = path[:-3].replace(os.sep,'.')
            if verbose: print " Importing '%s'..."% imp
            output.append( "from %s import *" % imp )
        elif verbose: print " skipping '%s'" % path
    return output

if __name__ == "__main__":
    for e in importdir(
        os.path.join( os.curdir, 'gdhost' ),
        verbose=('-v' in sys.argv) ):
        exec e
    unittest.main()