commit 2e7c33d5309e45153777443000bb9a2c84b656ab
author Przemek Stekiel <przemyslaw.stekiel@mobica.com> Thu Apr 27 12:29:45 2023 +0200
committer Przemek Stekiel <przemyslaw.stekiel@mobica.com> Thu Apr 27 12:31:42 2023 +0200
tree c456c09aa6e9fd5666ae13d67fee599e3931ca06
parent d1cf1bae5dbbd25912aa5f6a6e2079cef83d6ef3

Use import_with_data for testing FFDH invalid key length

Signed-off-by: Przemek Stekiel <przemyslaw.stekiel@mobica.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 61ac74e..8e25ad2 100644
--- a/tests/suites/test_suite_psa_crypto.function
+++ b/tests/suites/test_suite_psa_crypto.function
@@ -1503,7 +1503,6 @@
     size_t reexported_length;
     psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT;
     psa_key_attributes_t got_attributes = PSA_KEY_ATTRIBUTES_INIT;
-    psa_status_t expected_import_result = PSA_SUCCESS;
 
     export_size = (ptrdiff_t) data->len + export_size_delta;
     ASSERT_ALLOC(exported, export_size);
@@ -1523,19 +1522,9 @@
         export_size -= 1;
     }
 
-    if (PSA_KEY_TYPE_IS_DH(type) &&
-        (data->len != 256 && data->len != 384 &&
-         data->len != 512 && data->len != 768 && data->len != 1024)) {
-        expected_import_result =  PSA_ERROR_INVALID_ARGUMENT;
-    }
-
     /* Import the key */
     TEST_EQUAL(psa_import_key(&attributes, data->x, data->len, &key),
-               expected_import_result);
-
-    if (expected_import_result != PSA_SUCCESS) {
-        goto exit;
-    }
+               PSA_SUCCESS);
 
     /* Test the key information */
     PSA_ASSERT(psa_get_key_attributes(key, &got_attributes));