commit fe2bda32572ed90813f6c2c1181593ac82a684e5
author Thomas Daubney <thomas.daubney@arm.com> Thu Feb 15 13:35:06 2024 +0000
committer Thomas Daubney <thomas.daubney@arm.com> Thu Feb 15 13:35:06 2024 +0000
tree 168361fd5dbf9bfb2d2858398dddec441561155f
parent 9739ac047abd97b4a18b7cebc232b15c406cfed4

Generate test wrappers

Signed-off-by: Thomas Daubney <thomas.daubney@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 c65603b..b56bd34 100755
--- a/tests/scripts/generate_psa_wrappers.py
+++ b/tests/scripts/generate_psa_wrappers.py
@@ -163,6 +163,9 @@
                              'psa_hash_compute',
                              'psa_hash_compare'):
             return True
+        if function_name in ('psa_key_derivation_key_agreement',
+                             'psa_raw_key_agreement'):
+            return True
         return False
 
     def _write_function_call(self, out: typing_util.Writable,