commit d79c5c5105f0f05d26709e469ab279e3a3023121
author Paul Elliott <paul.elliott@arm.com> Wed Oct 06 21:49:41 2021 +0100
committer Paul Elliott <paul.elliott@arm.com> Thu Oct 07 18:06:03 2021 +0100
tree 063ac7dc6b2657a6af08b755c31302e05125513c
parent e4c08ed257c9602345b17ec9c3632a9e4c8279c8

Add Multipart AEAD CCM generate nonce tests

Signed-off-by: Paul Elliott <paul.elliott@arm.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 000806d..3e1355d 100644
--- a/tests/suites/test_suite_psa_crypto.function
+++ b/tests/suites/test_suite_psa_crypto.function
@@ -4039,12 +4039,13 @@
         TEST_EQUAL( actual_nonce_length, PSA_AEAD_NONCE_LENGTH( key_type,
                                                                 alg ) );
 
-    TEST_ASSERT( actual_nonce_length < PSA_AEAD_NONCE_MAX_SIZE );
+    TEST_ASSERT( actual_nonce_length <= PSA_AEAD_NONCE_MAX_SIZE );
 
     if( expected_status == PSA_SUCCESS )
     {
-
         /* Ensure we can still complete operation. */
+        PSA_ASSERT( psa_aead_set_lengths( &operation, additional_data->len,
+                                          input_data->len ) );
 
         PSA_ASSERT( psa_aead_update_ad( &operation, additional_data->x,
                                         additional_data->len ) );