# Copyright 2010, Google Inc.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
# * Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above
# copyright notice, this list of conditions and the following disclaimer
# in the documentation and/or other materials provided with the
# distribution.
# * Neither the name of Google Inc. nor the names of its
# contributors may be used to endorse or promote products derived from
# this software without specific prior written permission.
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
import socket
import unittest
import httpplus
# relative import to ease embedding the library
import util
class SimpleHttpTest(util.HttpTestBase, unittest.TestCase):
def _run_simple_test(self, host, server_data, expected_req, expected_data):
con = httpplus.HTTPConnection(host)
con._connect()
con.sock.data = server_data
con.request('GET', '/')
self.assertStringEqual(expected_req, con.sock.sent)
self.assertEqual(expected_data, con.getresponse().read())
def test_broken_data_obj(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
self.assertRaises(httpplus.BadRequestData,
con.request, 'POST', '/', body=1)
def test_no_keepalive_http_1_0(self):
expected_request_one = """GET /remote/.hg/requires HTTP/1.1
Host: localhost:9999
range: bytes=0-
accept-encoding: identity
accept: application/mercurial-0.1
user-agent: mercurial/proto-1.0
""".replace('\n', '\r\n')
expected_response_headers = """HTTP/1.0 200 OK
Server: SimpleHTTP/0.6 Python/2.6.1
Date: Sun, 01 May 2011 13:56:57 GMT
Content-type: application/octet-stream
Content-Length: 33
Last-Modified: Sun, 01 May 2011 13:56:56 GMT
""".replace('\n', '\r\n')
expected_response_body = """revlogv1
store
fncache
dotencode
"""
con = httpplus.HTTPConnection('localhost:9999')
con._connect()
con.sock.data = [expected_response_headers, expected_response_body]
con.request('GET', '/remote/.hg/requires',
headers={'accept-encoding': 'identity',
'range': 'bytes=0-',
'accept': 'application/mercurial-0.1',
'user-agent': 'mercurial/proto-1.0',
})
self.assertStringEqual(expected_request_one, con.sock.sent)
self.assertEqual(con.sock.closed, False)
self.assertNotEqual(con.sock.data, [])
self.assert_(con.busy())
resp = con.getresponse()
self.assertStringEqual(resp.read(), expected_response_body)
self.failIf(con.busy())
self.assertEqual(con.sock, None)
self.assertEqual(resp.sock.data, [])
self.assert_(resp.sock.closed)
def test_multiline_header(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
con.sock.data = ['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Multiline: Value\r\n',
' Rest of value\r\n',
'Content-Length: 10\r\n',
'\r\n'
'1234567890'
]
con.request('GET', '/')
expected_req = ('GET / HTTP/1.1\r\n'
'Host: 1.2.3.4\r\n'
'accept-encoding: identity\r\n\r\n')
self.assertEqual(('1.2.3.4', 80), con.sock.sa)
self.assertEqual(expected_req, con.sock.sent)
resp = con.getresponse()
self.assertEqual('1234567890', resp.read())
self.assertEqual(['Value\n Rest of value'],
resp.headers.getheaders('multiline'))
# Socket should not be closed
self.assertEqual(resp.sock.closed, False)
self.assertEqual(con.sock.closed, False)
def testSimpleRequest(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
con.sock.data = ['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'MultiHeader: Value\r\n'
'MultiHeader: Other Value\r\n'
'MultiHeader: One More!\r\n'
'Content-Length: 10\r\n',
'\r\n'
'1234567890'
]
con.request('GET', '/')
expected_req = ('GET / HTTP/1.1\r\n'
'Host: 1.2.3.4\r\n'
'accept-encoding: identity\r\n\r\n')
self.assertEqual(('1.2.3.4', 80), con.sock.sa)
self.assertEqual(expected_req, con.sock.sent)
resp = con.getresponse()
self.assertEqual('1234567890', resp.read())
self.assertEqual(['Value', 'Other Value', 'One More!'],
resp.headers.getheaders('multiheader'))
self.assertEqual(['BogusServer 1.0'],
resp.headers.getheaders('server'))
def testHeaderlessResponse(self):
con = httpplus.HTTPConnection('1.2.3.4', use_ssl=False)
con._connect()
con.sock.data = ['HTTP/1.1 200 OK\r\n',
'\r\n'
'1234567890'
]
con.sock.close_on_empty = True
con.request('GET', '/')
expected_req = ('GET / HTTP/1.1\r\n'
'Host: 1.2.3.4\r\n'
'accept-encoding: identity\r\n\r\n')
self.assertEqual(('1.2.3.4', 80), con.sock.sa)
self.assertEqual(expected_req, con.sock.sent)
resp = con.getresponse()
self.assertEqual('1234567890', resp.read())
self.assertEqual({}, dict(resp.headers))
self.assertEqual(resp.status, 200)
def testReadline(self):
con = httpplus.HTTPConnection('1.2.3.4')
con._connect()
con.sock.data = ['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Connection: Close\r\n',
'\r\n'
'1\n2\nabcdefg\n4\n5']
con.sock.close_on_empty = True
expected_req = ('GET / HTTP/1.1\r\n'
'Host: 1.2.3.4\r\n'
'accept-encoding: identity\r\n\r\n')
con.request('GET', '/')
self.assertEqual(('1.2.3.4', 80), con.sock.sa)
self.assertEqual(expected_req, con.sock.sent)
r = con.getresponse()
for expected in ['1\n', '2\n', 'abcdefg\n', '4\n', '5']:
actual = r.readline()
self.assertEqual(expected, actual,
'Expected %r, got %r' % (expected, actual))
def testReadlineTrickle(self):
con = httpplus.HTTPConnection('1.2.3.4')
con._connect()
# make sure it trickles in one byte at a time
# so that we touch all the cases in readline
con.sock.data = list(''.join(
['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Connection: Close\r\n',
'\r\n'
'1\n2\nabcdefg\n4\n5']))
con.sock.close_on_empty = True
expected_req = ('GET / HTTP/1.1\r\n'
'Host: 1.2.3.4\r\n'
'accept-encoding: identity\r\n\r\n')
con.request('GET', '/')
self.assertEqual(('1.2.3.4', 80), con.sock.sa)
self.assertEqual(expected_req, con.sock.sent)
r = con.getresponse()
for expected in ['1\n', '2\n', 'abcdefg\n', '4\n', '5']:
actual = r.readline()
self.assertEqual(expected, actual,
'Expected %r, got %r' % (expected, actual))
def testVariousReads(self):
con = httpplus.HTTPConnection('1.2.3.4')
con._connect()
# make sure it trickles in one byte at a time
# so that we touch all the cases in readline
con.sock.data = list(''.join(
['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Connection: Close\r\n',
'\r\n'
'1\n2',
'\na', 'bc',
'defg\n4\n5']))
con.sock.close_on_empty = True
expected_req = ('GET / HTTP/1.1\r\n'
'Host: 1.2.3.4\r\n'
'accept-encoding: identity\r\n\r\n')
con.request('GET', '/')
self.assertEqual(('1.2.3.4', 80), con.sock.sa)
self.assertEqual(expected_req, con.sock.sent)
r = con.getresponse()
for read_amt, expect in [(1, '1'), (1, '\n'),
(4, '2\nab'),
('line', 'cdefg\n'),
(None, '4\n5')]:
if read_amt == 'line':
self.assertEqual(expect, r.readline())
else:
self.assertEqual(expect, r.read(read_amt))
def testZeroLengthBody(self):
con = httpplus.HTTPConnection('1.2.3.4')
con._connect()
# make sure it trickles in one byte at a time
# so that we touch all the cases in readline
con.sock.data = list(''.join(
['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Content-length: 0\r\n',
'\r\n']))
expected_req = ('GET / HTTP/1.1\r\n'
'Host: 1.2.3.4\r\n'
'accept-encoding: identity\r\n\r\n')
con.request('GET', '/')
self.assertEqual(('1.2.3.4', 80), con.sock.sa)
self.assertEqual(expected_req, con.sock.sent)
r = con.getresponse()
self.assertEqual('', r.read())
def testIPv6(self):
self._run_simple_test('[::1]:8221',
['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Content-Length: 10',
'\r\n\r\n'
'1234567890'],
('GET / HTTP/1.1\r\n'
'Host: [::1]:8221\r\n'
'accept-encoding: identity\r\n\r\n'),
'1234567890')
self._run_simple_test('::2',
['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Content-Length: 10',
'\r\n\r\n'
'1234567890'],
('GET / HTTP/1.1\r\n'
'Host: ::2\r\n'
'accept-encoding: identity\r\n\r\n'),
'1234567890')
self._run_simple_test('[::3]:443',
['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Content-Length: 10',
'\r\n\r\n'
'1234567890'],
('GET / HTTP/1.1\r\n'
'Host: ::3\r\n'
'accept-encoding: identity\r\n\r\n'),
'1234567890')
def testEarlyContinueResponse(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
sock = con.sock
sock.data = ['HTTP/1.1 403 Forbidden\r\n',
'Server: BogusServer 1.0\r\n',
'Content-Length: 18',
'\r\n\r\n'
"You can't do that."]
expected_req = self.doPost(con, expect_body=False)
self.assertEqual(('1.2.3.4', 80), sock.sa)
self.assertStringEqual(expected_req, sock.sent)
self.assertEqual("You can't do that.", con.getresponse().read())
self.assertEqual(sock.closed, True)
def testEarlyContinueResponseNoContentLength(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
sock = con.sock
sock.data = ['HTTP/1.1 403 Forbidden\r\n',
'Server: BogusServer 1.0\r\n',
'\r\n'
"You can't do that."]
sock.close_on_empty = True
expected_req = self.doPost(con, expect_body=False)
self.assertEqual(('1.2.3.4', 80), sock.sa)
self.assertStringEqual(expected_req, sock.sent)
self.assertEqual("You can't do that.", con.getresponse().read())
self.assertEqual(sock.closed, True)
def testDeniedAfterContinueTimeoutExpires(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
sock = con.sock
sock.data = ['HTTP/1.1 403 Forbidden\r\n',
'Server: BogusServer 1.0\r\n',
'Content-Length: 18\r\n',
'Connection: close',
'\r\n\r\n'
"You can't do that."]
sock.read_wait_sentinel = 'Dear server, send response!'
sock.close_on_empty = True
# send enough data out that we'll chunk it into multiple
# blocks and the socket will close before we can send the
# whole request.
post_body = ('This is some POST data\n' * 1024 * 32 +
'Dear server, send response!\n' +
'This is some POST data\n' * 1024 * 32)
expected_req = self.doPost(con, expect_body=False,
body_to_send=post_body)
self.assertEqual(('1.2.3.4', 80), sock.sa)
self.assert_('POST data\n' in sock.sent)
self.assert_('Dear server, send response!\n' in sock.sent)
# We expect not all of our data was sent.
self.assertNotEqual(sock.sent, expected_req)
self.assertEqual("You can't do that.", con.getresponse().read())
self.assertEqual(sock.closed, True)
def testPostData(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
sock = con.sock
sock.read_wait_sentinel = 'POST data'
sock.early_data = ['HTTP/1.1 100 Co', 'ntinue\r\n\r\n']
sock.data = ['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Content-Length: 16',
'\r\n\r\n',
"You can do that."]
expected_req = self.doPost(con, expect_body=True)
self.assertEqual(('1.2.3.4', 80), sock.sa)
self.assertEqual(expected_req, sock.sent)
self.assertEqual("You can do that.", con.getresponse().read())
self.assertEqual(sock.closed, False)
def testServerWithoutContinue(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
sock = con.sock
sock.read_wait_sentinel = 'POST data'
sock.data = ['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Content-Length: 16',
'\r\n\r\n',
"You can do that."]
expected_req = self.doPost(con, expect_body=True)
self.assertEqual(('1.2.3.4', 80), sock.sa)
self.assertEqual(expected_req, sock.sent)
self.assertEqual("You can do that.", con.getresponse().read())
self.assertEqual(sock.closed, False)
def testServerWithSlowContinue(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
sock = con.sock
sock.read_wait_sentinel = 'POST data'
sock.data = ['HTTP/1.1 100 ', 'Continue\r\n\r\n',
'HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Content-Length: 16',
'\r\n\r\n',
"You can do that."]
expected_req = self.doPost(con, expect_body=True)
self.assertEqual(('1.2.3.4', 80), sock.sa)
self.assertEqual(expected_req, sock.sent)
resp = con.getresponse()
self.assertEqual("You can do that.", resp.read())
self.assertEqual(200, resp.status)
self.assertEqual(sock.closed, False)
def testSlowConnection(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
# simulate one byte arriving at a time, to check for various
# corner cases
con.sock.data = list('HTTP/1.1 200 OK\r\n'
'Server: BogusServer 1.0\r\n'
'Content-Length: 10'
'\r\n\r\n'
'1234567890')
con.request('GET', '/')
expected_req = ('GET / HTTP/1.1\r\n'
'Host: 1.2.3.4\r\n'
'accept-encoding: identity\r\n\r\n')
self.assertEqual(('1.2.3.4', 80), con.sock.sa)
self.assertEqual(expected_req, con.sock.sent)
self.assertEqual('1234567890', con.getresponse().read())
def testCloseAfterNotAllOfHeaders(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
con.sock.data = ['HTTP/1.1 200 OK\r\n',
'Server: NO CARRIER']
con.sock.close_on_empty = True
con.request('GET', '/')
self.assertRaises(httpplus.HTTPRemoteClosedError,
con.getresponse)
expected_req = ('GET / HTTP/1.1\r\n'
'Host: 1.2.3.4\r\n'
'accept-encoding: identity\r\n\r\n')
def testTimeout(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
con.sock.data = []
con.request('GET', '/')
self.assertRaises(httpplus.HTTPTimeoutException,
con.getresponse)
expected_req = ('GET / HTTP/1.1\r\n'
'Host: 1.2.3.4\r\n'
'accept-encoding: identity\r\n\r\n')
self.assertEqual(('1.2.3.4', 80), con.sock.sa)
self.assertEqual(expected_req, con.sock.sent)
def test_conn_keep_alive_but_server_close_anyway(self):
sockets = []
def closingsocket(*args, **kwargs):
s = util.MockSocket(*args, **kwargs)
sockets.append(s)
s.data = ['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Connection: Keep-Alive\r\n',
'Content-Length: 16',
'\r\n\r\n',
'You can do that.']
s.close_on_empty = True
return s
socket.socket = closingsocket
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
con.request('GET', '/')
r1 = con.getresponse()
r1.read()
self.assertFalse(con.sock.closed)
self.assert_(con.sock.remote_closed)
con.request('GET', '/')
self.assertEqual(2, len(sockets))
def test_server_closes_before_end_of_body(self):
con = httpplus.HTTPConnection('1.2.3.4:80')
con._connect()
s = con.sock
s.data = ['HTTP/1.1 200 OK\r\n',
'Server: BogusServer 1.0\r\n',
'Connection: Keep-Alive\r\n',
'Content-Length: 16',
'\r\n\r\n',
'You can '] # Note: this is shorter than content-length
s.close_on_empty = True
con.request('GET', '/')
r1 = con.getresponse()
self.assertRaises(httpplus.HTTPRemoteClosedError, r1.read)
def test_no_response_raises_response_not_ready(self):
con = httpplus.HTTPConnection('foo')
self.assertRaises(httpplus.httplib.ResponseNotReady, con.getresponse)
# no-check-code