commit edb20865c7a9cf9c608d0bf32ccc38461a37d267
author Neil Armstrong <narmstrong@baylibre.com> Mon Feb 07 15:47:44 2022 +0100
committer Neil Armstrong <narmstrong@baylibre.com> Tue Feb 22 10:56:18 2022 +0100
tree dcc6410e5e27cc4607bd84bee352e51fd82ed1b3
parent 9b545c04f7ee41692c11ac7454b25a3f975d70ba

Add One-Shot Hash setup test

Signed-off-by: Neil Armstrong <narmstrong@baylibre.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 c94c37e..9874d58 100644
--- a/tests/suites/test_suite_psa_crypto.function
+++ b/tests/suites/test_suite_psa_crypto.function
@@ -1803,12 +1803,24 @@
                  int expected_status_arg )
 {
     psa_algorithm_t alg = alg_arg;
+    uint8_t *output = NULL;
+    size_t output_size = 0;
+    size_t output_length = 0;
     psa_status_t expected_status = expected_status_arg;
     psa_hash_operation_t operation = PSA_HASH_OPERATION_INIT;
     psa_status_t status;
 
     PSA_ASSERT( psa_crypto_init( ) );
 
+    /* Hash Setup, one-shot */
+    output_size = PSA_HASH_LENGTH(alg);
+    ASSERT_ALLOC( output, output_size );
+
+    status = psa_hash_compute( alg, NULL, 0,
+                               output, output_size, &output_length );
+    TEST_EQUAL( status, expected_status );
+
+    /* Hash Setup, multi-part */
     status = psa_hash_setup( &operation, alg );
     TEST_EQUAL( status, expected_status );
 
@@ -1827,6 +1839,7 @@
 #endif
 
 exit:
+    mbedtls_free( output );
     PSA_DONE( );
 }
 /* END_CASE */