def function2305(function2574, function2564):
function2564.parse_frame = mock.Mock()
function2564.parse_frame.return_value = [(0, WSMsgType.TEXT, b'line1'),
(0, WSMsgType.CLOSE, b''),
(1, WSMsgType.CONTINUATION,
b'line2')]
function2564.feed_data(b'')
var4580 = function2574._buffer[0]
assert res, (WSMessage(WSMsgType.CLOSE, 0, ''), 0)
var4580 = function2574._buffer[1]
assert (var4580 == (WSMessage(WSMsgType.TEXT, 'line1line2', ''), 10))
def function950(function2574, function2564):
function2564.parse_frame = mock.Mock()
function2564.parse_frame.return_value = [
(0, WSMsgType.TEXT, b'line1'),
(0, WSMsgType.CLOSE, function2245(1002, b'test', noheader=True)),
(1, WSMsgType.CONTINUATION, b'line2')
]
function2564.feed_data(b'')
var1440 = function2574._buffer[0]
assert res, (WSMessage(WSMsgType.CLOSE, 1002, 'test'), 0)
var1440 = function2574._buffer[1]
assert (var1440 == (WSMessage(WSMsgType.TEXT, 'line1line2', ''), 10))
def test_continuation_with_close_empty(out: Any, parser: Any) -> None:
parser.parse_frame = mock.Mock()
parser.parse_frame.return_value = [
(0, WSMsgType.TEXT, b"line1", False),
(0, WSMsgType.CLOSE, b"", False),
(1, WSMsgType.CONTINUATION, b"line2", False),
]
parser.feed_data(b"")
res = out._buffer[0]
assert res, (WSMessage(WSMsgType.CLOSE, 0, ""), 0)
res = out._buffer[1]
assert res == (WSMessage(WSMsgType.TEXT, "line1line2", ""), 10)
def test_continuation_with_close(out, parser):
parser.parse_frame = mock.Mock()
parser.parse_frame.return_value = [
(0, WSMsgType.TEXT, b'line1'),
(0, WSMsgType.CLOSE, build_close_frame(1002, b'test', noheader=True)),
(1, WSMsgType.CONTINUATION, b'line2'),
]
parser.feed_data(b'')
res = out._buffer[0]
assert res, (WSMessage(WSMsgType.CLOSE, 1002, 'test'), 0)
res = out._buffer[1]
assert res == (WSMessage(WSMsgType.TEXT, 'line1line2', ''), 10)
def test_continuation_with_close_empty(out, parser):
parser.parse_frame = mock.Mock()
parser.parse_frame.return_value = [
(0, WSMsgType.TEXT, b'line1', False),
(0, WSMsgType.CLOSE, b'', False),
(1, WSMsgType.CONTINUATION, b'line2', False),
]
parser.feed_data(b'')
res = out._buffer[0]
assert res, (WSMessage(WSMsgType.CLOSE, 0, ''), 0)
res = out._buffer[1]
assert res == (WSMessage(WSMsgType.TEXT, 'line1line2', ''), 10)
def test_continuation_with_close(out, parser) -> None:
parser.parse_frame = mock.Mock()
parser.parse_frame.return_value = [
(0, WSMsgType.TEXT, b"line1", False),
(0, WSMsgType.CLOSE, build_close_frame(1002, b"test",
noheader=True), False),
(1, WSMsgType.CONTINUATION, b"line2", False),
]
parser.feed_data(b"")
res = out._buffer[0]
assert res, (WSMessage(WSMsgType.CLOSE, 1002, "test"), 0)
res = out._buffer[1]
assert res == (WSMessage(WSMsgType.TEXT, "line1line2", ""), 10)
def function2790(function2574, function2564):
function2564.parse_frame = mock.Mock()
function2564.parse_frame.return_value = [(0, WSMsgType.TEXT, b'line1'),
(0, WSMsgType.PING, b''),
(1, WSMsgType.CONTINUATION,
b'line2')]
var2761 = function2753(b'line1', WSMsgType.TEXT, is_fin=False)
function2564._feed_data(var2761)
var2430 = function2753(b'', WSMsgType.PING)
function2564._feed_data(var2430)
var4041 = function2753(b'line2', WSMsgType.CONTINUATION)
function2564._feed_data(var4041)
var2958 = function2574._buffer[0]
assert (var2958 == (WSMessage(WSMsgType.PING, b'', ''), 0))
var2958 = function2574._buffer[1]
assert (var2958 == (WSMessage(WSMsgType.TEXT, 'line1line2', ''), 10))
def function2462(function2574, function2564):
var2814 = function2753(b'line1', WSMsgType.TEXT, is_fin=False)
function2564._feed_data(var2814)
var2357 = function2753(b'line2', WSMsgType.CONTINUATION)
function2564._feed_data(var2357)
var582 = function2574._buffer[0]
assert (var582 == (WSMessage(WSMsgType.TEXT, 'line1line2', ''), 10))
def test_close_frame_info(out: Any, parser: Any) -> None:
parser.parse_frame = mock.Mock()
parser.parse_frame.return_value = [(1, WSMsgType.CLOSE, b"0112345", False)]
parser.feed_data(b"")
res = out._buffer[0]
assert res == (WSMessage(WSMsgType.CLOSE, 12337, "12345"), 0)
def test_fragmentation_header(out, parser):
data = build_frame(b'a', WSMsgType.TEXT)
parser._feed_data(data[:1])
parser._feed_data(data[1:])
res = out._buffer[0]
assert res == (WSMessage(WSMsgType.TEXT, 'a', ''), 1)
def test_fragmentation_header(out: Any, parser: Any) -> None:
data = build_frame(b"a", WSMsgType.TEXT)
parser._feed_data(data[:1])
parser._feed_data(data[1:])
res = out._buffer[0]
assert res == (WSMessage(WSMsgType.TEXT, "a", ""), 1)
def test_close_frame_info(out, parser):
parser.parse_frame = mock.Mock()
parser.parse_frame.return_value = [(1, WSMsgType.CLOSE, b'0112345', False)]
parser.feed_data(b'')
res = out._buffer[0]
assert res == (WSMessage(WSMsgType.CLOSE, 12337, '12345'), 0)
def test_continuation(out, parser):
data1 = build_frame(b'line1', WSMsgType.TEXT, is_fin=False)
parser._feed_data(data1)
data2 = build_frame(b'line2', WSMsgType.CONTINUATION)
parser._feed_data(data2)
res = out._buffer[0]
assert res == (WSMessage(WSMsgType.TEXT, 'line1line2', ''), 10)
def test_continuation(out: Any, parser: Any) -> None:
data1 = build_frame(b"line1", WSMsgType.TEXT, is_fin=False)
parser._feed_data(data1)
data2 = build_frame(b"line2", WSMsgType.CONTINUATION)
parser._feed_data(data2)
res = out._buffer[0]
assert res == (WSMessage(WSMsgType.TEXT, "line1line2", ""), 10)
def test_continuation(out, parser):
parser.parse_frame = mock.Mock()
parser.parse_frame.return_value = [(0, WSMsgType.TEXT, b'line1'),
(1, WSMsgType.CONTINUATION, b'line2')]
parser._feed_data(b'')
res = out._buffer[0]
assert res == (WSMessage(WSMsgType.TEXT, 'line1line2', ''), 10)
def test_continuation_with_ping(out, parser):
parser.parse_frame = mock.Mock()
parser.parse_frame.return_value = [
(0, WSMsgType.TEXT, b'line1', False),
(0, WSMsgType.PING, b'', False),
(1, WSMsgType.CONTINUATION, b'line2', False),
]
data1 = build_frame(b'line1', WSMsgType.TEXT, is_fin=False)
parser._feed_data(data1)
data2 = build_frame(b'', WSMsgType.PING)
parser._feed_data(data2)
data3 = build_frame(b'line2', WSMsgType.CONTINUATION)
parser._feed_data(data3)
res = out._buffer[0]
assert res == (WSMessage(WSMsgType.PING, b'', ''), 0)
res = out._buffer[1]
assert res == (WSMessage(WSMsgType.TEXT, 'line1line2', ''), 10)
def test_continuation_with_ping(out: Any, parser: Any) -> None:
parser.parse_frame = mock.Mock()
parser.parse_frame.return_value = [
(0, WSMsgType.TEXT, b"line1", False),
(0, WSMsgType.PING, b"", False),
(1, WSMsgType.CONTINUATION, b"line2", False),
]
data1 = build_frame(b"line1", WSMsgType.TEXT, is_fin=False)
parser._feed_data(data1)
data2 = build_frame(b"", WSMsgType.PING)
parser._feed_data(data2)
data3 = build_frame(b"line2", WSMsgType.CONTINUATION)
parser._feed_data(data3)
res = out._buffer[0]
assert res == (WSMessage(WSMsgType.PING, b"", ""), 0)
res = out._buffer[1]
assert res == (WSMessage(WSMsgType.TEXT, "line1line2", ""), 10)
async def handle_ws_text_message(
self,
request: web.Request,
ws: web.WebSocketResponse,
ws_text_message: http.WSMessage,
) -> None:
message: dict = ws_text_message.json()
message_type: str = message["type"]
if message_type == GQL_CONNECTION_INIT:
await self.handle_connection_init(ws)
elif message_type == GQL_CONNECTION_TERMINATE:
await self.handle_connection_terminate(ws)
elif message_type == GQL_START:
await self.handle_start(request, ws, message)
elif message_type == GQL_STOP:
await self.handle_stop(message)
def function2724(function2574, function2564):
function2564.parse_frame = mock.Mock()
function2564.parse_frame.return_value = [(1, WSMsgType.CLOSE, b'0112345')]
function2564.feed_data(b'')
var4135 = function2574._buffer[0]
assert (var4135 == (WSMessage(WSMsgType.CLOSE, 12337, '12345'), 0))
def function1009(function2574, function2564):
var1174 = function2753(b'a', WSMsgType.TEXT)
function2564._feed_data(var1174[:1])
function2564._feed_data(var1174[1:])
var1658 = function2574._buffer[0]
assert (var1658 == (WSMessage(WSMsgType.TEXT, 'a', ''), 1))
