Fix Airflow 3 tests raising generate_run_id() takes 0 positional arguments

[tatiana]

Action Link: https://github.com/astronomer/astronomer-cosmos/actions/runs/14703034847/job/41256451294?pr=1646#step:7:4023

A few tests in the tests/operators/test_local.py failing with the below error

FAILED tests/operators/test_local.py::test_run_operator_dataset_url_encoded_names - TypeError: generate_run_id() takes 0 positional arguments but 2 were given
FAILED tests/operators/test_local.py::test_run_operator_caches_partial_parsing - TypeError: generate_run_id() takes 0 positional arguments but 2 were given
FAILED tests/operators/test_local.py::test_run_test_operator_with_callback[InvocationMode.SUBPROCESS] - TypeError: generate_run_id() takes 0 positional arguments but 2 were given
FAILED tests/operators/test_local.py::test_run_test_operator_with_callback[InvocationMode.DBT_RUNNER] - TypeError: generate_run_id() takes 0 positional arguments but 2 were given
FAILED tests/operators/test_local.py::test_run_test_operator_without_callback[InvocationMode.SUBPROCESS] - TypeError: generate_run_id() takes 0 positional arguments but 2 were given
FAILED tests/operators/test_local.py::test_run_test_operator_without_callback[InvocationMode.DBT_RUNNER] - TypeError: generate_run_id() takes 0 positional arguments but 2 were given

The only test this PR is not uncommenting is test_run_operator_dataset_url_encoded_names - that is being addressed in PR #1716

Closes: #1705

[netlify]

✅ Deploy Preview for sunny-pastelito-5ecb04 canceled.

Name	Link
🔨 Latest commit	b486dfc
🔍 Latest deploy log	https://app.netlify.com/sites/sunny-pastelito-5ecb04/deploys/6810c807a95b89000890dbbd

[cloudflare-workers-and-pages]

Deploying astronomer-cosmos with    Cloudflare Pages

Latest commit:	b486dfc
Status:	✅  Deploy successful!
Preview URL:	https://0c1c0858.astronomer-cosmos.pages.dev
Branch Preview URL:	https://fix-1705.astronomer-cosmos.pages.dev

View logs

[Copilot]

Pull Request Overview

This PR fixes the issue with tests failing in Airflow 3 by addressing the "generate_run_id() takes 0 positional arguments" error and updating test routines accordingly.

• Updated tests/utils.py to add a workaround for Airflow 2.10 versions by catching sqlalchemy.exc.PendingRollbackError.
• Removed redundant exception handling in tests/test_example_dags.py.
• Removed skip decorators in tests/operators/test_local.py to enable running tests on Airflow 3.

Reviewed Changes

Copilot reviewed 3 out of 3 changed files in this pull request and generated 1 comment.

File	Description
tests/utils.py	Introduces a new test_dag function with version-specific handling and exception fallback.
tests/test_example_dags.py	Eliminates redundant try/except block and warnings in favor of the updated test_dag approach.
tests/operators/test_local.py	Removes skipif decorators to support Airflow 3 compatibility in operator tests.

[pankajkoti]

LGTM. Thanks for consolidating the method that's getting used across tests to decide on Airflow version specific test function.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 97.58%. Comparing base (a4e0805) to head (b486dfc).
Report is 2 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #1725   +/-   ##
=======================================
  Coverage   97.58%   97.58%           
=======================================
  Files          83       83           
  Lines        5174     5174           
=======================================
  Hits         5049     5049           
  Misses        125      125           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.