Python FlushMessageList.generate示例

示例#1

文件： test_tlsfuzzer_messages.py 项目： NetPainter/tlsfuzzer

    def test_with_different_message_types(self):
        msg_list = [messages.Message(20, bytearray(b"\x20")), messages.Message(30, bytearray(b"\x10"))]

        msg_gen = FlushMessageList(msg_list)

        with self.assertRaises(AssertionError):
            msg_gen.generate(None)

示例#2

    def test_with_different_message_types(self):
        msg_list = [
            messages.Message(20, bytearray(b'\x20')),
            messages.Message(30, bytearray(b'\x10'))
        ]

        msg_gen = FlushMessageList(msg_list)

        with self.assertRaises(AssertionError):
            msg_gen.generate(None)

示例#3

文件： test_tlsfuzzer_messages.py 项目： majinxin2003/tlsfuzzer

    def test_with_message_list(self):
        msg_list = []

        msg_gen = FlushMessageList(msg_list)

        self.assertEqual(msg_list, [])

        msg_list.append(messages.Message(20, bytearray(b'\x20\x30')))
        msg_list.append(messages.Message(20, bytearray(b'\x60\x70')))

        msg = msg_gen.generate(None)

        self.assertEqual(msg.contentType, 20)
        self.assertEqual(msg.write(), bytearray(b'\x20\x30\x60\x70'))

        self.assertEqual(msg_list, [])

示例#4

    def test_with_message_list(self):
        msg_list = []

        msg_gen = FlushMessageList(msg_list)

        self.assertEqual(msg_list, [])

        msg_list.append(messages.Message(20, bytearray(b'\x20\x30')))
        msg_list.append(messages.Message(20, bytearray(b'\x60\x70')))

        msg = msg_gen.generate(None)

        self.assertEqual(msg.contentType, 20)
        self.assertEqual(msg.write(), bytearray(b'\x20\x30\x60\x70'))

        self.assertEqual(msg_list, [])