def test_payload_enc(self):
packets = (
PayloadsTestCase.GOOD_STR_PACKET,
)
expected = [
PayloadsTestCase.GOOD_STR_PACKET_PAYLOAD,
]
actual = list(encbinpayloadsgen(packets))
self.assertEqual(expected, actual)
packets = (
PayloadsTestCase.GOOD_BIN_PACKET,
)
expected = [
PayloadsTestCase.GOOD_BIN_PACKET_PAYLOAD,
]
actual = list(encbinpayloadsgen(packets))
self.assertEqual(expected, actual)
packets = (
PayloadsTestCase.GOOD_STR_PACKET,
PayloadsTestCase.GOOD_BIN_PACKET,
PayloadsTestCase.GOOD_STR_PACKET,
PayloadsTestCase.GOOD_BIN_PACKET,
)
expected = [
PayloadsTestCase.GOOD_STR_PACKET_PAYLOAD,
PayloadsTestCase.GOOD_BIN_PACKET_PAYLOAD,
PayloadsTestCase.GOOD_STR_PACKET_PAYLOAD,
PayloadsTestCase.GOOD_BIN_PACKET_PAYLOAD,
]
actual = list(encbinpayloadsgen(packets))
self.assertEqual(expected, actual)
def test_payload_enc_bad_type(self):
packets = (
PayloadsTestCase.GOOD_STR_PACKET,
PayloadsTestCase.GOOD_BIN_PACKET,
42,
PayloadsTestCase.GOOD_STR_PACKET,
PayloadsTestCase.GOOD_BIN_PACKET,
)
expected = [
PayloadsTestCase.GOOD_STR_PACKET_PAYLOAD,
PayloadsTestCase.GOOD_BIN_PACKET_PAYLOAD,
]
actual = []
with self.assertRaisesRegex(TypeError, r'^packet type must be one of bytes or str, not .+ for packet\[2\]$'):
for i in encbinpayloadsgen(packets):
actual.append(i)
self.assertEqual(expected, actual)
