From 8ffe3e5451fa0a4494d4d15ee3ed58245b287b19 Mon Sep 17 00:00:00 2001 From: Rob Austein Date: Fri, 4 Nov 2016 16:41:11 -0400 Subject: .set_attribute() speed tests. --- unit-tests.py | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) (limited to 'unit-tests.py') diff --git a/unit-tests.py b/unit-tests.py index 5f472fd..3f74b36 100644 --- a/unit-tests.py +++ b/unit-tests.py @@ -614,6 +614,57 @@ class TestPKeyAttribute(TestCaseLoggedIn): self.load_and_fill(HAL_KEY_FLAG_TOKEN, n_attrs = 16, n_fill = 1024) +class TestPKeyAttributeSpeedToken(TestCaseLoggedIn): + """ + Attribute speed tests. + """ + + def setUp(self): + der = PreloadedKey.db[HAL_KEY_TYPE_EC_PRIVATE, HAL_CURVE_P256].der + self.k = hsm.pkey_load(HAL_KEY_TYPE_EC_PRIVATE, HAL_CURVE_P256, der, HAL_KEY_FLAG_TOKEN) + self.addCleanup(self.k.delete) + super(TestPKeyAttributeSpeedToken, self).setUp() + + def set_attributes(self, n_attrs): + pinwheel = Pinwheel() + for i in xrange(n_attrs): + pinwheel() + self.k.set_attribute(i, "Attribute {}".format(i)) + + def test_set_1_attribute(self): + self.set_attributes(1) + + def test_set_6_attributes(self): + self.set_attributes(6) + + def test_set_12_attributes(self): + self.set_attributes(12) + +class TestPKeyAttributeSpeedVolatile(TestCaseLoggedIn): + """ + Attribute speed tests. + """ + + def setUp(self): + der = PreloadedKey.db[HAL_KEY_TYPE_EC_PRIVATE, HAL_CURVE_P256].der + self.k = hsm.pkey_load(HAL_KEY_TYPE_EC_PRIVATE, HAL_CURVE_P256, der, 0) + self.addCleanup(self.k.delete) + super(TestPKeyAttributeSpeedVolatile, self).setUp() + + def set_attributes(self, n_attrs): + for i in xrange(n_attrs): + self.k.set_attribute(i, "Attribute {}".format(i)) + + def test_set_1_attribute(self): + self.set_attributes(1) + + def test_set_6_attributes(self): + self.set_attributes(6) + + def test_set_12_attributes(self): + self.set_attributes(12) + + @unittest.skipUnless(ecdsa_loaded, "Requires Python ECDSA package") class TestPkeyECDSAVerificationNIST(TestCaseLoggedIn): """ -- cgit v1.2.3