add has_been_closed_server_side assertion

Signed-off-by: varun-edachali-dbx <[email protected]>

Author: varun-edachali-dbx

tests/unit/test_client.py

@@ -102,6 +102,10 @@ def test_closing_connection_closes_commands(self):
                 connection.close()
 
                 # Check that the manually created mock result set's close method was called
+                self.assertEqual(
+                    mock_result_set.has_been_closed_server_side, 
+                    closed,
+                )
                 mock_result_set.close.assert_called_once_with()
 
     @patch("%s.session.ThriftDatabricksClient" % PACKAGE_NAME)