From: Chris Glass <tribaal@gmail.com>
Date: Wed, 25 Jun 2014 08:39:50 +0000 (+0200)
Subject: Added more bitcoin tests
X-Git-Url: https://git.novaco.in/?p=electrum-nvc.git;a=commitdiff_plain;h=bc98223188cf9a285a5024ef38783f14b37219e3

Added more bitcoin tests
---

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