Generate test wrappers
Signed-off-by: Thomas Daubney <thomas.daubney@arm.com>
diff --git a/tests/scripts/generate_psa_wrappers.py b/tests/scripts/generate_psa_wrappers.py
index e5b4256..005a324 100755
--- a/tests/scripts/generate_psa_wrappers.py
+++ b/tests/scripts/generate_psa_wrappers.py
@@ -145,6 +145,11 @@
# Proof-of-concept: just instrument one function for now
if function_name == 'psa_cipher_encrypt':
return True
+ if function_name in ('psa_sign_message',
+ 'psa_verify_message',
+ 'psa_sign_hash',
+ 'psa_verify_hash'):
+ return True
return False
def _write_function_call(self, out: typing_util.Writable,