diff --git a/NewIntegrationTest.py b/NewIntegrationTest.py index 51d507f2..7a68531f 100644 --- a/NewIntegrationTest.py +++ b/NewIntegrationTest.py @@ -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()