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()
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_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
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
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
