LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Packages Projects Releases 282 Wiki Activity

separate cases and make it deterministic

Browse source
This commit is contained in:
Victor Shyba 2020-01-13 18:04:42 -03:00 committed by Lex Berezhny
parent 3946bc6662
commit a56dd66c98
1 changed files with 15 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
tests/unit/wallet/test_hash.py
View file

@ -41,7 +41,11 @@ class TestAESEncryptDecrypt(TestCase):
def test_edge_case_invalid_password_valid_padding_invalid_unicode(self): def test_edge_case_invalid_password_valid_padding_invalid_unicode(self):
with self.assertRaises(InvalidPasswordError): with self.assertRaises(InvalidPasswordError):
aes_decrypt('notbubblegum', 'gy3/mNq3FWB/xAXirOQnlAqQLuvhLGXZaeGBUIg1w6yY4PDLDT7BU83XOfBsJoluWU5zEU4+upOFH35HDqyV8EMQhcKSufN9WkT1izEbFtweBUTK8nTSkV7NBppE1Jaz') aes_decrypt(
'notbubblegum',
'gy3/mNq3FWB/xAXirOQnlAqQLuvhLGXZaeGBUIg1w6yY4PDLDT7BU83XOfBsJol'
'uWU5zEU4+upOFH35HDqyV8EMQhcKSufN9WkT1izEbFtweBUTK8nTSkV7NBppE1Jaz'
)
def test_better_encrypt_decrypt(self): def test_better_encrypt_decrypt(self):
self.assertEqual( self.assertEqual(
@ -50,16 +54,15 @@ class TestAESEncryptDecrypt(TestCase):
'super secret', 'super secret',
better_aes_encrypt('super secret', b'valuable value'))) better_aes_encrypt('super secret', b'valuable value')))
def test_better_decrypt_error(self): @mock.patch('os.urandom', side_effect=lambda i: b'd'*i)
try: def test_better_decrypt_error(self, _):
value = better_aes_encrypt('super secret', b'valuable value') with self.assertRaises(InvalidPasswordError):
decrypted = better_aes_decrypt( better_aes_decrypt(
'super secret but wrong', 'super secret but wrong',
value better_aes_encrypt('super secret', b'valuable value')
) )
print(f'{value} accepted as valid decryption.')
# FIXME: example that hit here: czo4MTkyOjE2OjE6VrwsN8FSJlegxHVEQePoyjWT1k8yAXBCUbbGCFKcsNY= def test_edge_case_invalid_password_valid_everything(self):
self.assertNotEqual(decrypted, b'valuable value') value = b'czo4MTkyOjE2OjE6VrwsN8FSJlegxHVEQePoyjWT1k8yAXBCUbbGCFKcsNY='
except InvalidPasswordError: self.assertEqual(b'valuable value', better_aes_decrypt('super secret', value))
# success self.assertNotEqual(b'valuable value', better_aes_decrypt('super secret but wrong', value))
pass