def function113(function271): var2027 = function271('GET', '/') var2897 = WebSocketResponse() yield from var2897.prepare(var2027) var962 = ValueError() var2897._writer = mock.Mock() var2897._writer.close.side_effect = var962 yield from var2897.close() assert var2897.closed assert (var2897.exception() is var962) var2897._closed = False var2897._writer.close.side_effect = asyncio.CancelledError() with pytest.raises(asyncio.CancelledError): yield from var2897.close()
def test_close_exc2(self): req = self.make_request("GET", "/") ws = WebSocketResponse() self.loop.run_until_complete(ws.prepare(req)) exc = ValueError() self.writer.close.side_effect = exc ws._writer = self.writer self.loop.run_until_complete(ws.close()) self.assertTrue(ws.closed) self.assertIs(ws.exception(), exc) ws._closed = False self.writer.close.side_effect = asyncio.CancelledError() self.assertRaises(asyncio.CancelledError, self.loop.run_until_complete, ws.close())
async def test_close_exc2(make_request): req = make_request('GET', '/') ws = WebSocketResponse() await ws.prepare(req) exc = ValueError() ws._writer = mock.Mock() ws._writer.close.side_effect = exc await ws.close() assert ws.closed assert ws.exception() is exc ws._closed = False ws._writer.close.side_effect = asyncio.CancelledError() with pytest.raises(asyncio.CancelledError): await ws.close()
def test_close_exc2(self): req = self.make_request('GET', '/') ws = WebSocketResponse() ws.start(req) exc = ValueError() self.writer.close.side_effect = exc ws._writer = self.writer self.loop.run_until_complete(ws.close()) self.assertTrue(ws.closed) self.assertIs(ws.exception(), exc) ws._closed = False self.writer.close.side_effect = asyncio.CancelledError() self.assertRaises(asyncio.CancelledError, self.loop.run_until_complete, ws.close())
def test_close_exc(make_request, reader, loop): req = make_request('GET', '/') ws = WebSocketResponse() yield from ws.prepare(req) exc = ValueError() reader.read.return_value = helpers.create_future(loop) reader.read.return_value.set_exception(exc) yield from ws.close() assert ws.closed assert ws.exception() is exc ws._closed = False reader.read.return_value = helpers.create_future(loop) reader.read.return_value.set_exception(asyncio.CancelledError()) with pytest.raises(asyncio.CancelledError): yield from ws.close() assert ws.close_code == 1006
def test_close_exc(self): req = self.make_request("GET", "/") reader = self.reader.set_parser.return_value = mock.Mock() ws = WebSocketResponse() self.loop.run_until_complete(ws.prepare(req)) exc = ValueError() reader.read.return_value = asyncio.Future(loop=self.loop) reader.read.return_value.set_exception(exc) self.loop.run_until_complete(ws.close()) self.assertTrue(ws.closed) self.assertIs(ws.exception(), exc) ws._closed = False reader.read.return_value = asyncio.Future(loop=self.loop) reader.read.return_value.set_exception(asyncio.CancelledError()) self.assertRaises(asyncio.CancelledError, self.loop.run_until_complete, ws.close()) self.assertEqual(ws.close_code, 1006)
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)
def test_close_exc(self): req = self.make_request('GET', '/') reader = self.reader.set_parser.return_value = mock.Mock() ws = WebSocketResponse() ws.start(req) exc = ValueError() reader.read.return_value = asyncio.Future(loop=self.loop) reader.read.return_value.set_exception(exc) self.loop.run_until_complete(ws.close()) self.assertTrue(ws.closed) self.assertIs(ws.exception(), exc) ws._closed = False reader.read.return_value = asyncio.Future(loop=self.loop) reader.read.return_value.set_exception(asyncio.CancelledError()) self.assertRaises(asyncio.CancelledError, self.loop.run_until_complete, ws.close()) self.assertEqual(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