Added tests for ec key creation from raw material and for private key

[oss/ctypescrypto.git] / tests / testec.py

diff --git a/tests/testec.py b/tests/testec.py
new file mode 100644 (file)
index 0000000..496a747
--- /dev/null
+++ b/tests/testec.py
@@ -0,0 +1,32 @@
+from ctypescrypto.oid import Oid
+from ctypescrypto.ec import create
+from base64 import b16decode
+import unittest
+
+
+
+class TestEcCreation(unittest.TestCase):
+       ec1priv="""-----BEGIN PRIVATE KEY-----
+MIGEAgEAMBAGByqGSM49AgEGBSuBBAAKBG0wawIBAQQgKnG6neqZvB98EEuuxnHs
+fv+L/5abuNNG20wzUqRpncOhRANCAARWKXWeUZ6WiCKZ2kHx87jmJyx0G3ZB1iQC
++Gp2AJYswbQPhGPigKolzIbZYfwnn7QOca6N8QDhPAn3QQK8trZI
+-----END PRIVATE KEY-----
+"""
+       bigkey="""-----BEGIN PRIVATE KEY-----
+MHUCAQAwEAYHKoZIzj0CAQYFK4EEAAoEXjBcAgEBBBEBRVEjGVC3X8RALaFzL8m+
+vqFEA0IABJFmwom5+QXlX549+fadfzVrSiIJX4lPRxVxSqS1Zgav8YHrlmvkrLXP
++eFrZtgJvpTiFPBsk/0JEJmvmEmSVec=
+-----END PRIVATE KEY-----
+"""
+       def test_keyone(self):
+               key=create(Oid("secp256k1"),b16decode("2A71BA9DEA99BC1F7C104BAEC671EC7EFF8BFF969BB8D346DB4C3352A4699DC3",True))
+                       
+               out=key.exportpriv()
+               self.assertEqual(out,self.ec1priv)
+
+       def test_bignum(self):
+               keyval='\xff'*32
+               key=create(Oid("secp256k1"),keyval)
+               self.assertEqual(key.exportpriv(),self.bigkey)
+if __name__ == "__main__":
+       unittest.main()