Add usage parameter to mbedtls_pk_can_do_ext()

Signed-off-by: Neil Armstrong <narmstrong@baylibre.com>
diff --git a/tests/suites/test_suite_pk.function b/tests/suites/test_suite_pk.function
index c7b514e..468f22a 100644
--- a/tests/suites/test_suite_pk.function
+++ b/tests/suites/test_suite_pk.function
@@ -246,7 +246,7 @@
 
 /* BEGIN_CASE depends_on:MBEDTLS_USE_PSA_CRYPTO */
 void pk_can_do_ext( int key_type, int key_usage, int key_alg, int key_alg2,
-                    int key_bits, int alg_check, int result )
+                    int key_bits, int alg_check, int usage_check, int result )
 {
     mbedtls_pk_context pk;
     mbedtls_svc_key_id_t key = MBEDTLS_SVC_KEY_ID_INIT;
@@ -272,7 +272,7 @@
 
     TEST_EQUAL( mbedtls_pk_get_type( &pk ), MBEDTLS_PK_OPAQUE );
 
-    TEST_EQUAL( mbedtls_pk_can_do_ext( &pk, alg_check ), result );
+    TEST_EQUAL( mbedtls_pk_can_do_ext( &pk, alg_check, usage_check ), result );
 
 exit:
     psa_reset_key_attributes( &attributes );