From 657ee187301a1b810716f524f486178dd8b5b70b Mon Sep 17 00:00:00 2001 From: Pol Henarejos Date: Mon, 13 Mar 2023 18:02:31 +0100 Subject: [PATCH] Added assert on deletion Signed-off-by: Pol Henarejos --- tests/pico-hsm/test_020_keypair_gen.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/tests/pico-hsm/test_020_keypair_gen.py b/tests/pico-hsm/test_020_keypair_gen.py index be0775a..06b46a9 100644 --- a/tests/pico-hsm/test_020_keypair_gen.py +++ b/tests/pico-hsm/test_020_keypair_gen.py @@ -30,8 +30,10 @@ def test_gen_ecc(device, curve): keyid = device.key_generation(KeyType.ECC, curve) resp = device.list_keys() assert((DOPrefixes.KEY_PREFIX.value, keyid) in resp) - device.delete_file(DOPrefixes.KEY_PREFIX.value << 8 | keyid) - device.delete_file(DOPrefixes.EE_CERTIFICATE_PREFIX.value << 8 | keyid) + device.delete_file(DOPrefixes.KEY_PREFIX.value, keyid) + device.delete_file(DOPrefixes.EE_CERTIFICATE_PREFIX.value, keyid) + resp = device.list_keys() + assert((DOPrefixes.KEY_PREFIX.value, keyid) not in resp) @pytest.mark.parametrize( "modulus", [1024, 2048, 4096] @@ -40,6 +42,6 @@ def test_gen_rsa(device, modulus): keyid = device.key_generation(KeyType.RSA, modulus) resp = device.list_keys() assert((DOPrefixes.KEY_PREFIX.value, keyid) in resp) - device.delete_file(DOPrefixes.KEY_PREFIX.value << 8 | keyid) - device.delete_file(DOPrefixes.EE_CERTIFICATE_PREFIX.value << 8 | keyid) + device.delete_file(DOPrefixes.KEY_PREFIX.value, keyid) + device.delete_file(DOPrefixes.EE_CERTIFICATE_PREFIX.value, keyid)