diff --git a/python/pyarrow/tests/parquet/test_encryption.py b/python/pyarrow/tests/parquet/test_encryption.py index 6d0cfc0024e1..5185bf6a3013 100644 --- a/python/pyarrow/tests/parquet/test_encryption.py +++ b/python/pyarrow/tests/parquet/test_encryption.py @@ -680,9 +680,12 @@ def check_rotated_external_keys(master_key_id: str) -> None: assert data_table.equals(table_read_after_rotation) -def recording_kms_factory(created_configs, client_class=InMemoryKmsClient): +def recording_kms_factory(created_configs, client_class=None): """Create a KMS client factory that appends the KMS instance ID and URL of each connection configuration it is given to created_configs""" + if client_class is None: + client_class = InMemoryKmsClient + def kms_factory(kms_connection_configuration): created_configs.append( (kms_connection_configuration.kms_instance_id,