mirror of https://github.com/nucypher/nucypher.git
Add test test_keypair_object to test keypair intialization
parent
506fd8c2d0
commit
d89fd7cf31
|
@ -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))
|
||||
|
|
Loading…
Reference in New Issue