From 8735fede00c27f85c1af767d75df1c2734ce21fb Mon Sep 17 00:00:00 2001 From: Amogh Desai Date: Tue, 4 Feb 2025 17:34:06 +0530 Subject: [PATCH] missed a file --- tests_common/pytest_plugin.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/tests_common/pytest_plugin.py b/tests_common/pytest_plugin.py index 0aaac8924e3722..57f379ec10d50d 100644 --- a/tests_common/pytest_plugin.py +++ b/tests_common/pytest_plugin.py @@ -32,6 +32,8 @@ import pytest import time_machine +from tests_common.test_utils.version_compat import AIRFLOW_V_3_0_PLUS + if TYPE_CHECKING: from itsdangerous import URLSafeSerializer from sqlalchemy.orm import Session @@ -1549,7 +1551,12 @@ def _disable_redact(request: pytest.FixtureRequest, mocker): yield return - mocked_redact = mocker.patch("airflow.sdk.execution_time.secrets_masker.SecretsMasker.redact") + target = ( + "airflow.sdk.execution_time.secrets_masker.redact" + if AIRFLOW_V_3_0_PLUS + else "airflow.utils.log.secrets_masker.mask_secret.redact" + ) + mocked_redact = mocker.patch(target) mocked_redact.side_effect = lambda item, name=None, max_depth=None: item with pytest.MonkeyPatch.context() as mp_ctx: mp_ctx.setattr(settings, "MASK_SECRETS_IN_LOGS", False)