Skip to content

Commit 0d51d1b

Browse files
op_state -> status in ResultSet
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 4148aac commit 0d51d1b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/unit/test_client.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -679,7 +679,7 @@ def test_resultset_close_handles_cursor_already_closed_error(self):
679679
result_set.backend.CLOSED_OP_STATE = CommandState.CLOSED
680680
result_set.connection = Mock()
681681
result_set.connection.open = True
682-
result_set.op_state = "RUNNING"
682+
result_set.status = CommandState.RUNNING
683683
result_set.has_been_closed_server_side = False
684684
result_set.command_id = Mock()
685685

@@ -693,7 +693,7 @@ def __init__(self):
693693
try:
694694
try:
695695
if (
696-
result_set.op_state != result_set.backend.CLOSED_OP_STATE
696+
result_set.status != result_set.backend.CLOSED_OP_STATE
697697
and not result_set.has_been_closed_server_side
698698
and result_set.connection.open
699699
):
@@ -703,15 +703,15 @@ def __init__(self):
703703
pass
704704
finally:
705705
result_set.has_been_closed_server_side = True
706-
result_set.op_state = result_set.backend.CLOSED_OP_STATE
706+
result_set.status = result_set.backend.CLOSED_OP_STATE
707707

708708
result_set.backend.close_command.assert_called_once_with(
709709
result_set.command_id
710710
)
711711

712712
assert result_set.has_been_closed_server_side is True
713713

714-
assert result_set.op_state == result_set.backend.CLOSED_OP_STATE
714+
assert result_set.status == result_set.backend.CLOSED_OP_STATE
715715
finally:
716716
pass
717717

0 commit comments

Comments
 (0)