From bc98223188cf9a285a5024ef38783f14b37219e3 Mon Sep 17 00:00:00 2001 From: Chris Glass Date: Wed, 25 Jun 2014 10:39:50 +0200 Subject: [PATCH] Added more bitcoin tests --- lib/tests/test_bitcoin.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/lib/tests/test_bitcoin.py b/lib/tests/test_bitcoin.py index c1146a48..aa0dd98d 100644 --- a/lib/tests/test_bitcoin.py +++ b/lib/tests/test_bitcoin.py @@ -81,6 +81,26 @@ class Test_bitcoin(unittest.TestCase): dec = pw_decode(enc, password) self.assertEqual(dec, payload) + def test_aes_encode_without_password(self): + """When not passed a password, pw_encode is noop on the payload.""" + payload = u'\u66f4\u7a33\u5b9a\u7684\u4ea4\u6613\u5e73\u53f0' + enc = pw_encode(payload, None) + self.assertEqual(payload, enc) + + def test_aes_deencode_without_password(self): + """When not passed a password, pw_decode is noop on the payload.""" + payload = u'\u66f4\u7a33\u5b9a\u7684\u4ea4\u6613\u5e73\u53f0' + enc = pw_decode(payload, None) + self.assertEqual(payload, enc) + + def test_aes_decode_with_invalid_password(self): + """pw_decode raises an Exception when supplied an invalid password.""" + payload = u"blah" + password = u"uber secret" + wrong_password = u"not the password" + enc = pw_encode(payload, password) + self.assertRaises(Exception, pw_decode, enc, wrong_password) + def test_hash(self): """Make sure the Hash function does sha256 twice""" payload = u"test"