From 027796c0cc2619a770d3024422fd53b5a9e271c0 Mon Sep 17 00:00:00 2001 From: Valerio Setti Date: Wed, 20 Mar 2024 16:55:35 +0100 Subject: [PATCH] test_suite_pk: uniformly generate RSA and EC keys in pk_psa_sign() Signed-off-by: Valerio Setti --- tests/suites/test_suite_pk.function | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/suites/test_suite_pk.function b/tests/suites/test_suite_pk.function index 702c23e5ca..272e9f67a9 100644 --- a/tests/suites/test_suite_pk.function +++ b/tests/suites/test_suite_pk.function @@ -1862,8 +1862,7 @@ void pk_psa_sign(int psa_type, int bits, int rsa_padding) if (PSA_KEY_TYPE_IS_RSA(psa_type)) { TEST_ASSERT(mbedtls_pk_setup(&pk, mbedtls_pk_info_from_type(MBEDTLS_PK_RSA)) == 0); - TEST_ASSERT(mbedtls_rsa_gen_key(mbedtls_pk_rsa(pk), mbedtls_test_rnd_std_rand, NULL, - bits, 3) == 0); + TEST_EQUAL(pk_genkey(&pk, bits), 0); TEST_EQUAL(mbedtls_rsa_set_padding(mbedtls_pk_rsa(pk), rsa_padding, MBEDTLS_MD_NONE), 0); } #else /* MBEDTLS_RSA_C && MBEDTLS_GENPRIME */