Ensure that recorded calls are all replayed

This commit is contained in:
Vincent Jacques
2012-05-08 11:23:48 +01:00
parent 00904f3333
commit 6565563412
+6
View File
@@ -96,6 +96,11 @@ class TestCase( unittest.TestCase ):
self.g = github.Github( GithubCredentials.login, GithubCredentials.password )
def tearDown( self ):
unittest.TestCase.tearDown( self )
self.assertEqual( self.__file.readline(), "" )
self.__file.close()
def tearDownForRecord( self ):
unittest.TestCase.tearDown( self )
self.__file.close()
@@ -193,6 +198,7 @@ if len( sys.argv ) > 1 and sys.argv[ 1 ] == "--record":
exec "testCase = " + class_ + "( methodName = method )"
method = getattr( testCase, method )
TestCase.setUp = TestCase.setUpForRecord
TestCase.tearDown = TestCase.tearDownForRecord
testCase.setUp()
method()
testCase.tearDown()