From 3075cf854b5f5cace8e27d28d79f96b52aeef052 Mon Sep 17 00:00:00 2001
From: "Sybren A. St?vel" <sybren@stuvel.eu>
Date: Sat, 30 Jul 2011 20:39:02 +0200
Subject: Removed insecure code from rsa/__init__.py

---
 tests/test_integers.py | 10 +++++-----
 tests/test_strings.py  | 15 +--------------
 2 files changed, 6 insertions(+), 19 deletions(-)

(limited to 'tests')

diff --git a/tests/test_integers.py b/tests/test_integers.py
index 1e318ab..111f8ab 100644
--- a/tests/test_integers.py
+++ b/tests/test_integers.py
@@ -2,7 +2,7 @@
 
 import unittest
 
-import rsa
+import rsa.core
 
 class IntegerTest(unittest.TestCase):
 
@@ -14,10 +14,10 @@ class IntegerTest(unittest.TestCase):
         message = 42
         print "\tMessage:   %d" % message
 
-        encrypted = rsa.encrypt_int(message, self.pub['e'], self.pub['n'])
+        encrypted = rsa.core.encrypt_int(message, self.pub.e, self.pub.n)
         print "\tEncrypted: %d" % encrypted
 
-        decrypted = rsa.decrypt_int(encrypted, self.priv['d'], self.pub['n'])
+        decrypted = rsa.core.decrypt_int(encrypted, self.priv.d, self.pub.n)
         print "\tDecrypted: %d" % decrypted
 
         self.assertEqual(message, decrypted)
@@ -26,10 +26,10 @@ class IntegerTest(unittest.TestCase):
 
         message = 42
 
-        signed = rsa.encrypt_int(message,self.priv['d'], self.pub['n'])
+        signed = rsa.core.encrypt_int(message,self.priv.d, self.pub.n)
         print "\tSigned:    %d" % signed
 
-        verified = rsa.decrypt_int(signed, self.pub['e'],self.pub['n'])
+        verified = rsa.core.decrypt_int(signed, self.pub.e,self.pub.n)
         print "\tVerified:  %d" % verified
 
         self.assertEqual(message, verified)
diff --git a/tests/test_strings.py b/tests/test_strings.py
index 8baa63d..38fae06 100644
--- a/tests/test_strings.py
+++ b/tests/test_strings.py
@@ -7,7 +7,7 @@ import rsa
 class StringTest(unittest.TestCase):
 
     def setUp(self):
-        (self.pub, self.priv) = rsa.newkeys(64)
+        (self.pub, self.priv) = rsa.newkeys(384)
 
     def test_enc_dec(self):
 
@@ -22,16 +22,3 @@ class StringTest(unittest.TestCase):
 
         self.assertEqual(message, decrypted)
 
-    def test_sign_verify(self):
-
-        message = u"Euro=\u20ac ABCDEFGHIJKLMNOPQRSTUVWXYZ".encode('utf-8')
-        print "\tMessage:   %s" % message
-
-        signed = rsa.sign(message, self.priv)
-        print "\tSigned:    %s" % signed
-
-        verified = rsa.verify(signed, self.pub)
-        print "\tVerified:  %s" % verified
-
-        self.assertEqual(message, verified)
-
-- 
cgit v1.2.1