Added more bitcoin tests

diff --git a/lib/tests/test_bitcoin.py b/lib/tests/test_bitcoin.py
index c1146a4..aa0dd98 100644 (file)
--- 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"