From 90bd914ac145c3344f9fd503678e974fc8eb4e41 Mon Sep 17 00:00:00 2001 From: Nikolaos Korasidis Date: Tue, 16 Sep 2014 16:50:00 +0300 Subject: [PATCH] Test that encryption is static. * The curve used for encryption is derived from the key provided. --- test/test_ec.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/test/test_ec.py b/test/test_ec.py index f4db1e703..6841473c0 100644 --- a/test/test_ec.py +++ b/test/test_ec.py @@ -52,7 +52,9 @@ def setUpClass(cls): cls.alice = ec.ECC(curve=cls.ecc_curve) cls.bob = ec.ECC(curve=cls.ecc_curve) cls.bob_pubkey = cls.bob.get_pubkey() + cls.bob_privkey = cls.bob.get_privkey() cls.alice_pubkey = cls.alice.get_pubkey() + cls.data = "YELLOW SUBMARINE" def test_asymmetric_enc_dec(self): plaintext = "Hello Bob" @@ -78,6 +80,22 @@ def test_curve_mismatch(self): with self.assertRaises(Exception): agent2.get_ecdh_key(agent1.get_pubkey()) + def test_encrypt_is_static(self): + obj_agent1 = ec.ECC(curve=self.ecc_curve) + obj_agent2 = ec.ECC(curve='sect283k1') + cls_agent3 = ec.ECC + + encd1 = obj_agent1.encrypt(self.data, self.bob_pubkey) + encd2 = obj_agent2.encrypt(self.data, self.bob_pubkey) + encd3 = cls_agent3.encrypt(self.data, self.bob_pubkey) + + dcd1 = self.bob.decrypt(encd1) + dcd2 = self.bob.decrypt(encd2) + dcd3 = self.bob.decrypt(encd3) + + self.assertEqual(dcd1, dcd2) + self.assertEqual(dcd2, dcd3) + if __name__ == "__main__": unittest.main()