tests/testKeypair.py

   1 #!/usr/bin/env python3
   2 import sys
   3 sys.path.append('..')
   4
   5 import unittest
   6 import base64
   7 from sfa.trust.certificate import Keypair
   8
   9 class TestKeypair(unittest.TestCase):
  10    def setUp(self):
  11       pass
  12
  13    def testCreate(self):
  14       k = Keypair()
  15       k.create()
  16
  17    def testSaveLoadFile(self):
  18       k = Keypair()
  19       k.create()
  20
  21       k.save_to_file("test.key")
  22
  23       k2 = Keypair()
  24       k2.load_from_file("test.key")
  25
  26       self.assertEqual(k.as_pem(), k2.as_pem())
  27
  28    def test_get_m2_pubkey(self):
  29       k = Keypair()
  30       k.create()
  31
  32       m2 = k.get_m2_pubkey()
  33       self.assertTrue(m2 != None)
  34
  35    def test_get_openssl_pkey(self):
  36       k = Keypair()
  37       k.create()
  38
  39       pk = k.get_openssl_pkey()
  40       self.assertTrue(pk != None)
  41
  42    def test_sign_verify(self):
  43       k = Keypair()
  44       k.create()
  45
  46       data = "this is a test"
  47       sig = k.sign_string(data)
  48
  49       print(k.verify_string(data, sig))
  50
  51 if __name__ == "__main__":
  52     unittest.main()