diff -r 000000000000 -r ae805ac0140d python-2.5.2/win32/Lib/test/test_httplib.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/python-2.5.2/win32/Lib/test/test_httplib.py Fri Apr 03 17:19:34 2009 +0100 @@ -0,0 +1,172 @@ +import httplib +import StringIO +import sys + +from unittest import TestCase + +from test import test_support + +class FakeSocket: + def __init__(self, text, fileclass=StringIO.StringIO): + self.text = text + self.fileclass = fileclass + + def sendall(self, data): + self.data = data + + def makefile(self, mode, bufsize=None): + if mode != 'r' and mode != 'rb': + raise httplib.UnimplementedFileMode() + return self.fileclass(self.text) + +class NoEOFStringIO(StringIO.StringIO): + """Like StringIO, but raises AssertionError on EOF. + + This is used below to test that httplib doesn't try to read + more from the underlying file than it should. + """ + def read(self, n=-1): + data = StringIO.StringIO.read(self, n) + if data == '': + raise AssertionError('caller tried to read past EOF') + return data + + def readline(self, length=None): + data = StringIO.StringIO.readline(self, length) + if data == '': + raise AssertionError('caller tried to read past EOF') + return data + + +class HeaderTests(TestCase): + def test_auto_headers(self): + # Some headers are added automatically, but should not be added by + # .request() if they are explicitly set. + + import httplib + + class HeaderCountingBuffer(list): + def __init__(self): + self.count = {} + def append(self, item): + kv = item.split(':') + if len(kv) > 1: + # item is a 'Key: Value' header string + lcKey = kv[0].lower() + self.count.setdefault(lcKey, 0) + self.count[lcKey] += 1 + list.append(self, item) + + for explicit_header in True, False: + for header in 'Content-length', 'Host', 'Accept-encoding': + conn = httplib.HTTPConnection('example.com') + conn.sock = FakeSocket('blahblahblah') + conn._buffer = HeaderCountingBuffer() + + body = 'spamspamspam' + headers = {} + if explicit_header: + headers[header] = str(len(body)) + conn.request('POST', '/', body, headers) + self.assertEqual(conn._buffer.count[header.lower()], 1) + +# Collect output to a buffer so that we don't have to cope with line-ending +# issues across platforms. Specifically, the headers will have \r\n pairs +# and some platforms will strip them from the output file. + +def test(): + buf = StringIO.StringIO() + _stdout = sys.stdout + try: + sys.stdout = buf + _test() + finally: + sys.stdout = _stdout + + # print individual lines with endings stripped + s = buf.getvalue() + for line in s.split("\n"): + print line.strip() + +def _test(): + # Test HTTP status lines + + body = "HTTP/1.1 200 Ok\r\n\r\nText" + sock = FakeSocket(body) + resp = httplib.HTTPResponse(sock, 1) + resp.begin() + print resp.read() + resp.close() + + body = "HTTP/1.1 400.100 Not Ok\r\n\r\nText" + sock = FakeSocket(body) + resp = httplib.HTTPResponse(sock, 1) + try: + resp.begin() + except httplib.BadStatusLine: + print "BadStatusLine raised as expected" + else: + print "Expect BadStatusLine" + + # Check invalid host_port + + for hp in ("www.python.org:abc", "www.python.org:"): + try: + h = httplib.HTTP(hp) + except httplib.InvalidURL: + print "InvalidURL raised as expected" + else: + print "Expect InvalidURL" + + for hp,h,p in (("[fe80::207:e9ff:fe9b]:8000", "fe80::207:e9ff:fe9b", 8000), + ("www.python.org:80", "www.python.org", 80), + ("www.python.org", "www.python.org", 80), + ("[fe80::207:e9ff:fe9b]", "fe80::207:e9ff:fe9b", 80)): + try: + http = httplib.HTTP(hp) + except httplib.InvalidURL: + print "InvalidURL raised erroneously" + c = http._conn + if h != c.host: raise AssertionError, ("Host incorrectly parsed", h, c.host) + if p != c.port: raise AssertionError, ("Port incorrectly parsed", p, c.host) + + # test response with multiple message headers with the same field name. + text = ('HTTP/1.1 200 OK\r\n' + 'Set-Cookie: Customer="WILE_E_COYOTE"; Version="1"; Path="/acme"\r\n' + 'Set-Cookie: Part_Number="Rocket_Launcher_0001"; Version="1";' + ' Path="/acme"\r\n' + '\r\n' + 'No body\r\n') + hdr = ('Customer="WILE_E_COYOTE"; Version="1"; Path="/acme"' + ', ' + 'Part_Number="Rocket_Launcher_0001"; Version="1"; Path="/acme"') + s = FakeSocket(text) + r = httplib.HTTPResponse(s, 1) + r.begin() + cookies = r.getheader("Set-Cookie") + if cookies != hdr: + raise AssertionError, "multiple headers not combined properly" + + # Test that the library doesn't attempt to read any data + # from a HEAD request. (Tickles SF bug #622042.) + sock = FakeSocket( + 'HTTP/1.1 200 OK\r\n' + 'Content-Length: 14432\r\n' + '\r\n', + NoEOFStringIO) + resp = httplib.HTTPResponse(sock, 1, method="HEAD") + resp.begin() + if resp.read() != "": + raise AssertionError, "Did not expect response from HEAD request" + resp.close() + + +class OfflineTest(TestCase): + def test_responses(self): + self.assertEquals(httplib.responses[httplib.NOT_FOUND], "Not Found") + +def test_main(verbose=None): + tests = [HeaderTests,OfflineTest] + test_support.run_unittest(*tests) + +test()