Skip to content

Commit 450b80d

Browse files
remove un-necessary initialisation assertions
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 75752bf commit 450b80d

File tree

1 file changed

+0
-13
lines changed

1 file changed

+0
-13
lines changed

tests/unit/test_client.py

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -121,25 +121,12 @@ def test_closing_connection_closes_commands(self, mock_thrift_client_class):
121121
thrift_client=mock_backend,
122122
)
123123

124-
# Verify initial state
125-
self.assertEqual(real_result_set.has_been_closed_server_side, closed)
126-
expected_op_state = (
127-
CommandState.CLOSED if closed else CommandState.SUCCEEDED
128-
)
129-
self.assertEqual(real_result_set.op_state, expected_op_state)
130-
131124
# Mock execute_command to return our real result set
132125
cursor.backend.execute_command = Mock(return_value=real_result_set)
133126

134127
# Execute a command - this should set cursor.active_result_set to our real result set
135128
cursor.execute("SELECT 1")
136129

137-
# Verify that cursor.execute() set up the result set correctly
138-
self.assertIsInstance(cursor.active_result_set, ThriftResultSet)
139-
self.assertEqual(
140-
cursor.active_result_set.has_been_closed_server_side, closed
141-
)
142-
143130
# Close the connection - this should trigger the real close chain:
144131
# connection.close() -> cursor.close() -> result_set.close()
145132
connection.close()

0 commit comments

Comments
 (0)