Skip to content

SNOW-2912540: mock _connect on v5 so local testing skips UD connection_init #4326

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
sfc-gh-fpawlowski wants to merge 10 commits into
base: SNOW-2912540-log-batch-mock-patch
Choose a base branch
from
+5 −0
Open

SNOW-2912540: mock _connect on v5 so local testing skips UD connection_init #4326

Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
6ca591c
SNOW-2912540: mock _connect on v5 so local testing skips UD connectio…
sfc-gh-fpawlowski Aug 21, 2026
c62cfd1
Update _connection.py
sfc-gh-fpawlowski Aug 21, 2026
f0eb42e
Update _connection.py
sfc-gh-fpawlowski Aug 21, 2026
9673aec
Merge branch 'SNOW-2912540-log-batch-mock-patch' into SNOW-2912540-mo…
cursoragent Sep 2, 2026
e3ff578
SNOW-2912540: keep mock connect() and alias _connect on v5
cursoragent Sep 2, 2026
3b58c04
Merge branch 'SNOW-2912540-log-batch-mock-patch' into SNOW-2912540-mo…
cursoragent Sep 2, 2026
ffcdec2
Merge branch 'SNOW-2912540-log-batch-mock-patch' into SNOW-2912540-mo…
cursoragent Sep 2, 2026
cc68f4e
Merge branch 'SNOW-2912540-log-batch-mock-patch' into SNOW-2912540-mo…
cursoragent Sep 2, 2026
b2940c1
Merge branch 'SNOW-2912540-log-batch-mock-patch' into SNOW-2912540-mo…
cursoragent Sep 2, 2026
17b2848
Merge branch 'SNOW-2912540-log-batch-mock-patch' into SNOW-2912540-mo…
sfc-gh-fpawlowski Sep 3, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions src/snowflake/snowpark/mock/_connection.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@
from snowflake.snowpark._internal.error_message import SnowparkClientExceptionMessages
from snowflake.snowpark._internal.server_connection import DEFAULT_STRING_SIZE
from snowflake.snowpark._internal.utils import (
IS_V5_DRIVER,
is_in_stored_procedure,
result_set_to_rows,
)
Expand Down Expand Up @@ -78,6 +79,10 @@ def connect(self, **kwargs) -> None:
}
self._rest = Mock(**attrs)

if IS_V5_DRIVER:
# UD Connection.__init__ calls _connect(); legacy called connect().
_connect = connect

def close(self, retry: bool = True) -> None:
self._rest = None

Expand Down
Loading