def function2107(function271, arg2001): var1118 = function271('GET', '/') var3191 = WebSocketResponse() yield from var3191.prepare(var1118) var3191._reader = mock.Mock() var4512 = helpers.create_future(arg2001) var4512.set_exception(asyncio.TimeoutError()) var3191._reader.read = make_mocked_coro(var4512) with pytest.raises(asyncio.TimeoutError): yield from var3191.function1900()
def function2691(function271, arg857): var342 = function271('GET', '/') var4624 = WebSocketResponse() yield from var4624.prepare(var342) var4624._reader = mock.Mock() var3438 = helpers.create_future(arg857) var3438.set_exception(asyncio.CancelledError()) var4624._reader.read = make_mocked_coro(var3438) with pytest.raises(asyncio.CancelledError): yield from var4624.function1900()
async def test_receive_timeouterror(make_request, loop): req = make_request('GET', '/') ws = WebSocketResponse() await ws.prepare(req) ws._reader = mock.Mock() res = loop.create_future() res.set_exception(asyncio.TimeoutError()) ws._reader.read = make_mocked_coro(res) with pytest.raises(asyncio.TimeoutError): await ws.receive()
def test_receive_cancelled(make_request, loop): req = make_request('GET', '/') ws = WebSocketResponse() yield from ws.prepare(req) ws._reader = mock.Mock() res = helpers.create_future(loop) res.set_exception(asyncio.CancelledError()) ws._reader.read = make_mocked_coro(res) with pytest.raises(asyncio.CancelledError): yield from ws.receive()
def test_receive_client_disconnected(make_request, loop): req = make_request('GET', '/') ws = WebSocketResponse() yield from ws.prepare(req) ws._reader = mock.Mock() exc = errors.ClientDisconnectedError() res = helpers.create_future(loop) res.set_exception(exc) ws._reader.read = make_mocked_coro(res) msg = yield from ws.receive() assert ws.closed assert msg.type == WSMsgType.CLOSED assert msg.type is msg.tp assert msg.data is None assert ws.exception() is None
async def test_receive_eofstream_in_reader(make_request, loop): req = make_request('GET', '/') ws = WebSocketResponse() await ws.prepare(req) ws._reader = mock.Mock() exc = EofStream() res = loop.create_future() res.set_exception(exc) ws._reader.read = make_mocked_coro(res) ws._payload_writer.drain = mock.Mock() ws._payload_writer.drain.return_value = loop.create_future() ws._payload_writer.drain.return_value.set_result(True) msg = await ws.receive() assert msg.type == WSMsgType.CLOSED assert ws.closed
def function2536(function271, arg68): var3402 = function271('GET', '/') var3771 = WebSocketResponse() yield from var3771.prepare(var3402) var3771._reader = mock.Mock() var2011 = ValueError() var1794 = helpers.create_future(arg68) var1794.set_exception(var2011) var3771._reader.read = make_mocked_coro(var1794) var3771._payload_writer.drain = mock.Mock() var3771._payload_writer.drain.return_value = helpers.create_future(arg68) var3771._payload_writer.drain.return_value.set_result(True) var2571 = yield from var3771.function1900() assert (var2571.type == WSMsgType.ERROR) assert (var2571.type is var2571.tp) assert (var2571.data is var2011) assert (var3771.exception() is var2011)
async def test_receive_exc_in_reader(make_request, loop): req = make_request('GET', '/') ws = WebSocketResponse() await ws.prepare(req) ws._reader = mock.Mock() exc = ValueError() res = loop.create_future() res.set_exception(exc) ws._reader.read = make_mocked_coro(res) ws._payload_writer.drain = mock.Mock() ws._payload_writer.drain.return_value = loop.create_future() ws._payload_writer.drain.return_value.set_result(True) msg = await ws.receive() assert msg.type == WSMsgType.ERROR assert msg.data is exc assert ws.exception() is exc
def function479(function271, arg926, arg363): var1796 = function271('GET', '/') var300 = WebSocketResponse() yield from var300.prepare(var1796) var300._reader = mock.Mock() var4357 = ValueError() var300._reader.read.return_value = helpers.create_future(arg926) var300._reader.read.return_value.set_exception(var4357) var300._payload_writer.drain = mock.Mock() var300._payload_writer.drain.return_value = helpers.create_future(arg926) var300._payload_writer.drain.return_value.set_result(True) yield from var300.close() assert var300.closed assert (var300.exception() is var4357) var300._closed = False var300._reader.read.return_value = helpers.create_future(arg926) var300._reader.read.return_value.set_exception(asyncio.CancelledError()) with pytest.raises(asyncio.CancelledError): yield from var300.close() assert (var300.close_code == 1006)
async def test_close_exc(make_request, loop, mocker): req = make_request('GET', '/') ws = WebSocketResponse() await ws.prepare(req) ws._reader = mock.Mock() exc = ValueError() ws._reader.read.return_value = loop.create_future() ws._reader.read.return_value.set_exception(exc) ws._payload_writer.drain = mock.Mock() ws._payload_writer.drain.return_value = loop.create_future() ws._payload_writer.drain.return_value.set_result(True) await ws.close() assert ws.closed assert ws.exception() is exc ws._closed = False ws._reader.read.return_value = loop.create_future() ws._reader.read.return_value.set_exception(asyncio.CancelledError()) with pytest.raises(asyncio.CancelledError): await ws.close() assert ws.close_code == 1006