diff --git a/tests/scripts/generate_psa_tests.py b/tests/scripts/generate_psa_tests.py index 492810bf0..d34663188 100755 --- a/tests/scripts/generate_psa_tests.py +++ b/tests/scripts/generate_psa_tests.py @@ -78,6 +78,11 @@ def automatic_dependencies(*expressions: str) -> List[str]: """ used = set() for expr in expressions: + # HKDF_EXTRACT and HKDF_EXPAND algs depend on HKDF + if "HKDF_EXTRACT" in expr: + expr = expr.replace("HKDF_EXTRACT", "HKDF") + if "HKDF_EXPAND" in expr: + expr = expr.replace("HKDF_EXPAND", "HKDF") used.update(re.findall(r'PSA_(?:ALG|ECC_FAMILY|KEY_TYPE)_\w+', expr)) used.difference_update(SYMBOLS_WITHOUT_DEPENDENCY) return sorted(psa_want_symbol(name) for name in used)