DEBUG - run single test workflow

.github/workflows/run_single_test.yml

@@ -0,0 +1,35 @@
+name: Run custom pytest
+
+on:
+  push:
+
+jobs:
+  run-pytest:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        cloud-provider: [aws, azure, gcp]
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.11'
+
+      - name: Setup parameters file
+        shell: bash
+        env:
+          PARAMETERS_SECRET: ${{ secrets.PARAMETERS_SECRET }}
+        run: |
+          gpg --quiet --batch --yes --decrypt --passphrase="$PARAMETERS_SECRET" \
+          .github/workflows/parameters/public/parameters_${{ matrix.cloud-provider }}.py.gpg > test/parameters.py
+
+      - name: Install dependencies
+        run: |
+          python -m pip install ".[development,aio,secure-local-storage]"
+
+      - name: Run pytest
+        run: |
+          pytest -vvv -k "otel_error" .

src/snowflake/connector/aio/_network.py

@@ -217,11 +217,19 @@ async def request(
             HTTP_HEADER_USER_AGENT: PYTHON_CONNECTOR_USER_AGENT,
         }
         try:
-            from opentelemetry.propagate import inject
+            # SNOW-1763555: inject OpenTelemetry headers if available specifically in WC3 format
+            #  into our request headers in case tracing is enabled. This should make sure that
+            #  our requests are accounted for properly if OpenTelemetry is used by users.
+            from opentelemetry.trace.propagation.tracecontext import (
+                TraceContextTextMapPropagator,
+            )
 
-            inject(headers)
-        except ModuleNotFoundError as e:
-            logger.debug(f"Opentelemtry otel injection failed because of: {e}")
+            TraceContextTextMapPropagator().inject(headers)
+        except Exception:
+            logger.debug(
+                "Opentelemtry otel injection failed",
+                exc_info=True,
+            )
         if self._connection.service_name:
             headers[HTTP_HEADER_SERVICE_NAME] = self._connection.service_name
         if method == "post":

test/integ/aio/conftest.py

@@ -141,6 +141,7 @@ async def aio_connection(db_parameters):
         warehouse=db_parameters["warehouse"],
         protocol=db_parameters["protocol"],
         timezone="UTC",
+        role=db_parameters.get("role", None),
     )
     yield cnx
     await cnx.close()

test/integ/conftest.py

@@ -182,6 +182,7 @@ def init_test_schema(db_parameters) -> Generator[None, None, None]:
         database=ret["database"],
         account=ret["account"],
         protocol=ret["protocol"],
+        role=ret.get("role", None),
     ) as con:
         con.cursor().execute(f"CREATE SCHEMA IF NOT EXISTS {TEST_SCHEMA}")
         yield

test/unit/aio/test_connection_async_unit.py

@@ -551,3 +551,19 @@ async def test_ssl_error_hint(caplog):
         exc.value, OperationalError
     )
     assert "SSL error" in caplog.text and _CONNECTIVITY_ERR_MSG in caplog.text
+
+
+async def test_otel_error_message_async(caplog, mock_post_requests):
+    """This test assumes that OpenTelemetry is not installed when tests are running."""
+    with mock.patch("snowflake.connector.aio._network.SnowflakeRestful._post_request"):
+        with caplog.at_level(logging.DEBUG):
+            async with fake_db_conn():
+                ...
+    assert caplog.records
+    important_records = [
+        record
+        for record in caplog.records
+        if "Opentelemtry otel injection failed" in record.message
+    ]
+    assert len(important_records) == 1
+    assert important_records[0].exc_text is not None