Fix: some thrift RPCs failed with BadStatusLine (#141)

---------

Signed-off-by: Jesse Whitehouse <[email protected]>

Author: Jesse

CHANGELOG.md

@@ -2,6 +2,8 @@
 
 ## 2.6.x (Unreleased)
 
+- Fix: http.client would raise a BadStatusLine exception in some cases
+
 ## 2.6.0 (2023-06-07)
 
 - Add support for HTTP 1.1 connections (connection pools)

src/databricks/sql/auth/thrift_http_client.py

@@ -115,6 +115,7 @@ def open(self):
             self.__pool = pool_class(self.host, self.port, **_pool_kwargs)
 
     def close(self):
+        self.__resp and self.__resp.drain_conn()
         self.__resp and self.__resp.release_conn()
         self.__resp = None