Testing (#11)

* testing

* test rsa

* tests

setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-__version__ = "0.0.1"
+__version__ = "0.0.2"
 
 with open("README.md", "r", encoding="utf-8") as fh:
     long_description = fh.read()

src/crypto_pkg/rsa/rsa_scheme.py

@@ -60,14 +60,12 @@ def encrypt_message(message: str, e: int, n: int):
         m = str_to_int(message)
         if m < 0 or m > n - 1:
             raise MessageTooBigError
-        c = RSA.encrypt(m=m, e=e, n=n)
-        return int_to_str(int(c))
+        return RSA.encrypt(m=m, e=e, n=n)
 
     @staticmethod
-    def decrypt_message(cipher_text: str, d: int, n: int):
-        c = str_to_int(cipher_text)
-        m = RSA.decrypt(c=c, d=d, n=n)
-        return int_to_str(int(m))
+    def decrypt_message(cipher_text: int, d: int, n: int):
+        m = RSA.decrypt(c=cipher_text, d=d, n=n)
+        return int_to_str(m)
 
     @staticmethod
     def encrypt(m, e, n):

tests/test_rsa.py

@@ -0,0 +1,15 @@
+import unittest
+
+from crypto_pkg.rsa.rsa_scheme import RSA
+
+
+class TestRSA(unittest.TestCase):
+
+    def test_rsa(self):
+        rsa = RSA(k=100)
+        primes = rsa.generate_primes()
+        message = 'hello'
+        keys = rsa.generate_keys(primes.q, primes.p)
+        c = RSA.encrypt_message(message=message, e=keys.public, n=keys.modulus)
+        m = RSA.decrypt_message(cipher_text=c, d=keys.private, n=keys.modulus)
+        self.assertEqual(message, m)