commit 810612e14ec26021f56e065fcdab09600f93fb4e
author Raef Coles <raef.coles@arm.com> Tue Oct 11 13:16:53 2022 +0100
committer Raef Coles <raef.coles@arm.com> Thu Oct 13 14:29:51 2022 +0100
tree 6aa6943d8622116dae1f7c96aba6165b67c2b5e7
parent 90e13fc3c62132e4dd1bc3b36f6dcee4895c7ef6

Update LMS and LMOTS to use TEST_EQUAL

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 f69ca9e..7ba02c5 100644
--- a/tests/suites/test_suite_lms.function
+++ b/tests/suites/test_suite_lms.function
@@ -155,9 +155,9 @@
     uint8_t exported_pub_key[MBEDTLS_LMS_PUBLIC_KEY_LEN(MBEDTLS_LMS_SHA256_M32_H10)];
 
     mbedtls_lms_public_init(&ctx);
-    TEST_ASSERT( mbedtls_lms_import_public_key( &ctx, pub_key->x, pub_key->len ) == 0 );
-    TEST_ASSERT( mbedtls_lms_export_public_key( &ctx, exported_pub_key,
-                                                sizeof(exported_pub_key), NULL ) == 0 );
+    TEST_EQUAL( mbedtls_lms_import_public_key( &ctx, pub_key->x, pub_key->len ), 0 );
+    TEST_EQUAL( mbedtls_lms_export_public_key( &ctx, exported_pub_key,
+                                                sizeof(exported_pub_key), NULL ), 0 );
 
     ASSERT_COMPARE( pub_key->x, MBEDTLS_LMS_PUBLIC_KEY_LEN(MBEDTLS_LMS_SHA256_M32_H10),
                     exported_pub_key, MBEDTLS_LMS_PUBLIC_KEY_LEN(MBEDTLS_LMS_SHA256_M32_H10) );