def test_close_idempotent(self): req = self.make_request("GET", "/") ws = WebSocketResponse() self.loop.run_until_complete(ws.prepare(req)) writer = mock.Mock() ws._writer = writer self.assertTrue(self.loop.run_until_complete(ws.close(code=1, message="message1"))) self.assertTrue(ws.closed) self.assertFalse(self.loop.run_until_complete(ws.close(code=2, message="message2")))
def test_close_idempotent(self): req = self.make_request('GET', '/') ws = WebSocketResponse() ws.start(req) writer = mock.Mock() ws._writer = writer self.assertTrue( self.loop.run_until_complete(ws.close(code=1, message='message1'))) self.assertTrue(ws.closed) self.assertFalse( self.loop.run_until_complete(ws.close(code=2, message='message2')))
def test_double_close(self): req = self.make_request('GET', '/') ws = WebSocketResponse() ws.start(req) writer = mock.Mock() ws._writer = writer ws.close(code=1, message='message1') self.assertTrue(ws.closing) with self.assertRaisesRegex(RuntimeError, 'Already closing'): ws.close(code=2, message='message2') self.assertTrue(ws.closing) writer.close.assert_called_once_with(1, 'message1')
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()
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())