Fix test for new version of urllib that uses HTTPConnection directly.

Changes the way the httplib classes are stubbed out.
This commit is contained in:
Jeremy Hylton 2007-08-04 19:23:09 +00:00
parent 39b198d835
commit 77553ab531

View file

@ -111,11 +111,11 @@ class urlopen_HttpTests(unittest.TestCase):
class FakeHTTPConnection(httplib.HTTPConnection):
def connect(self):
self.sock = FakeSocket(fakedata)
assert httplib.HTTP._connection_class == httplib.HTTPConnection
httplib.HTTP._connection_class = FakeHTTPConnection
self._connection_class = httplib.HTTPConnection
httplib.HTTPConnection = FakeHTTPConnection
def unfakehttp(self):
httplib.HTTP._connection_class = httplib.HTTPConnection
httplib.HTTPConnection = self._connection_class
def test_read(self):
self.fakehttp(b"Hello!")