type issues

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

src/databricks/sql/result_set.py

@@ -73,7 +73,7 @@ def __init__(
         self.status: CommandState = status
         self.has_been_closed_server_side: bool = has_been_closed_server_side
         self.is_direct_results: bool = is_direct_results
-        self.results: Optional[ResultSetQueue] = None  # Children will set this
+        self.results: Optional[ResultSetQueue] = None
         self._is_staging_operation: bool = is_staging_operation
         self.lz4_compressed: bool = lz4_compressed
         self._arrow_schema_bytes: Optional[bytes] = arrow_schema_bytes
@@ -166,7 +166,8 @@ def close(self) -> None:
         been closed on the server for some other reason, issue a request to the server to close it.
         """
         try:
-            self.results.close()
+            if self.results:
+                self.results.close()
             if (
                 self.status != CommandState.CLOSED
                 and not self.has_been_closed_server_side

tests/unit/test_client.py

@@ -219,7 +219,7 @@ def test_closing_result_set_hard_closes_commands(self):
         mock_results_response = Mock()
         mock_results_response.has_been_closed_server_side = False
         mock_connection = Mock()
-        mock_thrift_backend = Mock()
+        mock_thrift_backend = Mock(spec=ThriftDatabricksClient)
         mock_results = Mock()
         # Setup session mock on the mock_connection
         mock_session = Mock()