Fixed unit test not restoring things to prepatched state

This commit is contained in:
gholt 2011-05-12 22:54:54 +00:00
parent 4ef6644e26
commit d7a9a51201

View File

@ -85,13 +85,17 @@ class TestBufferedHTTP(unittest.TestCase):
def endheaders(self): def endheaders(self):
pass pass
origHTTPSConnection = bufferedhttp.HTTPSConnection
bufferedhttp.HTTPSConnection = MockHTTPSConnection bufferedhttp.HTTPSConnection = MockHTTPSConnection
bufferedhttp.http_connect('127.0.0.1', 8080, 'sda', 1, 'GET', '/', try:
headers={'x-one': '1', 'x-two': 2, 'x-three': 3.0, bufferedhttp.http_connect('127.0.0.1', 8080, 'sda', 1, 'GET', '/',
'x-four': {'crazy': 'value'}}, ssl=True) headers={'x-one': '1', 'x-two': 2, 'x-three': 3.0,
bufferedhttp.http_connect_raw('127.0.0.1', 8080, 'GET', '/', 'x-four': {'crazy': 'value'}}, ssl=True)
headers={'x-one': '1', 'x-two': 2, 'x-three': 3.0, bufferedhttp.http_connect_raw('127.0.0.1', 8080, 'GET', '/',
'x-four': {'crazy': 'value'}}, ssl=True) headers={'x-one': '1', 'x-two': 2, 'x-three': 3.0,
'x-four': {'crazy': 'value'}}, ssl=True)
finally:
bufferedhttp.HTTPSConnection = origHTTPSConnection
if __name__ == '__main__': if __name__ == '__main__':