commit 81797fc27052223bebd6336fb6ac5d0d5716f875
author Kusumit Ghoderao <Kusumit.Ghoderao@silabs.com> Mon Jun 05 15:05:09 2023 +0530
committer Kusumit Ghoderao <Kusumit.Ghoderao@silabs.com> Tue Jun 06 15:05:42 2023 +0530
tree a53a1009d426b11273ee366d5edc973c609d3c8a
parent f6a0d57e4d08806723d63fd108805f6e34bcdf7d

Add cost and password steps to derive_output_test function

Signed-off-by: Kusumit Ghoderao <Kusumit.Ghoderao@silabs.com>

tests/suites/test_suite_psa_crypto.function

diff --git a/tests/suites/test_suite_psa_crypto.function b/tests/suites/test_suite_psa_crypto.function
index b0123d9..cdd7a81 100644
--- a/tests/suites/test_suite_psa_crypto.function
+++ b/tests/suites/test_suite_psa_crypto.function
@@ -8713,6 +8713,20 @@
         switch (steps[i]) {
             case 0:
                 break;
+            case PSA_KEY_DERIVATION_INPUT_COST:
+                TEST_EQUAL(psa_key_derivation_input_integer(
+                                &operation, steps[i],
+                                parse_binary_string(inputs[i])),
+                            statuses[i]);
+                if (statuses[i] != PSA_SUCCESS) {
+                    goto exit;
+                }
+                break;
+            case PSA_KEY_DERIVATION_INPUT_PASSWORD:
+                if (!PSA_ALG_IS_PBKDF2_HMAC(alg)) {
+                    goto exit;
+                }
+                // fall through
             case PSA_KEY_DERIVATION_INPUT_SECRET:
                 switch (key_input_type) {
                     case 0: // input bytes
@@ -8740,9 +8754,14 @@
                                       PSA_TLS12_PSK_TO_MS_PSK_MAX_SIZE);
                         }
 
-                        PSA_ASSERT(psa_key_derivation_input_key(&operation,
+                        TEST_EQUAL(psa_key_derivation_input_key(&operation,
                                                                 steps[i],
-                                                                keys[i]));
+                                                                keys[i]),
+                                   statuses[i]);
+
+                        if (statuses[i] != PSA_SUCCESS) {
+                            goto exit;
+                        }
                         break;
                     default:
                         TEST_ASSERT(!"default case not supported");