From 8fac5909a294f7c2271c3ccfffd93815895e8195 Mon Sep 17 00:00:00 2001
From: "Sybren A. St?vel" <sybren@stuvel.eu>
Date: Mon, 20 Jun 2011 01:06:39 +0200
Subject: Added block padding to be able to work with leading zeroes, breaks
 all kind of stuff

---
 tests/test_binary.py  | 37 +++++++++++++++++++++++++++++++++++++
 tests/test_strings.py |  7 +++----
 2 files changed, 40 insertions(+), 4 deletions(-)
 create mode 100644 tests/test_binary.py

(limited to 'tests')

diff --git a/tests/test_binary.py b/tests/test_binary.py
new file mode 100644
index 0000000..f770b72
--- /dev/null
+++ b/tests/test_binary.py
@@ -0,0 +1,37 @@
+'''Tests string operations.'''
+
+import struct
+import unittest
+
+import rsa
+
+class BinaryTest(unittest.TestCase):
+
+    def setUp(self):
+        (self.pub, self.priv) = rsa.newkeys(64)
+
+    def test_enc_dec(self):
+
+        message = struct.pack('>IIII', 0, 0, 0, 1) + 20 * '\x00'
+        print "\tMessage:   %r" % message
+
+        encrypted = rsa.encrypt(message, self.pub)
+        print "\tEncrypted: %r" % encrypted
+
+        decrypted = rsa.decrypt(encrypted, self.priv)
+        print "\tDecrypted: %r" % decrypted
+
+        self.assertEqual(message, decrypted)
+
+    def test_sign_verify(self):
+
+        message = struct.pack('>IIII', 0, 0, 0, 1) + 20 * '\x00'
+        print "\tMessage:   %r" % message
+
+        signed = rsa.sign(message, self.priv)
+        print "\tSigned:    %r" % signed
+
+        verified = rsa.verify(signed, self.pub)
+        print "\tVerified:  %r" % verified
+
+        self.assertEqual(message, verified)
diff --git a/tests/test_strings.py b/tests/test_strings.py
index c5803e4..8baa63d 100644
--- a/tests/test_strings.py
+++ b/tests/test_strings.py
@@ -11,8 +11,7 @@ class StringTest(unittest.TestCase):
 
     def test_enc_dec(self):
 
-        # TODO: test with unicode strings and non-ascii chars
-        message = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+        message = u"Euro=\u20ac ABCDEFGHIJKLMNOPQRSTUVWXYZ".encode('utf-8')
         print "\tMessage:   %s" % message
 
         encrypted = rsa.encrypt(message, self.pub)
@@ -25,8 +24,8 @@ class StringTest(unittest.TestCase):
 
     def test_sign_verify(self):
 
-        # TODO: test with unicode strings and non-ascii chars
-        message = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+        message = u"Euro=\u20ac ABCDEFGHIJKLMNOPQRSTUVWXYZ".encode('utf-8')
+        print "\tMessage:   %s" % message
 
         signed = rsa.sign(message, self.priv)
         print "\tSigned:    %s" % signed
-- 
cgit v1.2.1