Apply fixes to test driver from lib implementation

Signed-off-by: Paul Elliott <paul.elliott@arm.com>
diff --git a/tests/src/drivers/test_driver_aead.c b/tests/src/drivers/test_driver_aead.c
index ac116ff..84e69e0 100644
--- a/tests/src/drivers/test_driver_aead.c
+++ b/tests/src/drivers/test_driver_aead.c
@@ -282,7 +282,7 @@
                                    plaintext_size,
                                    plaintext_length,
                                    check_tag,
-                                   tag_length,
+                                   sizeof( check_tag ),
                                    &check_tag_length );
 
        if( mbedtls_test_driver_aead_hooks.driver_status == PSA_SUCCESS )
@@ -293,6 +293,8 @@
              mbedtls_test_driver_aead_hooks.driver_status =
                                                     PSA_ERROR_INVALID_SIGNATURE;
        }
+
+       mbedtls_platform_zeroize( check_tag, sizeof( check_tag ) );
     }
 
     return( mbedtls_test_driver_aead_hooks.driver_status );