SNOW-3121122: Add SQLA2.1 to CI

Author: Dev-iL

.github/workflows/build_test.yml

@@ -56,6 +56,8 @@ jobs:
             python-version: "3.13"
           - hatch-env: sa14
             python-version: "3.12"
+          - hatch-env: sa21
+            python-version: "3.12"
     steps:
       - uses: actions/checkout@v4
         with:
@@ -73,8 +75,8 @@ jobs:
           python -m hatch -e ${{ matrix.hatch-env }} build --clean
       - name: Install and check import
         run: |
-          python -m uv pip install dist/snowflake_sqlalchemy-*.whl
-          python -c "import snowflake.sqlalchemy; print(snowflake.sqlalchemy.__version__)"
+          hatch run ${{ matrix.hatch-env }}:pip install dist/snowflake_sqlalchemy-*.whl --force-reinstall --no-deps
+          hatch run ${{ matrix.hatch-env }}:check-import
 
   test-dialect:
     name: Test dialect ${{ matrix.os }}-${{ fromJson('{"default":"3.13","sa14":"3.12"}')[matrix.hatch-env] }}-${{ matrix.cloud-provider }}-${{ matrix.hatch-env }}
@@ -261,6 +263,97 @@ jobs:
           path: |
             ./coverage.xml
 
+  test-dialect-v21:
+    name: Test dialect sa21 ${{ matrix.os }}-${{ matrix.python-version }}-${{ matrix.cloud-provider }}
+    needs: [lint, build-install]
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest, macos-15, windows-latest]
+        python-version: ["3.12"]
+        cloud-provider: [aws, azure, gcp]
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          persist-credentials: false
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Setup parameters file
+        shell: bash
+        env:
+          PARAMETERS_SECRET: ${{ secrets.PARAMETERS_SECRET }}
+        run: |
+          printf "%s" "$PARAMETERS_SECRET" | gpg --quiet --batch --yes \
+            --decrypt \
+            --passphrase-fd 0 \
+            --pinentry-mode loopback \
+            --output tests/parameters.py \
+            .github/workflows/parameters/parameters_${{ matrix.cloud-provider }}.py.gpg
+      - name: Upgrade pip and install hatch
+        run: |
+          python -m pip install -U uv
+          python -m uv pip install -U hatch
+          python -m hatch env create sa21
+      - name: Verify SQLAlchemy version
+        run: hatch run sa21:python -c "import sqlalchemy; print(sqlalchemy.__version__); assert sqlalchemy.__version__.startswith('2.1'), f'Expected SQLAlchemy 2.1.x, got {{sqlalchemy.__version__}}'"
+      - name: Run test for AWS
+        run: hatch run sa21:test-dialect-aws
+        if: matrix.cloud-provider == 'aws'
+      - name: Run tests
+        run: hatch run sa21:test-dialect
+      - uses: actions/upload-artifact@v4
+        with:
+          name: coverage.xml_dialect-sa21-${{ matrix.os }}-${{ matrix.python-version }}-${{ matrix.cloud-provider }}
+          path: |
+            ./coverage.xml
+
+  test-dialect-compatibility-v21:
+    name: Test dialect sa21 compatibility ${{ matrix.os }}-${{ matrix.python-version }}-${{ matrix.cloud-provider }}
+    needs: lint
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest, macos-15, windows-latest]
+        python-version: ["3.12"]
+        cloud-provider: [aws, azure, gcp]
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          persist-credentials: false
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Upgrade pip and install hatch
+        run: |
+          python -m pip install -U uv
+          python -m uv pip install -U hatch
+          python -m hatch env create sa21
+      - name: Verify SQLAlchemy version
+        run: hatch run sa21:python -c "import sqlalchemy; print(sqlalchemy.__version__); assert sqlalchemy.__version__.startswith('2.1'), f'Expected SQLAlchemy 2.1.x, got {{sqlalchemy.__version__}}'"
+      - name: Setup parameters file
+        shell: bash
+        env:
+          PARAMETERS_SECRET: ${{ secrets.PARAMETERS_SECRET }}
+        run: |
+          printf "%s" "$PARAMETERS_SECRET" | gpg --quiet --batch --yes \
+            --decrypt \
+            --passphrase-fd 0 \
+            --pinentry-mode loopback \
+            --output tests/parameters.py \
+            .github/workflows/parameters/parameters_${{ matrix.cloud-provider }}.py.gpg
+      - name: Run tests
+        run: hatch run sa21:test-dialect-compatibility
+      - uses: actions/upload-artifact@v4
+        with:
+          name: coverage.xml_dialect-sa21-compatibility-${{ matrix.os }}-${{ matrix.python-version }}-${{ matrix.cloud-provider }}
+          path: |
+            ./coverage.xml
+
   combine-coverage:
     name: Combine coverage
     if: ${{ success() || failure() }}
@@ -270,6 +363,8 @@ jobs:
         test-dialect-compatibility,
         test-dialect-v14,
         test-dialect-compatibility-v14,
+        test-dialect-v21,
+        test-dialect-compatibility-v21,
       ]
     runs-on: ubuntu-latest
     steps:

pyproject.toml

@@ -89,8 +89,14 @@ extra-dependencies = [
   "pandas>=2.1.1,<2.2",
   "numpy<2",
 ]
+
+[tool.hatch.envs.sa21]
+installer = "uv"
+builder = true
+extra-dependencies = ["SQLAlchemy>=2.1.0b1"]
 features = ["development", "pandas"]
 python = "3.12"
+pip-pre = true
 
 [tool.hatch.envs.sa14.scripts]
 test-dialect = "pytest --ignore_v20_test -ra -vvv --tb=short --cov snowflake.sqlalchemy --cov-append --junitxml ./junit.xml --ignore=tests/sqlalchemy_test_suite tests/"