From e6912d8b44160aaf0ca6834f05a734f526518296 Mon Sep 17 00:00:00 2001
From: Dev-iL <gid.cto+github@gmail.com>
Date: Tue, 1 Jul 2025 18:16:14 +0300
Subject: [PATCH] CI: enable the SQLA2 special test for airflow-core

---
 .github/workflows/run-unit-tests.yml |  6 ++++++
 .github/workflows/special-tests.yml  | 24 ++++++++++++++++++++++++
 2 files changed, 30 insertions(+)

diff --git a/.github/workflows/run-unit-tests.yml b/.github/workflows/run-unit-tests.yml
index 035248113dac1..0355256276861 100644
--- a/.github/workflows/run-unit-tests.yml
+++ b/.github/workflows/run-unit-tests.yml
@@ -92,6 +92,11 @@ on:  # yamllint disable-line rule:truthy
         required: false
         default: "false"
         type: string
+      upgrade-sqlalchemy:
+        description: "Whether to upgrade SQLAlchemy or not (true/false)"
+        required: false
+        default: "false"
+        type: string
       upgrade-boto:
         description: "Whether to upgrade boto or not (true/false)"
         required: false
@@ -159,6 +164,7 @@ jobs:
       PARALLEL_TEST_TYPES: ${{ matrix.test-types.test_types }}
       PYTHON_MAJOR_MINOR_VERSION: "${{ matrix.python-version }}"
       UPGRADE_BOTO: "${{ inputs.upgrade-boto }}"
+      UPGRADE_SQLALCHEMY: "${{ inputs.upgrade-sqlalchemy }}"
       AIRFLOW_MONITOR_DELAY_TIME_IN_SECONDS: "${{inputs.monitor-delay-time-in-seconds}}"
       VERBOSE: "true"
     if: inputs.test-group == 'core' || inputs.skip-providers-tests != 'true'
diff --git a/.github/workflows/special-tests.yml b/.github/workflows/special-tests.yml
index a41ab1693c830..1b5bdb1ef1534 100644
--- a/.github/workflows/special-tests.yml
+++ b/.github/workflows/special-tests.yml
@@ -135,6 +135,30 @@ jobs:
       skip-providers-tests: ${{ inputs.skip-providers-tests }}
       use-uv: ${{ inputs.use-uv }}
 
+  tests-latest-sqlalchemy:
+    name: "Latest SQLAlchemy test: core"
+    uses: ./.github/workflows/run-unit-tests.yml
+    permissions:
+      contents: read
+      packages: read
+    with:
+      runners: ${{ inputs.runners }}
+      platform: ${{ inputs.platform }}
+      upgrade-sqlalchemy: "true"
+      test-name: "LatestSQLAlchemy-Postgres"
+      test-scope: "DB"
+      test-group: "core"
+      backend: "postgres"
+      python-versions: "['${{ inputs.default-python-version }}']"
+      backend-versions: "['${{ inputs.default-postgres-version }}']"
+      excluded-providers-as-string: ${{ inputs.excluded-providers-as-string }}
+      excludes: "[]"
+      test-types-as-strings-in-json: ${{ inputs.core-test-types-list-as-strings-in-json }}
+      run-coverage: ${{ inputs.run-coverage }}
+      debug-resources: ${{ inputs.debug-resources }}
+      skip-providers-tests: true  # TODO: After merging FAB5 set to: ${{ inputs.skip-providers-tests }}
+      use-uv: ${{ inputs.use-uv }}
+
   tests-boto-core:
     name: "Latest Boto test: core"
     uses: ./.github/workflows/run-unit-tests.yml