def test_padding(self): key = base64.urlsafe_b64encode(os.urandom(16)) feistel = MyFeistel(key, 10) msg = os.urandom(41) h_msg = msg.encode('hex') length = len(msg) L, R = '0' + h_msg[:length], '0' + h_msg[length:] assert feistel._pad_string(msg)[1] == (L + R).decode('hex')
def test_paddingAndUnpadding(self): key = base64.urlsafe_b64encode(os.urandom(16)) feistel = MyFeistel(key, 10) msg = os.urandom(41) assert feistel._unpad_string(True, feistel._pad_string(msg)[1]) == msg