add refresh timeout parameter to listener session

jeffersonkr · jeffersonkr · commit d778e17ce326 · 2019-07-02T12:37:19.000-03:00
diff --git a/firebase_admin/_sseclient.py b/firebase_admin/_sseclient.py
@@ -33,8 +33,8 @@
 class KeepAuthSession(transport.requests.AuthorizedSession):
     """A session that does not drop authentication on redirects between domains."""
 
-    def __init__(self, credential):
-        super(KeepAuthSession, self).__init__(credential)
+    def __init__(self, credential, refresh_timeout):
+        super(KeepAuthSession, self).__init__(credential, refresh_timeout=refresh_timeout)
 
     def rebuild_auth(self, prepared_request, response):
         pass
@@ -107,7 +107,7 @@ def _connect(self):
         if self.should_connect:
             if self.last_id:
                 self.requests_kwargs['headers']['Last-Event-ID'] = self.last_id
-            self.resp = self.session.get(self.url, stream=True, **self.requests_kwargs)
+            self.resp = self.session.get(self.url, stream=True, timeout=self.session.http_timeout, **self.requests_kwargs)
             self.resp_iterator = self.resp.iter_content(decode_unicode=True)
             self.resp.raise_for_status()
         else:
