SNOW-UD: Use shared install script for UD connector swap

Move the Universal Driver connector swap logic into the shared
tox_install_cmd.sh (conditional on ud_connector_path) so the UD
workflow runs the same tox environments as normal CI instead of
dedicated ud/ud-datasource envs.

- Add UD swap block to scripts/tox_install_cmd.sh (no-op when unset)
- Add ud_connector_path to [testenv] passenv
- Remove [testenv:ud] and [testenv:ud-datasource] from tox.ini
- Delete scripts/ud_tox_install_cmd.sh
- Workflow now invokes py${VER}-notdoctest-ci and datasource directly

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: sfc-gh-fpawlowski

.github/workflows/ud-inline-tests.yml

@@ -83,7 +83,7 @@ jobs:
           sudo ACCEPT_EULA=Y apt-get install -y msodbcsql18 unixodbc-dev
 
       - name: Run integration tests
-        run: python -m tox -e ud
+        run: python -m tox -e "py${PYTHON_VERSION//.}-notdoctest-ci"
         env:
           cloud_provider: ${{ env.CLOUD_PROVIDER }}
           ud_connector_path: "git+https://github.com/snowflakedb/universal-driver@${{ env.UD_BRANCH }}#subdirectory=python"
@@ -92,7 +92,7 @@ jobs:
           TOX_PARALLEL_NO_SPINNER: 1
 
       - name: Run datasource tests
-        run: python -m tox -e ud-datasource
+        run: python -m tox -e datasource
         env:
           cloud_provider: ${{ env.CLOUD_PROVIDER }}
           ud_connector_path: "git+https://github.com/snowflakedb/universal-driver@${{ env.UD_BRANCH }}#subdirectory=python"

scripts/tox_install_cmd.sh

@@ -2,6 +2,11 @@
 
 set -e
 
+# ── Standard install ─────────────────────────────────────────────────
+# Install project dependencies using uv.  This section is shared by
+# every tox environment and must remain provider-agnostic.
+# ─────────────────────────────────────────────────────────────────────
+
 # Check if uv is installed, and install it if not
 if ! command -v uv &> /dev/null; then
     echo "uv not found, installing it..."
@@ -33,3 +38,18 @@ else
   uv pip install ${snowflake_path}/snowflake_connector_python*${python_version}*.whl
   uv pip install ${uv_options[@]}
 fi
+
+# ── Universal Driver connector swap ──────────────────────────────────
+# When ud_connector_path is set, replace snowflake-connector-python
+# with the Universal Driver build.  This is a no-op for normal CI.
+# ─────────────────────────────────────────────────────────────────────
+
+ud_connector_path=${ud_connector_path:-""}
+if [[ -n "${ud_connector_path}" ]]; then
+  echo "Swapping snowflake-connector-python → Universal Driver"
+  echo "  UD connector path: ${ud_connector_path}"
+  # --reinstall ensures wheel files are always extracted even if the
+  # version matches a previous install (uv otherwise skips silently).
+  uv pip uninstall snowflake-connector-python
+  uv pip install --reinstall "${ud_connector_path}"
+fi

scripts/ud_tox_install_cmd.sh

@@ -120,6 +120,8 @@ passenv =
     GITHUB_ENV
     SNOWPARK_PYTHON_API_TEST_BUCKET_PATH
     SNOWPARK_PYTHON_API_S3_STORAGE_INTEGRATION
+    ; Universal Driver connector path (no-op when unset)
+    ud_connector_path
 commands =
     notudf: {env:SNOWFLAKE_PYTEST_CMD} -m "{env:SNOWFLAKE_TEST_TYPE} and not udf" {posargs:} {env:RERUN_FLAGS} src/snowflake/snowpark tests
     udf: {env:SNOWFLAKE_PYTEST_CMD} -m "{env:SNOWFLAKE_TEST_TYPE} or udf" {posargs:} {env:RERUN_FLAGS} src/snowflake/snowpark tests
@@ -264,44 +266,6 @@ deps =
 commands =
     {env:SNOWFLAKE_PYTEST_CMD} -m "{env:SNOWFLAKE_TEST_TYPE}" {posargs:} tests/integ/datasource -n 8
 
-[testenv:ud]
-description = run integration tests with Universal Driver connector
-allowlist_externals = bash
-install_command = bash ./scripts/ud_tox_install_cmd.sh {opts} {packages}
-passenv =
-    {[testenv]passenv}
-    ud_connector_path
-    LD_PRELOAD
-setenv =
-    {[testenv]setenv}
-    SNOWFLAKE_PYTEST_VERBOSITY = -vvv
-    SNOWFLAKE_PYTEST_PARALLELISM = -n logical
-    SNOWFLAKE_TEST_TYPE = (unit or integ or doctest)
-    LD_PRELOAD = {env:LD_PRELOAD:}
-commands =
-    {env:SNOWFLAKE_PYTEST_CMD} -m "{env:SNOWFLAKE_TEST_TYPE} or udf" {posargs:} {env:RERUN_FLAGS} tests
-
-[testenv:ud-datasource]
-description = run datasource tests with Universal Driver connector
-allowlist_externals = bash
-install_command = bash ./scripts/ud_tox_install_cmd.sh {opts} {packages}
-passenv =
-    {[testenv]passenv}
-    ud_connector_path
-    LD_PRELOAD
-setenv =
-    {[testenv]setenv}
-    LD_PRELOAD = {env:LD_PRELOAD:}
-deps =
-    {[testenv]deps}
-    databricks-sql-connector > 4.0.0
-    oracledb
-    psycopg2-binary
-    pymysql
-    pyodbc
-commands =
-    {env:SNOWFLAKE_PYTEST_CMD} -m "{env:SNOWFLAKE_TEST_TYPE}" {posargs:} tests/integ/datasource -n 8
-
 [pytest]
 log_cli = True
 log_cli_level = DEBUG