mirror of
https://github.com/status-im/PyGithub.git
synced 2026-09-02 03:41:08 +00:00
Ensure that recorded calls are all replayed
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user