commit c8b6c050141fc24243bcd6c0a3c71b9cec44bb00
author Ryan Everett <144035422+Ryan-Everett-arm@users.noreply.github.com> Thu Jan 25 15:20:09 2024 +0000
committer Ryan Everett <ryan.everett@arm.com> Fri Feb 02 10:50:51 2024 +0000
tree e3ca145eca3b0cdf1ea54cd99c798cf477acf4a1
parent cb4f8554567078b346b64a06a018e6d314a64536

Update tests/scripts/generate_psa_wrappers.py

Co-authored-by: David Horstmann <david.horstmann@arm.com>
Signed-off-by: Ryan Everett <ryan.everett@arm.com>

tests/scripts/generate_psa_wrappers.py

diff --git a/tests/scripts/generate_psa_wrappers.py b/tests/scripts/generate_psa_wrappers.py
index 7fc852e..d7970c0 100755
--- a/tests/scripts/generate_psa_wrappers.py
+++ b/tests/scripts/generate_psa_wrappers.py
@@ -153,9 +153,9 @@
         # Proof-of-concept: just instrument one function for now
         if function_name == 'psa_cipher_encrypt':
             return True
-            if function_name in ('psa_import_key',
-                                 'psa_export_key',
-                                 'psa_export_public_key'):
+        if function_name in ('psa_import_key',
+                             'psa_export_key',
+                             'psa_export_public_key'):
             return True
         if function_name in ('psa_sign_message',
                              'psa_verify_message',