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)))
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))