diff --git a/tests/models/test_mappedoperator.py b/tests/models/test_mappedoperator.py index dd335f9ef89d3..473a2224830f7 100644 --- a/tests/models/test_mappedoperator.py +++ b/tests/models/test_mappedoperator.py @@ -1786,6 +1786,7 @@ def group(n: int) -> None: assert states == expected +@pytest.mark.skip_if_database_isolation_mode # Does not work in db isolation mode def test_mapped_tasks_in_mapped_task_group_waits_for_upstreams_to_complete(dag_maker, session): """Test that one failed trigger rule works well in mapped task group""" with dag_maker() as dag: @@ -1817,6 +1818,7 @@ def t3(a): assert not ti3.state +@pytest.mark.skip_if_database_isolation_mode # Does not work in db isolation mode def test_mapped_tasks_in_mapped_task_group_waits_for_upstreams_to_complete__mapped_skip_with_all_success( dag_maker, session ): diff --git a/tests/models/test_skipmixin.py b/tests/models/test_skipmixin.py index 72ad956ae4c55..87001d0236ffd 100644 --- a/tests/models/test_skipmixin.py +++ b/tests/models/test_skipmixin.py @@ -116,6 +116,7 @@ def test_skip_none_tasks(self): assert not session.query.called assert not session.commit.called + @pytest.mark.skip_if_database_isolation_mode # Does not work in db isolation mode def test_skip_mapped_task(self, mock_session): SkipMixin()._skip( dag_run=MagicMock(spec=DagRun), @@ -128,6 +129,7 @@ def test_skip_mapped_task(self, mock_session): mock_session.execute.assert_not_called() mock_session.commit.assert_not_called() + @pytest.mark.skip_if_database_isolation_mode # Does not work in db isolation mode @patch("airflow.models.skipmixin.update") def test_skip_none_mapped_task(self, mock_update, mock_session): SkipMixin()._skip( diff --git a/tests/models/test_taskinstance.py b/tests/models/test_taskinstance.py index b5dcb43de262d..15a9eb1ba3307 100644 --- a/tests/models/test_taskinstance.py +++ b/tests/models/test_taskinstance.py @@ -3768,6 +3768,7 @@ def test_echo_env_variables(self, dag_maker): ti.refresh_from_db() assert ti.state == State.SUCCESS + @pytest.mark.skip_if_database_isolation_mode # Does not work in db isolation mode, fails in context serialization @pytest.mark.parametrize( "code, expected_state", [