tests/testec.py

   1 from ctypescrypto.oid import Oid
   2 from ctypescrypto.ec import create
   3 from base64 import b16decode
   4 import unittest
   5
   6
   7
   8 class TestEcCreation(unittest.TestCase):
   9     ec1priv="""-----BEGIN PRIVATE KEY-----
  10 MIGEAgEAMBAGByqGSM49AgEGBSuBBAAKBG0wawIBAQQgKnG6neqZvB98EEuuxnHs
  11 fv+L/5abuNNG20wzUqRpncOhRANCAARWKXWeUZ6WiCKZ2kHx87jmJyx0G3ZB1iQC
  12 +Gp2AJYswbQPhGPigKolzIbZYfwnn7QOca6N8QDhPAn3QQK8trZI
  13 -----END PRIVATE KEY-----
  14 """
  15     bigkey="""-----BEGIN PRIVATE KEY-----
  16 MIGEAgEAMBAGByqGSM49AgEGBSuBBAAKBG0wawIBAQQgAAAAAAAAAAAAAAAAAAAA
  17 AUVRIxlQt1/EQC2hcy/Jvr6hRANCAASRZsKJufkF5V+ePfn2nX81a0oiCV+JT0cV
  18 cUqktWYGr/GB65Zr5Ky1z/nha2bYCb6U4hTwbJP9CRCZr5hJklXn
  19 -----END PRIVATE KEY-----
  20 """
  21     def test_keyone(self):
  22         key=create(Oid("secp256k1"),b16decode("2A71BA9DEA99BC1F7C104BAEC671EC7EFF8BFF969BB8D346DB4C3352A4699DC3",True))
  23
  24         out=key.exportpriv()
  25         self.assertEqual(out,self.ec1priv)
  26
  27     def test_bignum(self):
  28         keyval='\xff'*32
  29         key=create(Oid("secp256k1"),keyval)
  30         self.assertEqual(key.exportpriv(),self.bigkey)
  31 if __name__ == "__main__":
  32     unittest.main()