More encoding madness

This commit is contained in:
Manuel Ebert 2015-04-05 06:38:38 +04:00
parent 109a11d18e
commit c720c15787
2 changed files with 4 additions and 4 deletions

View file

@ -167,7 +167,7 @@ def check_output_inline(context, text):
out = context.stdout_capture.getvalue()
if isinstance(out, bytes):
out = out.decode('utf-8')
assert text in out
assert text in out, text
@then('the output should not contain "{text}"')

View file

@ -80,13 +80,13 @@ class LegacyEncryptedJournal(Journal.LegacyJournal):
try:
plain_padded = decryptor.update(cipher) + decryptor.finalize()
self.config['password'] = password
if plain_padded[-1] == " ":
if plain_padded[-1] in (" ", 32):
# Ancient versions of jrnl. Do not judge me.
plain = plain_padded.rstrip(" ")
return plain_padded.decode('utf-8').rstrip(" ")
else:
unpadder = padding.PKCS7(algorithms.AES.block_size).unpadder()
plain = unpadder.update(plain_padded) + unpadder.finalize()
return plain.decode('utf-8')
return plain.decode('utf-8')
except ValueError:
return None
if password: