commit 0a75adcf4e943941d88deaff98951170348cc723
author Janos Follath <janos.follath@arm.com> Thu Aug 22 20:00:23 2024 +0100
committer Manuel Pégourié-Gonnard <manuel.pegourie-gonnard@arm.com> Mon Sep 02 10:30:47 2024 +0200
tree c413fbf52a157c8b1c08cd4284e9e765b557bdb6
parent e9cc10d2affa668a6847d4cffc3a53160f5239d9

Prepare codepath tests for early termination

Signed-off-by: Janos Follath <janos.follath@arm.com>

tests/suites/test_suite_rsa.function

diff --git a/tests/suites/test_suite_rsa.function b/tests/suites/test_suite_rsa.function
index 75f3f42..98ea9ef 100644
--- a/tests/suites/test_suite_rsa.function
+++ b/tests/suites/test_suite_rsa.function
@@ -496,7 +496,7 @@
 #endif
     TEST_ASSERT(mbedtls_rsa_public(&ctx, message_str->x, output) == result);
 #if defined(MBEDTLS_TEST_HOOKS) && !defined(MBEDTLS_THREADING_C)
-    TEST_EQUAL(mbedtls_codepath_check, MBEDTLS_MPI_IS_PUBLIC);
+    ASSERT_RSA_CODEPATH(MBEDTLS_MPI_IS_PUBLIC, result);
 #endif
     if (result == 0) {
 
@@ -569,7 +569,7 @@
                                         &rnd_info, message_str->x,
                                         output) == result);
 #if defined(MBEDTLS_TEST_HOOKS) && !defined(MBEDTLS_THREADING_C)
-        TEST_EQUAL(mbedtls_codepath_check, MBEDTLS_MPI_IS_SECRET);
+        ASSERT_RSA_CODEPATH(MBEDTLS_MPI_IS_SECRET, result);
 #endif
         if (result == 0) {