commit 9a855f21aa610b5f1f8aafce90d279fc5d3524ff
author Valerio Setti <valerio.setti@nordicsemi.no> Thu Apr 27 12:07:23 2023 +0200
committer Valerio Setti <valerio.setti@nordicsemi.no> Tue May 02 15:45:39 2023 +0200
tree 0f864ff91775c2eca1dedf8f14d467a46958ae64
parent f5451717af843d95f7ed4f7d2154dfa7b464332b

test: check for exact length of returned pub key

Signed-off-by: Valerio Setti <valerio.setti@nordicsemi.no>

tests/suites/test_suite_pkwrite.function

diff --git a/tests/suites/test_suite_pkwrite.function b/tests/suites/test_suite_pkwrite.function
index 7fdda80..c148c8a 100644
--- a/tests/suites/test_suite_pkwrite.function
+++ b/tests/suites/test_suite_pkwrite.function
@@ -149,8 +149,8 @@
     derived_key_len = pub_key_len;
     ASSERT_ALLOC(derived_key_raw, derived_key_len);
 
-    TEST_LE_U(1, mbedtls_pk_write_pubkey_der(&priv_key, derived_key_raw,
-                                             derived_key_len));
+    TEST_EQUAL(mbedtls_pk_write_pubkey_der(&priv_key, derived_key_raw,
+                                           derived_key_len), pub_key_len);
 
     ASSERT_COMPARE(derived_key_raw, derived_key_len,
                    pub_key_raw, pub_key_len);
@@ -162,8 +162,8 @@
                                          PSA_ALG_NONE, PSA_KEY_USAGE_EXPORT,
                                          PSA_ALG_NONE), 0);
 
-    TEST_LE_U(1, mbedtls_pk_write_pubkey_der(&priv_key, derived_key_raw,
-                                             derived_key_len));
+    TEST_EQUAL(mbedtls_pk_write_pubkey_der(&priv_key, derived_key_raw,
+                                           derived_key_len), pub_key_len);
 
     ASSERT_COMPARE(derived_key_raw, derived_key_len,
                    pub_key_raw, pub_key_len);