Skip to content

Commit b4f9e2d

Browse files
remove session_id_hex args
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent bd2e83b commit b4f9e2d

File tree

2 files changed

+0
-5
lines changed

2 files changed

+0
-5
lines changed

tests/unit/test_client.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,6 @@ def test_closing_connection_closes_commands(self, mock_thrift_client_class):
127127
connection=connection,
128128
execute_response=mock_execute_response,
129129
thrift_client=mock_backend,
130-
session_id_hex=Mock(),
131130
)
132131

133132
# Mock execute_command to return our real result set
@@ -187,7 +186,6 @@ def test_closing_result_set_with_closed_connection_soft_closes_commands(self):
187186
connection=mock_connection,
188187
execute_response=Mock(),
189188
thrift_client=mock_backend,
190-
session_id_hex=Mock(),
191189
)
192190
result_set.results = mock_results
193191

@@ -217,7 +215,6 @@ def test_closing_result_set_hard_closes_commands(self):
217215
mock_connection,
218216
mock_results_response,
219217
mock_thrift_backend,
220-
session_id_hex=Mock(),
221218
)
222219
result_set.results = mock_results
223220

tests/unit/test_fetches.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,6 @@ def make_dummy_result_set_from_initial_results(initial_results):
6363
),
6464
thrift_client=mock_thrift_backend,
6565
t_row_set=None,
66-
session_id_hex=Mock(),
6766
)
6867
return rs
6968

@@ -108,7 +107,6 @@ def fetch_results(
108107
is_staging_operation=False,
109108
),
110109
thrift_client=mock_thrift_backend,
111-
session_id_hex=Mock(),
112110
)
113111
return rs
114112

0 commit comments

Comments
 (0)