def test_binary_message(self): reader = asyncio.StreamReader(loop=self.loop) parser = WebSocketParser(reader) task = asyncio.Task(parser.get_message(), loop=self.loop) frame = FrameBuilder.binary(b'binary') self.loop.call_soon(lambda: reader.feed_data(frame)) frame = self.loop.run_until_complete(task) self.assertEqual(frame.opcode, OpCode.binary) self.assertEqual(frame.payload, b'binary')
def test_binary_frame(self, _build_method): data = b"Hello" FrameBuilder.binary(data) _build_method.assert_called_with(opcode=OpCode.binary, fin=True, masked=True, payload=data)