def test_raise_on_text_after_text(self): reader = asyncio.StreamReader(loop=self.loop) parser = WebSocketParser(reader) frame1 = FrameBuilder.text("Hello", fin=False) frame2 = FrameBuilder.text(" world!", fin=True) self.loop.call_soon(lambda: reader.feed_data(b''.join((frame1, frame2)))) task = asyncio.Task(parser.get_message(), loop=self.loop) self.assertRaises(WebSocketFormatException, self.loop.run_until_complete, task)
def test_eof_when_reading_length(self): reader = asyncio.StreamReader(loop=self.loop) parser = WebSocketParser(reader) frame = FrameBuilder.text(' ' * 129)[:2] self.loop.call_soon(feed_data_and_eof, reader, frame) task = asyncio.Task(parser.get_message(), loop=self.loop) self.assertIsNone(self.loop.run_until_complete(task)) frame = FrameBuilder.text(' ' * 67000)[:2] self.loop.call_soon(feed_data_and_eof, reader, frame) task = asyncio.Task(parser.get_message(), loop=self.loop) self.assertIsNone(self.loop.run_until_complete(task))
def test_eight_byte_length(self): reader = asyncio.StreamReader(loop=self.loop) parser = WebSocketParser(reader) frame = FrameBuilder.text(" "*67000, fin=True) self.loop.call_soon(lambda: reader.feed_data(frame)) task = asyncio.Task(parser.get_message(), loop=self.loop) msg = self.loop.run_until_complete(task) self.assertEqual(len(msg.payload), 67000)
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!')
def test_text_frame(self, _build_method): data = "Hello" FrameBuilder.text(data) _build_method.assert_called_with(opcode=OpCode.text, fin=True, masked=True, payload=data)
def test_build(self): frame = FrameBuilder.text("Hello world", masked=False) self.assertEqual(frame, b'\x81\x0bHello world')