IPython/testing/util.py

# encoding: utf-8
"""This file contains utility classes for performing tests with Deferreds.
"""
#-----------------------------------------------------------------------------
#  Copyright (C) 2009  The IPython Development Team
#
#  Distributed under the terms of the BSD License.  The full license is in
#  the file COPYING, distributed as part of this software.
#-----------------------------------------------------------------------------

#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------

from twisted.trial import unittest
from twisted.internet import defer

#-----------------------------------------------------------------------------
# Classes and functions
#-----------------------------------------------------------------------------

class DeferredTestCase(unittest.TestCase):
    
    def assertDeferredEquals(self, deferred, expectedResult,
                                      chainDeferred=None):
        """Calls assertEquals on the result of the deferred and expectedResult.
        
        chainDeferred can be used to pass in previous Deferred objects that
        have tests being run on them.  This chaining of Deferred's in tests
        is needed to insure that all Deferred's are cleaned up at the end of
        a test.
        """
        
        if chainDeferred is None:
            chainDeferred = defer.succeed(None)
               
        def gotResult(actualResult):
            self.assertEquals(actualResult, expectedResult)
        
        deferred.addCallback(gotResult)
                    
        return chainDeferred.addCallback(lambda _: deferred)
    
    def assertDeferredRaises(self, deferred, expectedException,
                                      chainDeferred=None):
        """Calls assertRaises on the Failure of the deferred and expectedException.
        
        chainDeferred can be used to pass in previous Deferred objects that
        have tests being run on them.  This chaining of Deferred's in tests
        is needed to insure that all Deferred's are cleaned up at the end of
        a test.
        """
        
        if chainDeferred is None:
            chainDeferred = defer.succeed(None)

        def gotFailure(f):
            #f.printTraceback()
            self.assertRaises(expectedException, f.raiseException)
            #return f
        
        deferred.addBoth(gotFailure)
            
        return chainDeferred.addCallback(lambda _: deferred)