Example #1
0
    def test_payload_enc_dec(self):
        packets = []
        raw = encbinpayloads(packets)
        self.assertEqual(packets, list(decbinpayloadsgen(raw)))

        packets = [
            PayloadsTestCase.GOOD_STR_PACKET,
        ]
        raw = encbinpayloads(packets)
        self.assertEqual(packets, list(decbinpayloadsgen(raw)))

        packets = [
            PayloadsTestCase.GOOD_BIN_PACKET,
        ]
        raw = encbinpayloads(packets)
        self.assertEqual(packets, list(decbinpayloadsgen(raw)))

        packets = [
            PayloadsTestCase.GOOD_STR_PACKET,
            PayloadsTestCase.GOOD_BIN_PACKET,
            PayloadsTestCase.GOOD_STR_PACKET,
            PayloadsTestCase.GOOD_BIN_PACKET,
        ]
        raw = encbinpayloads(packets)
        self.assertEqual(packets, list(decbinpayloadsgen(raw)))
Example #2
0
    def test_payload_dec_enc(self):
        raw = b''
        packets = list(decbinpayloadsgen(raw))
        self.assertEqual(raw, encbinpayloads(packets))

        raw = PayloadsTestCase.GOOD_STR_PACKET_PAYLOAD
        packets = list(decbinpayloadsgen(raw))
        self.assertEqual(raw, encbinpayloads(packets))

        raw = PayloadsTestCase.GOOD_BIN_PACKET_PAYLOAD
        packets = list(decbinpayloadsgen(raw))
        self.assertEqual(raw, encbinpayloads(packets))

        raw = PayloadsTestCase.GOOD_BIN_PACKET_PAYLOAD \
            + PayloadsTestCase.GOOD_STR_PACKET_PAYLOAD \
            + PayloadsTestCase.GOOD_BIN_PACKET_PAYLOAD \
            + PayloadsTestCase.GOOD_STR_PACKET_PAYLOAD
        packets = list(decbinpayloadsgen(raw))
        self.assertEqual(raw, encbinpayloads(packets))