Extending pyunit (Python recipe) by Tim Watson

ActiveState Code (http://code.activestate.com/recipes/511475/)

#!/usr/bin/env python

"""
    pyunit2.py - extends the facilities in the unittest module (PyUnit ),
    which is supplied with the Python Standard Library (since version 2.1,
    see the unittest.py module or go to the original project home page at
    http://pyunit.sourceforge.net/pyunit.html for more info).

    Example:

    #!/usr/bin/env python

    from pyunit2 import * #includes class Fixture

    class ExampleTestFixture(Fixture):
        @test
        def anExampleTest(self):
            self.assertEqual(1, 2) #fails

        @test
        @expected_exception(NameError)
        def usingExpectedExceptions(self):
            class Foo: pass
            f = Foo()
            print f.non_existent_attribute

    if __name__ == "__main__":
        #the following line(s) equate to
        #import unittest
        #unittest.main()
        #... etc
        import pyunit2
        pyunit2.main()
"""
__author__ = "Tim Watson"
__version__ = "$Revision: 0.2 $"
__license__ = "Python"

##############################################################################
# Exported classes and functions
##############################################################################
__all__ = [
    'expected_exception',
    'test',
    'Fixture'
]

import unittest

##############################################################################
# utility classes
##############################################################################

class TestDeclaration( object ):
    declarations = {}
    def __init__( self, func, doc_string=None ):
        if func.func_name.startswith( "test" ): return
        func.__doc__ = doc_string or func.__doc__
        fname = "test_%s" % func.func_name
        if not fname in TestDeclaration.declarations:
            TestDeclaration.declarations[ fname ] = func

    def __call__( self, func ):
        if func.func_name.startswith( "test" ):
            def execute( *args, **kwargs ):
                return func( *args, **kwargs )
            return execute
        raise Exception( 'should not have arrived here!?' )

class ExpectedException( object ):
    def __init__( self, exception_class ):
        self.exception_class = exception_class

    def __call__( self, func ):
        def execute( *args, **kwargs ):
            self_pointer = args [ 0 ]
            assert not self_pointer is None
            self_pointer.assertRaises( self.exception_class,
                func, *args, **kwargs )
        return execute

##############################################################################
# replacement base class 'Fixture' and supporting meta-class(es)
##############################################################################

class TestFixtureManager( type ):
    """
    A meta-class for mapping the decorator based test syntax
    from this module, to standard PyUnit style test method names, etc.
    """
    def __new__( cls, name, bases, attrs ):
        new_class = type.__new__( cls, name, bases, attrs )
        if not bases: return new_class
        [ setattr( new_class, func_name, func )
            for func_name, func in TestDeclaration.declarations.iteritems() ]
        TestDeclaration.declarations.clear()
        return new_class

    def __init__( cls, name, bases, dict ):
        #todo: deal with fixture setup/teardown here!
        pass

class Fixture( unittest.TestCase ): __metaclass__ = TestFixtureManager

##############################################################################
# decorators to make declaring tests/expected exceptions easier!
##############################################################################

def test( func, doc_string=None ):
    """
        Marks a method as a test method. Removes the need
        to call all your test methods testXXX and so on.
    """
    return TestDeclaration( func, doc_string )

def expected_exception( ex_cls ):
    """
        Marks a method as expecting an exception of class -> ex_cls
    """
    return ExpectedException( ex_cls )

##############################################################################
# support for fixture wide setup/teardown (NOT IMPLEMENTED YET)
##############################################################################

def testFixtureSetUp():
    """
        Adds support for a setup method that runs
        only once per testcase/fixture. The method must be defined
        as a staticmethod.
    """
    pass

def testFixtureTearDown():
    """
        Adds support for a teardown method that runs
        only once per testcase/fixture. The method should be defined
        as a staticmethod.
    """
    pass

def main():
    unittest.main()

if __name__ == "__main__":
    main()