asymmetric_encrypt: check output length only if return code is PSA_SU…

…CCESS. Signed-off-by: Stephan Koch <[email protected]> Signed-off-by: Dave Rodgman <[email protected]>
Mbed-TLS · Feb 28, 2023 · 8a23f49 · 8a23f49
1 parent 7a5168e
commit 8a23f49
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/suites/test_suite_psa_crypto.function b/tests/suites/test_suite_psa_crypto.function
@@ -4173,7 +4173,9 @@ void asymmetric_encrypt(int key_type_arg,
                                            output, output_size,
                                            &output_length);
     TEST_EQUAL(actual_status, expected_status);
-    TEST_EQUAL(output_length, expected_output_length);
+    if (actual_status == PSA_SUCCESS) {
+        TEST_EQUAL(output_length, expected_output_length);
+    }
 
     /* If the label is empty, the test framework puts a non-null pointer
      * in label->x. Test that a null pointer works as well. */
@@ -4188,7 +4190,9 @@ void asymmetric_encrypt(int key_type_arg,
                                                output, output_size,
                                                &output_length);
         TEST_EQUAL(actual_status, expected_status);
-        TEST_EQUAL(output_length, expected_output_length);
+        if (actual_status == PSA_SUCCESS) {
+            TEST_EQUAL(output_length, expected_output_length);
+        }
     }
 
 exit: