示例#1
0
    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')
示例#2
0
 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)