Skip to content

Commit 3a5a357

Browse files
pass test_input to get_protocol_version instead of session_id to maintain previous API
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 5c9d6fe commit 3a5a357

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

tests/unit/test_parameters.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -64,13 +64,7 @@ class TestSessionHandleChecks(object):
6464
],
6565
)
6666
def test_get_protocol_version_fallback_behavior(self, test_input, expected):
67-
properties = (
68-
{"serverProtocolVersion": test_input.serverProtocolVersion}
69-
if test_input.serverProtocolVersion
70-
else {}
71-
)
72-
session_id = SessionId.from_thrift_handle(test_input.sessionHandle, properties)
73-
assert Connection.get_protocol_version(session_id) == expected
67+
assert Connection.get_protocol_version(test_input) == expected
7468

7569
@pytest.mark.parametrize(
7670
"test_input,expected",

0 commit comments

Comments
 (0)