Update tests

wiggzz · wiggzz · commit af4a41186e37 · 2025-10-06T13:47:09.000-05:00
diff --git a/dbt-snowflake/tests/unit/test_snowflake_adapter.py b/dbt-snowflake/tests/unit/test_snowflake_adapter.py
@@ -292,6 +292,7 @@ def test_client_session_keep_alive_false_by_default(self):
                     insecure_mode=False,
                     session_parameters={},
                     reuse_connections=True,
+                    client_prefetch_threads=1,
                 ),
             ]
         )
@@ -325,6 +326,7 @@ def test_client_session_keep_alive_true(self):
                     insecure_mode=False,
                     session_parameters={},
                     reuse_connections=None,
+                    client_prefetch_threads=1,
                 )
             ]
         )
@@ -352,6 +354,7 @@ def test_client_has_query_tag(self):
                     application="dbt",
                     insecure_mode=False,
                     session_parameters={"QUERY_TAG": "test_query_tag"},
+                    client_prefetch_threads=1,
                 )
             ]
         )
@@ -387,6 +390,7 @@ def test_user_pass_authentication(self):
                     insecure_mode=False,
                     session_parameters={},
                     reuse_connections=True,
+                    client_prefetch_threads=1,
                 )
             ]
         )
@@ -421,6 +425,7 @@ def test_authenticator_user_pass_authentication(self):
                     insecure_mode=False,
                     session_parameters={},
                     reuse_connections=True,
+                    client_prefetch_threads=1,
                 )
             ]
         )
@@ -451,6 +456,7 @@ def test_authenticator_externalbrowser_authentication(self):
                     insecure_mode=False,
                     session_parameters={},
                     reuse_connections=True,
+                    client_prefetch_threads=1,
                 )
             ]
         )
@@ -485,6 +491,7 @@ def test_authenticator_oauth_authentication(self):
                     insecure_mode=False,
                     session_parameters={},
                     reuse_connections=True,
+                    client_prefetch_threads=1,
                 )
             ]
         )
@@ -519,6 +526,7 @@ def test_authenticator_private_key_authentication(self, mock_get_private_key):
                     insecure_mode=False,
                     session_parameters={},
                     reuse_connections=True,
+                    client_prefetch_threads=1,
                 )
             ]
         )
@@ -553,6 +561,7 @@ def test_authenticator_private_key_authentication_no_passphrase(self, mock_get_p
                     insecure_mode=False,
                     session_parameters={},
                     reuse_connections=True,
+                    client_prefetch_threads=1,
                 )
             ]
         )
@@ -585,6 +594,7 @@ def test_authenticator_jwt_authentication(self):
                     insecure_mode=False,
                     session_parameters={},
                     reuse_connections=True,
+                    client_prefetch_threads=1,
                 )
             ]
         )
@@ -615,6 +625,7 @@ def test_query_tag(self):
                     insecure_mode=False,
                     session_parameters={"QUERY_TAG": "test_query_tag"},
                     reuse_connections=True,
+                    client_prefetch_threads=1,
                 )
             ]
         )
@@ -644,6 +655,7 @@ def test_reuse_connections_with_keep_alive(self):
                     insecure_mode=False,
                     session_parameters={},
                     reuse_connections=True,
+                    client_prefetch_threads=1,
                 )
             ]
         )
@@ -678,6 +690,7 @@ def test_authenticator_private_key_string_authentication(self, mock_get_private_
                     insecure_mode=False,
                     session_parameters={},
                     reuse_connections=True,
+                    client_prefetch_threads=1,
                 )
             ]
         )
@@ -714,6 +727,7 @@ def test_authenticator_private_key_string_authentication_no_passphrase(
                     insecure_mode=False,
                     session_parameters={},
                     reuse_connections=True,
+                    client_prefetch_threads=1,
                 )
             ]
         )

Original file line number	Diff line number	Diff line change
`@@ -292,6 +292,7 @@ def test_client_session_keep_alive_false_by_default(self):`
`292`	`292`	`insecure_mode=False,`
`293`	`293`	`session_parameters={},`
`294`	`294`	`reuse_connections=True,`
	`295`	`+ client_prefetch_threads=1,`
`295`	`296`	`),`
`296`	`297`	`]`
`297`	`298`	`)`
`@@ -325,6 +326,7 @@ def test_client_session_keep_alive_true(self):`
`325`	`326`	`insecure_mode=False,`
`326`	`327`	`session_parameters={},`
`327`	`328`	`reuse_connections=None,`
	`329`	`+ client_prefetch_threads=1,`
`328`	`330`	`)`
`329`	`331`	`]`
`330`	`332`	`)`
`@@ -352,6 +354,7 @@ def test_client_has_query_tag(self):`
`352`	`354`	`application="dbt",`
`353`	`355`	`insecure_mode=False,`
`354`	`356`	`session_parameters={"QUERY_TAG": "test_query_tag"},`
	`357`	`+ client_prefetch_threads=1,`
`355`	`358`	`)`
`356`	`359`	`]`
`357`	`360`	`)`
`@@ -387,6 +390,7 @@ def test_user_pass_authentication(self):`
`387`	`390`	`insecure_mode=False,`
`388`	`391`	`session_parameters={},`
`389`	`392`	`reuse_connections=True,`
	`393`	`+ client_prefetch_threads=1,`
`390`	`394`	`)`
`391`	`395`	`]`
`392`	`396`	`)`
`@@ -421,6 +425,7 @@ def test_authenticator_user_pass_authentication(self):`
`421`	`425`	`insecure_mode=False,`
`422`	`426`	`session_parameters={},`
`423`	`427`	`reuse_connections=True,`
	`428`	`+ client_prefetch_threads=1,`
`424`	`429`	`)`
`425`	`430`	`]`
`426`	`431`	`)`
`@@ -451,6 +456,7 @@ def test_authenticator_externalbrowser_authentication(self):`
`451`	`456`	`insecure_mode=False,`
`452`	`457`	`session_parameters={},`
`453`	`458`	`reuse_connections=True,`
	`459`	`+ client_prefetch_threads=1,`
`454`	`460`	`)`
`455`	`461`	`]`
`456`	`462`	`)`
`@@ -485,6 +491,7 @@ def test_authenticator_oauth_authentication(self):`
`485`	`491`	`insecure_mode=False,`
`486`	`492`	`session_parameters={},`
`487`	`493`	`reuse_connections=True,`
	`494`	`+ client_prefetch_threads=1,`
`488`	`495`	`)`
`489`	`496`	`]`
`490`	`497`	`)`
`@@ -519,6 +526,7 @@ def test_authenticator_private_key_authentication(self, mock_get_private_key):`
`519`	`526`	`insecure_mode=False,`
`520`	`527`	`session_parameters={},`
`521`	`528`	`reuse_connections=True,`
	`529`	`+ client_prefetch_threads=1,`
`522`	`530`	`)`
`523`	`531`	`]`
`524`	`532`	`)`
`@@ -553,6 +561,7 @@ def test_authenticator_private_key_authentication_no_passphrase(self, mock_get_p`
`553`	`561`	`insecure_mode=False,`
`554`	`562`	`session_parameters={},`
`555`	`563`	`reuse_connections=True,`
	`564`	`+ client_prefetch_threads=1,`
`556`	`565`	`)`
`557`	`566`	`]`
`558`	`567`	`)`
`@@ -585,6 +594,7 @@ def test_authenticator_jwt_authentication(self):`
`585`	`594`	`insecure_mode=False,`
`586`	`595`	`session_parameters={},`
`587`	`596`	`reuse_connections=True,`
	`597`	`+ client_prefetch_threads=1,`
`588`	`598`	`)`
`589`	`599`	`]`
`590`	`600`	`)`
`@@ -615,6 +625,7 @@ def test_query_tag(self):`
`615`	`625`	`insecure_mode=False,`
`616`	`626`	`session_parameters={"QUERY_TAG": "test_query_tag"},`
`617`	`627`	`reuse_connections=True,`
	`628`	`+ client_prefetch_threads=1,`
`618`	`629`	`)`
`619`	`630`	`]`
`620`	`631`	`)`
`@@ -644,6 +655,7 @@ def test_reuse_connections_with_keep_alive(self):`
`644`	`655`	`insecure_mode=False,`
`645`	`656`	`session_parameters={},`
`646`	`657`	`reuse_connections=True,`
	`658`	`+ client_prefetch_threads=1,`
`647`	`659`	`)`
`648`	`660`	`]`
`649`	`661`	`)`
`@@ -678,6 +690,7 @@ def test_authenticator_private_key_string_authentication(self, mock_get_private_`
`678`	`690`	`insecure_mode=False,`
`679`	`691`	`session_parameters={},`
`680`	`692`	`reuse_connections=True,`
	`693`	`+ client_prefetch_threads=1,`
`681`	`694`	`)`
`682`	`695`	`]`
`683`	`696`	`)`
`@@ -714,6 +727,7 @@ def test_authenticator_private_key_string_authentication_no_passphrase(`
`714`	`727`	`insecure_mode=False,`
`715`	`728`	`session_parameters={},`
`716`	`729`	`reuse_connections=True,`
	`730`	`+ client_prefetch_threads=1,`
`717`	`731`	`)`
`718`	`732`	`]`
`719`	`733`	`)`