commit d137c8612572546995f47809aae93ab47f7660a8
author Raef Coles <raef.coles@arm.com> Wed Oct 12 15:55:25 2022 +0100
committer Raef Coles <raef.coles@arm.com> Thu Oct 13 14:30:17 2022 +0100
tree ff127f7093a609aa9973fb8f2b2b5b7a64b103f5
parent 76563399fd10c835b40ff5f8cc770e0c629752b9

Don't skip LMS tests due to out of memory error

Signed-off-by: Raef Coles <raef.coles@arm.com>

tests/suites/test_suite_lms.function

diff --git a/tests/suites/test_suite_lms.function b/tests/suites/test_suite_lms.function
index 4f49e8e..c0cf02c 100644
--- a/tests/suites/test_suite_lms.function
+++ b/tests/suites/test_suite_lms.function
@@ -14,7 +14,6 @@
     mbedtls_lms_public_t pub_ctx;
     mbedtls_lms_private_t priv_ctx;
     unsigned char sig[MBEDTLS_LMS_SIG_LEN(MBEDTLS_LMS_SHA256_M32_H10, MBEDTLS_LMOTS_SHA256_N32_W8)];
-    int rc;
 
     mbedtls_lms_public_init( &pub_ctx );
     mbedtls_lms_private_init( &priv_ctx );
@@ -22,12 +21,10 @@
     /* Allocation failure isn't a test failure, since it likely just means
      * there's not enough memory to run the test.
      */
-    rc = mbedtls_lms_generate_private_key( &priv_ctx, MBEDTLS_LMS_SHA256_M32_H10,
+    TEST_EQUAL( mbedtls_lms_generate_private_key( &priv_ctx, MBEDTLS_LMS_SHA256_M32_H10,
                                            MBEDTLS_LMOTS_SHA256_N32_W8,
                                            mbedtls_test_rnd_std_rand, NULL,
-                                           seed->x, seed->len );
-    TEST_ASSUME( rc != MBEDTLS_ERR_LMS_ALLOC_FAILED );
-    TEST_EQUAL( rc, 0 );
+                                           seed->x, seed->len ), 0 );
 
     TEST_EQUAL( mbedtls_lms_calculate_public_key( &pub_ctx, &priv_ctx ), 0 );
 
@@ -50,7 +47,6 @@
     mbedtls_lms_public_t pub_ctx;
     mbedtls_lms_private_t priv_ctx;
     unsigned char sig[MBEDTLS_LMS_SIG_LEN(MBEDTLS_LMS_SHA256_M32_H10, MBEDTLS_LMOTS_SHA256_N32_W8)];
-    int rc;
 
     mbedtls_lms_public_init( &pub_ctx );
     mbedtls_lms_private_init( &priv_ctx );
@@ -58,12 +54,10 @@
     /* Allocation failure isn't a test failure, since it likely just means
      * there's not enough memory to run the test.
      */
-    rc = mbedtls_lms_generate_private_key( &priv_ctx, MBEDTLS_LMS_SHA256_M32_H10,
+    TEST_EQUAL( mbedtls_lms_generate_private_key( &priv_ctx, MBEDTLS_LMS_SHA256_M32_H10,
                                            MBEDTLS_LMOTS_SHA256_N32_W8,
                                            mbedtls_test_rnd_std_rand, NULL,
-                                           seed->x, seed->len );
-    TEST_ASSUME( rc != MBEDTLS_ERR_LMS_ALLOC_FAILED );
-    TEST_EQUAL( rc, 0 );
+                                           seed->x, seed->len ), 0 );
 
     TEST_EQUAL( mbedtls_lms_calculate_public_key( &pub_ctx, &priv_ctx ), 0 );