Add test test_keypair_object to test keypair intialization

pull/80/head
tuxxy 2017-10-17 14:13:12 -06:00
parent 506fd8c2d0
commit d89fd7cf31
No known key found for this signature in database
GPG Key ID: 7BB971A0E89144D9
1 changed files with 39 additions and 0 deletions

View File

@ -28,3 +28,42 @@ class TestKeypairs(unittest.TestCase):
self.assertTrue(self.ecdsa_keypair.pubkey is not None)
self.assertEqual(bytes, type(self.ecdsa_keypair.pubkey))
self.assertEqual(64, len(self.ecdsa_keypair.pubkey))
def test_keypair_object(self):
# Test both keys
keypair = keypairs.SigningKeypair(self.ecdsa_keypair.privkey,
self.ecdsa_keypair.pubkey)
self.assertTrue(keypair.public_only is False)
self.assertEqual(bytes, type(keypair.privkey))
self.assertEqual(32, len(keypair.privkey))
self.assertEqual(bytes, type(keypair.pubkey))
self.assertEqual(64, len(keypair.pubkey))
# Test no keys (key generation)
keypair = keypairs.SigningKeypair()
self.assertTrue(keypair.public_only is False)
self.assertEqual(bytes, type(keypair.privkey))
self.assertEqual(32, len(keypair.privkey))
self.assertEqual(bytes, type(keypair.pubkey))
self.assertEqual(64, len(keypair.pubkey))
# Test privkey only
keypair = keypairs.SigningKeypair(privkey=self.ecdsa_keypair.privkey)
self.assertTrue(keypair.public_only is False)
self.assertEqual(bytes, type(keypair.privkey))
self.assertEqual(32, len(keypair.privkey))
self.assertEqual(bytes, type(keypair.pubkey))
self.assertEqual(64, len(keypair.pubkey))
# Test pubkey only
keypair = keypairs.SigningKeypair(pubkey=self.ecdsa_keypair.pubkey)
self.assertTrue(keypair.public_only is True)
self.assertEqual(bytes, type(keypair.pubkey))
self.assertEqual(64, len(keypair.pubkey))