def test_hiding_config(self, sensitive_variable_fields, key, expected_result): from airflow.utils.log.secrets_masker import get_sensitive_variables_fields with conf_vars({ ('core', 'sensitive_var_conn_names'): str(sensitive_variable_fields) }): get_sensitive_variables_fields.cache_clear() assert expected_result == should_hide_value_for_key(key) get_sensitive_variables_fields.cache_clear()
def should_hide_value_for_key(key_name): import warnings from airflow.utils.log.secrets_masker import should_hide_value_for_key warnings.warn( "This function is deprecated. Please use " "`airflow.utils.log.secrets_masker.should_hide_value_for_key`", DeprecationWarning, stacklevel=2, ) return should_hide_value_for_key(key_name)
def test_hiding_defaults(self, key, expected_result): assert expected_result == should_hide_value_for_key(key)