def test_parse_with_empty_data(self): parser = Parser(bytearray(0)) ext = PaddingExtension() ext.parse(parser) self.assertEqual(bytearray(b''), ext.paddingData)
def test_parse_with_nonempty_data(self): parser = Parser( bytearray(b'\x00\x00' + # 1st and 2nd null byte b'\x00\x00')) # 3rd and 4th null byte ext = PaddingExtension() ext.parse(parser) self.assertEqual(bytearray(b'\x00\x00\x00\x00'), ext.paddingData)
def test_parse_with_nonempty_data(self): parser = Parser(bytearray( b'\x00\x00' + # 1st and 2nd null byte b'\x00\x00')) # 3rd and 4th null byte ext = PaddingExtension() ext.parse(parser) self.assertEqual(bytearray(b'\x00\x00\x00\x00'), ext.paddingData)