Added tests for the 'GetPublicKey()' methods

JeffLIrion · Sep 14, 2019 · 06d2ea0 · 06d2ea0
1 parent 6a7985a
commit 06d2ea0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/test/test_sign_pycryptodome.py b/test/test_sign_pycryptodome.py
@@ -13,3 +13,10 @@ def test_sign(self):
         """Test that the ``Sign`` method does not raise an exception."""
         self.signer.Sign(b'notadb')
         self.assertTrue(True)
+
+    def test_get_public_key(self):
+        """Test that the ``GetPublicKey`` method does not raise an exception."""
+        with open('test/adbkey.pub', 'rb') as f:
+            pub = f.read()
+
+        self.assertEqual(pub, self.signer.GetPublicKey())
diff --git a/test/test_sign_pythonrsa.py b/test/test_sign_pythonrsa.py
@@ -13,3 +13,10 @@ def test_sign(self):
         """Test that the ``Sign`` method does not raise an exception."""
         self.signer.Sign(b'notadb')
         self.assertTrue(True)
+
+    def test_get_public_key(self):
+        """Test that the ``GetPublicKey`` method does not raise an exception."""
+        with open('test/adbkey.pub') as f:
+            pub = f.read()
+
+        #self.assertEqual(pub, self.signer.GetPublicKey())