home *** CD-ROM | disk | FTP | other *** search
/ H4CK3R 14 / hacker14.iso / programacao / pythonwin / python.exe / TEST_HTTPLIB.PY < prev    next >
Encoding:
Python Source  |  2003-07-08  |  3.5 KB  |  114 lines
  1. import httplib
  2. import StringIO
  3. import sys
  4.  
  5. from test.test_support import verify,verbose
  6.  
  7. class FakeSocket:
  8.     def __init__(self, text, fileclass=StringIO.StringIO):
  9.         self.text = text
  10.         self.fileclass = fileclass
  11.  
  12.     def makefile(self, mode, bufsize=None):
  13.         if mode != 'r' and mode != 'rb':
  14.             raise httplib.UnimplementedFileMode()
  15.         return self.fileclass(self.text)
  16.  
  17. class NoEOFStringIO(StringIO.StringIO):
  18.     """Like StringIO, but raises AssertionError on EOF.
  19.  
  20.     This is used below to test that httplib doesn't try to read
  21.     more from the underlying file than it should.
  22.     """
  23.     def read(self, n=-1):
  24.         data = StringIO.StringIO.read(self, n)
  25.         if data == '':
  26.             raise AssertionError('caller tried to read past EOF')
  27.         return data
  28.  
  29.     def readline(self, length=None):
  30.         data = StringIO.StringIO.readline(self, length)
  31.         if data == '':
  32.             raise AssertionError('caller tried to read past EOF')
  33.         return data
  34.  
  35. # Collect output to a buffer so that we don't have to cope with line-ending
  36. # issues across platforms.  Specifically, the headers will have \r\n pairs
  37. # and some platforms will strip them from the output file.
  38.  
  39. def test():
  40.     buf = StringIO.StringIO()
  41.     _stdout = sys.stdout
  42.     try:
  43.         sys.stdout = buf
  44.         _test()
  45.     finally:
  46.         sys.stdout = _stdout
  47.  
  48.     # print individual lines with endings stripped
  49.     s = buf.getvalue()
  50.     for line in s.split("\n"):
  51.         print line.strip()
  52.  
  53. def _test():
  54.     # Test HTTP status lines
  55.  
  56.     body = "HTTP/1.1 200 Ok\r\n\r\nText"
  57.     sock = FakeSocket(body)
  58.     resp = httplib.HTTPResponse(sock, 1)
  59.     resp.begin()
  60.     print resp.read()
  61.     resp.close()
  62.  
  63.     body = "HTTP/1.1 400.100 Not Ok\r\n\r\nText"
  64.     sock = FakeSocket(body)
  65.     resp = httplib.HTTPResponse(sock, 1)
  66.     try:
  67.         resp.begin()
  68.     except httplib.BadStatusLine:
  69.         print "BadStatusLine raised as expected"
  70.     else:
  71.         print "Expect BadStatusLine"
  72.  
  73.     # Check invalid host_port
  74.  
  75.     for hp in ("www.python.org:abc", "www.python.org:"):
  76.         try:
  77.             h = httplib.HTTP(hp)
  78.         except httplib.InvalidURL:
  79.             print "InvalidURL raised as expected"
  80.         else:
  81.             print "Expect InvalidURL"
  82.  
  83.     # test response with multiple message headers with the same field name.
  84.     text = ('HTTP/1.1 200 OK\r\n'
  85.             'Set-Cookie: Customer="WILE_E_COYOTE"; Version="1"; Path="/acme"\r\n'
  86.             'Set-Cookie: Part_Number="Rocket_Launcher_0001"; Version="1";'
  87.             ' Path="/acme"\r\n'
  88.             '\r\n'
  89.             'No body\r\n')
  90.     hdr = ('Customer="WILE_E_COYOTE"; Version="1"; Path="/acme"'
  91.            ', '
  92.            'Part_Number="Rocket_Launcher_0001"; Version="1"; Path="/acme"')
  93.     s = FakeSocket(text)
  94.     r = httplib.HTTPResponse(s, 1)
  95.     r.begin()
  96.     cookies = r.getheader("Set-Cookie")
  97.     if cookies != hdr:
  98.         raise AssertionError, "multiple headers not combined properly"
  99.  
  100.     # Test that the library doesn't attempt to read any data
  101.     # from a HEAD request.  (Tickles SF bug #622042.)
  102.     sock = FakeSocket(
  103.         'HTTP/1.1 200 OK\r\n'
  104.         'Content-Length: 14432\r\n'
  105.         '\r\n',
  106.         NoEOFStringIO)
  107.     resp = httplib.HTTPResponse(sock, 1, method="HEAD")
  108.     resp.begin()
  109.     if resp.read() != "":
  110.         raise AssertionError, "Did not expect response from HEAD request"
  111.     resp.close()
  112.  
  113. test()
  114.