def test_data_stream_not_bytes(self): @asyncio.coroutine def gen(): yield object() return b' result' req = HttpRequest('POST', 'http://python.org/', data=gen(), loop=self.loop) req.send(self.transport) self.assertRaises(ValueError, self.loop.run_until_complete, req._writer)
def test_data_stream_not_bytes(self): @asyncio.coroutine def gen(): yield object() return b' result' req = HttpRequest('POST', 'http://python.org/', data=gen(), loop=self.loop) req.send(self.transport, self.protocol) self.loop.run_until_complete(req._writer) self.assertTrue(self.protocol.set_exception.called)
def test_content_encoding_header(self, m_http): req = HttpRequest('get', 'http://python.org/', headers={'Content-Encoding': 'deflate'}, loop=self.loop) req.send(self.transport, self.protocol) self.assertEqual(req.headers['Transfer-encoding'], 'chunked') self.assertEqual(req.headers['Content-encoding'], 'deflate') m_http.Request.return_value\ .add_compression_filter.assert_called_with('deflate') m_http.Request.return_value\ .add_chunking_filter.assert_called_with(8196)
def test_close(self): @asyncio.coroutine def gen(): yield from asyncio.sleep(0.00001, loop=self.loop) return b'result' req = HttpRequest('POST', 'http://python.org/', data=gen(), loop=self.loop) req.send(self.transport) self.loop.run_until_complete(req.close()) self.assertEqual(self.transport.write.mock_calls[-3:], [ unittest.mock.call(b'result'), unittest.mock.call(b'\r\n'), unittest.mock.call(b'0\r\n\r\n') ])
def test_data_continue(self): req = HttpRequest('POST', 'http://python.org/', data=b'data', expect100=True, loop=self.loop) def coro(): yield from asyncio.sleep(0.0001, loop=self.loop) req._continue.set_result(1) asyncio. async (coro(), loop=self.loop) req.send(self.transport) self.assertEqual(1, len(self.transport.write.mock_calls)) self.loop.run_until_complete(req._writer) self.assertEqual(self.transport.write.mock_calls[-1], unittest.mock.call(b'data'))
def test_data_stream(self): def gen(): yield b'binary data' return b' result' req = HttpRequest('POST', 'http://python.org/', data=gen(), loop=self.loop) self.assertTrue(req.chunked) self.assertTrue(inspect.isgenerator(req.body)) self.assertEqual(req.headers['transfer-encoding'], 'chunked') resp = req.send(self.transport) self.assertIsInstance(req._writer, asyncio.Future) self.loop.run_until_complete(resp.wait_for_close()) self.assertIsNone(req._writer) self.assertEqual(self.transport.write.mock_calls[-3:], [ unittest.mock.call(b'binary data result'), unittest.mock.call(b'\r\n'), unittest.mock.call(b'0\r\n\r\n') ])
def test_version(self): req = HttpRequest('get', 'http://python.org/', version='1.0') self.assertEqual(req.version, (1, 0))
def test_chunked_explicit_size(self, m_http): req = HttpRequest('get', 'http://python.org/', chunked=1024) req.send(self.transport) self.assertEqual('chunked', req.headers['Transfer-encoding']) m_http.Request.return_value\ .add_chunking_filter.assert_called_with(1024)
def test_get_with_data(self): for meth in HttpRequest.GET_METHODS: req = HttpRequest(meth, 'http://python.org/', data={'life': '42'}) self.assertEqual('/?life=42', req.path)
def test_query_multivalued_param(self): for meth in HttpRequest.ALL_METHODS: req = HttpRequest(meth, 'http://python.org', params=(('test', 'foo'), ('test', 'baz'))) self.assertEqual(req.path, '/?test=foo&test=baz')
def test_path_is_not_double_encoded(self): req = HttpRequest('get', "http://0.0.0.0/get/test case") self.assertEqual(req.path, "/get/test%20case") req = HttpRequest('get', "http://0.0.0.0/get/test%20case") self.assertEqual(req.path, "/get/test%20case")
def test_basic_auth(self): req = HttpRequest('get', 'http://python.org', auth=('nkim', '1234')) self.assertIn('Authorization', req.headers) self.assertEqual('Basic bmtpbToxMjM0', req.headers['Authorization'])
def test_no_path(self): req = HttpRequest('get', 'http://python.org') self.assertEqual('/', req.path)
def test_headers_default(self): req = HttpRequest('get', 'http://python.org/', headers={'Accept-Encoding': 'deflate'}) self.assertEqual(req.headers['Accept-Encoding'], 'deflate')
def test_headers_list(self): req = HttpRequest('get', 'http://python.org/', headers=[('Content-Type', 'text/plain')]) self.assertIn('Content-Type', req.headers) self.assertEqual(req.headers['Content-Type'], 'text/plain')