fix session object not getting through

runawaycoast · runawaycoast · commit bfaa3b05ede7 · 2022-08-16T10:28:44.000-04:00
diff --git a/nasdaqdatalink/connection.py b/nasdaqdatalink/connection.py
@@ -42,8 +42,7 @@ def request(cls, http_verb, url, **options):
 
     @classmethod
     def execute_request(cls, http_verb, url, **options):
-        params = getattr(options, 'params', None)
-        session = getattr(params, 'session', None)
+        session = options.get('params', {}).get('session', None)
         if session is None:
             session = cls.get_session()
 
diff --git a/test/test_authorized_session.py b/test/test_authorized_session.py
@@ -1,6 +1,4 @@
 from unittest import TestCase
-from unittest.mock import patch
-
 from nasdaqdatalink.model.authorized_session import AuthorizedSession
 from nasdaqdatalink.api_config import ApiConfig
 from requests.sessions import Session
@@ -18,7 +16,10 @@ def test_authorized_session_assign_correct_internal_config(self):
         self.assertTrue(isinstance(authed_session._api_config, ApiConfig))
 
     def test_authorized_session_pass_created_session(self):
+        ApiConfig.use_retries = True
+        ApiConfig.number_of_retries = 130
         authed_session = AuthorizedSession()
         self.assertTrue(isinstance(authed_session._auth_session, Session))
         adapter = authed_session._auth_session.get_adapter(ApiConfig.api_protocol)
         self.assertTrue(isinstance(adapter, HTTPAdapter))
+        self.assertEqual(adapter.max_retries.connect, 130)
