diff options
| -rw-r--r-- | unit-tests.py | 51 |
1 files changed, 51 insertions, 0 deletions
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): """ |