root/trunk/geniusql/test/test.py

import getopt
import sys
import unittest

from geniusql.test import tools


class TestHarness(object):
    """A test harness for Geniusql."""
    
    def __init__(self, available_tests):
        """Constructor to populate the TestHarness instance.
        
        available_tests should be a list of module names (strings).
        """
        self.available_tests = available_tests
        self.tests = []
    
    def load(self, args=sys.argv[1:]):
        """Populate a TestHarness from sys.argv.
        
        args defaults to sys.argv[1:], but you can provide a different
            set of args if you like.
        """
        
        longopts = ['help']
        longopts.extend(self.available_tests)
        try:
            opts, args = getopt.getopt(args, "", longopts)
        except getopt.GetoptError:
            # print help information and exit
            self.help()
            sys.exit(2)
        
        self.tests = []
        
        for o, a in opts:
            if o == '--help':
                self.help()
                sys.exit()
            else:
                o = o[2:]
                if o in self.available_tests and o not in self.tests:
                    self.tests.append(o)
        
        if not self.tests:
            self.tests = self.available_tests[:]
    
    def help(self):
        """Print help for test.py command-line options."""
        
        print """
Geniusql Test Program
    Usage:
        test.py --<testname>
        
        tests:"""
        for name in self.available_tests:
            print '        --' + name
    
    def run(self, conf=None):
        """Run the test harness."""
        self.load()
        
        import geniusql
        print "Python version:", sys.version.split()[0]
        print "Geniusql version:", geniusql.__version__
        
        for testmod in self.tests:
            if testmod.startswith("test_store"):
                print
                print "Testing %s storage..." % testmod[10:]
                mod = __import__(testmod, globals(), locals(), [''])
                if hasattr(mod, 'proxied_opts'):
                    mod.proxied_opts['Prefix'] = 'test'
                elif hasattr(mod, 'opts'):
                    mod.opts['Prefix'] = 'test'
                mod.run()
            else:
                suite = unittest.TestLoader().loadTestsFromName(testmod)
                tools.TestRunner.run(suite)


def run():
    
    tools.prefer_parent_path()
    
    testList = [
        'test_firebird',
        'test_msaccess',
        'test_mysql',
        'test_psycopg',
        'test_pypgsql',
        'test_sqlite',
        'test_sqlserver',
    ]
    TestHarness(testList).run()


if __name__ == '__main__':
    run()