This is an automated email from the ASF dual-hosted git repository.
amoghdesai pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 493402d24ec Remove duplicate 'mock_supervisor_comms' pytest fixtures
from core (#49457)
493402d24ec is described below
commit 493402d24ece8c87b8004fa8a476b232408069cc
Author: Sneha Prabhu <[email protected]>
AuthorDate: Tue Apr 22 13:02:52 2025 +0530
Remove duplicate 'mock_supervisor_comms' pytest fixtures from core (#49457)
---
airflow-core/tests/conftest.py | 9 ---------
airflow-core/tests/unit/hooks/test_base.py | 8 --------
airflow-core/tests/unit/models/test_taskinstance.py | 8 --------
3 files changed, 25 deletions(-)
diff --git a/airflow-core/tests/conftest.py b/airflow-core/tests/conftest.py
index c605cc7648e..bffa1c61e89 100644
--- a/airflow-core/tests/conftest.py
+++ b/airflow-core/tests/conftest.py
@@ -24,7 +24,6 @@ import zipfile
from contextlib import contextmanager
from pathlib import Path
from typing import TYPE_CHECKING
-from unittest import mock
import pytest
@@ -126,14 +125,6 @@ def test_zip_path(tmp_path: Path):
return os.fspath(zipped)
[email protected]
-def mock_supervisor_comms():
- with mock.patch(
- "airflow.sdk.execution_time.task_runner.SUPERVISOR_COMMS", create=True
- ) as supervisor_comms:
- yield supervisor_comms
-
-
if TYPE_CHECKING:
# Static checkers do not know about pytest fixtures' types and return,
# In case if them distributed through third party packages.
diff --git a/airflow-core/tests/unit/hooks/test_base.py
b/airflow-core/tests/unit/hooks/test_base.py
index d1494ed308b..8c2d44f8485 100644
--- a/airflow-core/tests/unit/hooks/test_base.py
+++ b/airflow-core/tests/unit/hooks/test_base.py
@@ -29,14 +29,6 @@ from airflow.sdk.execution_time.comms import
ConnectionResult, ErrorResponse, Ge
from tests_common.test_utils.config import conf_vars
[email protected]
-def mock_supervisor_comms():
- with mock.patch(
- "airflow.sdk.execution_time.task_runner.SUPERVISOR_COMMS", create=True
- ) as supervisor_comms:
- yield supervisor_comms
-
-
class TestBaseHook:
def test_hook_has_default_logger_name(self):
hook = BaseHook()
diff --git a/airflow-core/tests/unit/models/test_taskinstance.py
b/airflow-core/tests/unit/models/test_taskinstance.py
index f6d61623a25..a398876f1be 100644
--- a/airflow-core/tests/unit/models/test_taskinstance.py
+++ b/airflow-core/tests/unit/models/test_taskinstance.py
@@ -127,14 +127,6 @@ def task_reschedules_for_ti():
return wrapper
[email protected]
-def mock_supervisor_comms():
- with mock.patch(
- "airflow.sdk.execution_time.task_runner.SUPERVISOR_COMMS", create=True
- ) as supervisor_comms:
- yield supervisor_comms
-
-
class CallbackWrapper:
task_id: str | None = None
dag_id: str | None = None
