ai: apply changes for #932 (1 review thread)

Addresses:
  - #3859712688 at src/databricks/sql/session.py:197

Signed-off-by: peco-engineer-bot[bot] <peco-engineer-bot[bot]@users.noreply.github.com>

Author: peco-engineer-bot[bot]

tests/unit/test_session.py

@@ -526,6 +526,48 @@ def test_azure_sp_m2m_kwargs_threaded_into_kernel_auth_options(self):
             finally:
                 conn.close()
 
+    def test_oauth_token_cache_enabled_threaded_into_kernel_auth_options(self):
+        # oauth_token_cache_enabled must reach the kernel auth bridge via
+        # auth_options; without this session.py mapping line the feature
+        # would silently regress to always-disabled (the safe default masks
+        # the failure). Guards the session.py -> kernel_auth_options map.
+        import sys
+        import types
+
+        pytest.importorskip(
+            "pyarrow",
+            reason="kernel client module imports pyarrow at load",
+        )
+
+        fake = types.ModuleType("databricks_sql_kernel")
+        fake.KernelError = type("KernelError", (Exception,), {})
+        fake.Session = MagicMock()
+
+        with patch.dict(sys.modules, {"databricks_sql_kernel": fake}), patch(
+            "databricks.sql.backend.kernel.client.KernelDatabricksClient"
+        ) as mock_kernel_client, patch(
+            "%s.session.get_python_sql_connector_auth_provider" % self.PACKAGE
+        ):
+            instance = mock_kernel_client.return_value
+            instance.open_session.return_value = SessionId(
+                BackendType.SEA, "sess-id", None
+            )
+
+            conn = databricks.sql.connect(
+                server_hostname="foo",
+                http_path="/sql/1.0/warehouses/abc",
+                use_kernel=True,
+                auth_type="databricks-oauth",
+                oauth_token_cache_enabled=True,
+                enable_telemetry=False,
+            )
+            try:
+                _, kwargs = mock_kernel_client.call_args
+                opts = kwargs["auth_options"]
+                assert opts["oauth_token_cache_enabled"] is True
+            finally:
+                conn.close()
+
 
 class TestKernelUserAgentForwarding:
     """user_agent_entry must reach the kernel on the use_kernel path —