Allow injection of sessions (#1111)

MrwanBaghdad · web-flow · commit a62ad5a2d702 · 2022-09-22T21:01:44.000-07:00
* Allow injection of session_factory to allow use of a custom session
diff --git a/tableauserverclient/server/server.py b/tableauserverclient/server/server.py
@@ -54,12 +54,13 @@ class PublishMode:
         Overwrite = "Overwrite"
         CreateNew = "CreateNew"
 
-    def __init__(self, server_address, use_server_version=False, http_options=None):
+    def __init__(self, server_address, use_server_version=False, http_options=None, session_factory=requests.Session):
         self._server_address = server_address
         self._auth_token = None
         self._site_id = None
         self._user_id = None
-        self._session = requests.Session()
+        self._session_factory = session_factory
+        self._session = session_factory()
         self._http_options = dict()
 
         self.version = default_server_version
@@ -120,7 +121,7 @@ def _clear_auth(self):
         self._site_id = None
         self._user_id = None
         self._auth_token = None
-        self._session = requests.Session()
+        self._session = self._session_factory()
 
     def _set_auth(self, site_id, user_id, auth_token):
         self._site_id = site_id
