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)
