def test_ping_frame(self, _build_method):
data = b"Hello"
FrameBuilder.ping(data)
_build_method.assert_called_with(opcode=OpCode.ping, fin=True, masked=True, payload=data)
FrameBuilder.ping(data, masked=False)
_build_method.assert_called_with(opcode=OpCode.ping, fin=True, masked=False, payload=data)
FrameBuilder.ping(masked=False)
_build_method.assert_called_with(opcode=OpCode.ping, fin=True, masked=False, payload=b'')
def test_ping_frame_between_text_frames(self):
reader = asyncio.StreamReader(loop=self.loop)
parser = WebSocketParser(reader)
first = FrameBuilder.text("Hello", fin=False)
ping = FrameBuilder.ping()
second = FrameBuilder.continuation(" world!", fin=True)
frame = FrameBuilder.build(fin=False, opcode=OpCode.close, payload=b'', masked=True)
self.loop.call_soon(lambda: reader.feed_data(b''.join((first, ping, second))))
task = asyncio.Task(parser.get_message(), loop=self.loop)
message = self.loop.run_until_complete(task)
self.assertEqual(message.opcode, OpCode.ping)
task = asyncio.Task(parser.get_message(), loop=self.loop)
message = self.loop.run_until_complete(task)
self.assertEqual(message.opcode, OpCode.text)
self.assertEqual(message.payload, 'Hello world!')
