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)