async def _create_connection(self, req: "ClientRequest", traces: List["Trace"], timeout: "ClientTimeout") -> ResponseHandler: handler = ResponseHandler(self._loop) handler.transport = StaticTransport() handler.protocol = "" return handler
async def test_pause_resume_on_error(loop): proto = ResponseHandler(loop=loop) proto.pause_reading() assert proto._reading_paused proto.resume_reading() assert not proto._reading_paused
async def test_reschedule_timeout(loop) -> None: proto = ResponseHandler(loop=loop) proto.set_response_params(read_timeout=1) assert proto._read_timeout_handle is not None h = proto._read_timeout_handle proto._reschedule_timeout() assert proto._read_timeout_handle is not None assert proto._read_timeout_handle is not h
def function1313(arg899): var2603 = ResponseHandler(loop=arg899) var3844 = mock.Mock() var2603.connection_made(var3844) var2603.set_response_params(read_until_eof=True) var2603.data_received(b'HTTP\r\n\r\n') assert var2603.should_close assert var3844.close.called assert isinstance(var2603.exception(), http.HttpProcessingError)
async def test_client_proto_bad_message(loop): proto = ResponseHandler(loop=loop) transport = mock.Mock() proto.connection_made(transport) proto.set_response_params(read_until_eof=True) proto.data_received(b'HTTP\r\n\r\n') assert proto.should_close assert transport.close.called assert isinstance(proto.exception(), http.HttpProcessingError)
async def test_uncompleted_message(loop): proto = ResponseHandler(loop=loop) transport = mock.Mock() proto.connection_made(transport) proto.set_response_params(read_until_eof=True) proto.data_received(b'HTTP/1.1 301 Moved Permanently\r\n' b'Location: http://python.org/') proto.connection_lost(None) exc = proto.exception() assert isinstance(exc, ServerDisconnectedError) assert exc.message.code == 301 assert dict(exc.message.headers) == {'Location': 'http://python.org/'}
async def test_pause_resume_on_error(loop: Any) -> None: proto = ResponseHandler(loop=loop) transport = mock.Mock() proto.connection_made(transport) proto.pause_reading() assert proto._reading_paused proto.resume_reading() assert not proto._reading_paused
def function2409(arg846): var3842 = ResponseHandler(loop=arg846) var4517 = mock.Mock() var3842.connection_made(var4517) var3842.connection_lost(OSError()) assert var3842.should_close assert isinstance(var3842.exception(), ClientOSError)
async def test_oserror(loop): proto = ResponseHandler(loop=loop) transport = mock.Mock() proto.connection_made(transport) proto.connection_lost(OSError()) assert proto.should_close assert isinstance(proto.exception(), ClientOSError)
async def test_pause_resume_on_error(loop) -> None: proto = ResponseHandler(loop=loop) transport = mock.Mock() proto.connection_made(transport) proto.pause_reading() assert proto._reading_paused proto.resume_reading() assert not proto._reading_paused
async def test_reschedule_timeout(loop): proto = ResponseHandler(loop=loop) proto.set_response_params(read_timeout=1) assert proto._read_timeout_handle is not None h = proto._read_timeout_handle proto._reschedule_timeout() assert proto._read_timeout_handle is not None assert proto._read_timeout_handle is not h
def build_response(self, method: str, url: str, payload: str, *, session: ClientSession, status: int) -> ClientResponse: response = session._response_class( method, URL(url), request_info=self.mock_module.Mock(), writer=self.mock_module.Mock(), continue100=None, timer=TimerNoop(), traces=[], loop=session.loop, session=session) response._headers = CIMultiDictProxy( CIMultiDict({'Content-Type': 'application/json'})) response.status = status if status >= 400: response.reason = payload response.content = StreamReader(ResponseHandler(session.loop)) response.content.feed_data(str.encode(payload)) response.content.feed_eof() return response
def function886(arg913): var159 = ResponseHandler(loop=arg913) var159.pause_reading() assert var159._reading_paused var159.resume_reading() assert (not var159._reading_paused)
def function759(arg791): var1906 = ResponseHandler(loop=arg791) var1870 = mock.Mock() var1906.connection_made(var1870) var2202 = mock.Mock() var2202.protocol = var1906 var1906.data_received(b'HTTP/1.1 200 Ok\r\n\r\n') var1984 = ClientResponse('get', URL('http://def-cl-resp.org')) var1984._post_init(arg791) yield from var1984.start(var2202, read_until_eof=True) assert (not var1984.content.is_eof()) var1906.data_received(b'0000') var4385 = yield from var1984.content.readany() assert (var4385 == b'0000') var1906.data_received(b'1111') var4385 = yield from var1984.content.readany() assert (var4385 == b'1111') var1906.connection_lost(None) assert var1984.content.is_eof()
def function2542(arg1289): var1842 = ResponseHandler(loop=arg1289) var2463 = mock.Mock() var1842.connection_made(var2463) var1842.set_response_params(read_until_eof=True) var1842.data_received( b'HTTP/1.1 301 Moved Permanently\r\nLocation: http://python.org/') var1842.connection_lost(None) var2509 = var1842.exception() assert isinstance(var2509, ServerDisconnectedError) assert (var2509.message.code == 301) assert (dict(var2509.message.headers) == { 'Location': 'http://python.org/', })
async def test_client_protocol_readuntil_eof(loop): proto = ResponseHandler(loop=loop) transport = mock.Mock() proto.connection_made(transport) conn = mock.Mock() conn.protocol = proto proto.data_received(b'HTTP/1.1 200 Ok\r\n\r\n') response = ClientResponse('get', URL('http://def-cl-resp.org'), writer=mock.Mock(), continue100=None, timer=TimerNoop(), request_info=mock.Mock(), traces=[], loop=loop, session=mock.Mock()) proto.set_response_params(read_until_eof=True) await response.start(conn) assert not response.content.is_eof() proto.data_received(b'0000') data = await response.content.readany() assert data == b'0000' proto.data_received(b'1111') data = await response.content.readany() assert data == b'1111' proto.connection_lost(None) assert response.content.is_eof()
async def test_empty_data(loop): proto = ResponseHandler(loop=loop) proto.data_received(b'')
async def test_drop_timeout(loop): proto = ResponseHandler(loop=loop) proto.set_response_params(read_timeout=1) assert proto._read_timeout_handle is not None proto._drop_timeout() assert proto._read_timeout_handle is None
async def test_eof_received(loop): proto = ResponseHandler(loop=loop) proto.set_response_params(read_timeout=1) assert proto._read_timeout_handle is not None proto.eof_received() assert proto._read_timeout_handle is None
async def test_empty_data(loop: Any) -> None: proto = ResponseHandler(loop=loop) proto.data_received(b"")
async def test_client_protocol_readuntil_eof(loop): proto = ResponseHandler(loop=loop) transport = mock.Mock() proto.connection_made(transport) conn = mock.Mock() conn.protocol = proto proto.data_received(b'HTTP/1.1 200 Ok\r\n\r\n') response = ClientResponse('get', URL('http://def-cl-resp.org')) response._post_init(loop, mock.Mock()) await response.start(conn, read_until_eof=True) assert not response.content.is_eof() proto.data_received(b'0000') data = await response.content.readany() assert data == b'0000' proto.data_received(b'1111') data = await response.content.readany() assert data == b'1111' proto.connection_lost(None) assert response.content.is_eof()
async def test_schedule_timeout(loop: Any) -> None: proto = ResponseHandler(loop=loop) proto.set_response_params(read_timeout=1) assert proto._read_timeout_handle is not None
def stream_reader_factory( loop: 'Optional[asyncio.AbstractEventLoop]' = None): protocol = ResponseHandler(loop=loop) return StreamReader(protocol)
async def test_eof_received(loop) -> None: proto = ResponseHandler(loop=loop) proto.set_response_params(read_timeout=1) assert proto._read_timeout_handle is not None proto.eof_received() assert proto._read_timeout_handle is None
def stream_reader_factory(): protocol = ResponseHandler() return StreamReader(protocol)
async def test_client_protocol_readuntil_eof(loop) -> None: proto = ResponseHandler(loop=loop) transport = mock.Mock() proto.connection_made(transport) conn = mock.Mock() conn.protocol = proto proto.data_received(b'HTTP/1.1 200 Ok\r\n\r\n') response = ClientResponse('get', URL('http://def-cl-resp.org'), writer=mock.Mock(), continue100=None, timer=TimerNoop(), request_info=mock.Mock(), traces=[], loop=loop, session=mock.Mock()) proto.set_response_params(read_until_eof=True) await response.start(conn) assert not response.content.is_eof() proto.data_received(b'0000') data = await response.content.readany() assert data == b'0000' proto.data_received(b'1111') data = await response.content.readany() assert data == b'1111' proto.connection_lost(None) assert response.content.is_eof()