Add requests timeout

adal/authentication_context.py

@@ -47,7 +47,7 @@ class AuthenticationContext(object):
 
     def __init__(
             self, authority, validate_authority=None, cache=None,
-            api_version='1.0'):
+            api_version='1.0', timeout=None):
         '''Creates a new AuthenticationContext object.
 
         By default the authority will be checked against a list of known Azure
@@ -70,6 +70,9 @@ def __init__(
             Developers are now encouraged to set it as None explicitly,
             which means the underlying API version will be automatically chosen.
             In next major release, this default value will be changed to None.
+        :param timeout: (optional) requests timeout. How long to wait for the server to send
+            data before giving up, as a float, or a `(connect timeout,
+            read timeout) <timeouts>` tuple.
         '''
         self.authority = Authority(authority, validate_authority is None or validate_authority)
         self._oauth2client = None
@@ -89,7 +92,8 @@ def __init__(
         self._call_context = {
             'options': GLOBAL_ADAL_OPTIONS,
             'api_version': api_version,
-            'verify_ssl': None if env_value is None else not env_value # mainly for tracing through proxy
+            'verify_ssl': None if env_value is None else not env_value, # mainly for tracing through proxy
+            'timeout':timeout
             }
         self._token_requests_with_user_code = {}
         self.cache = cache or TokenCache()

adal/oauth2_client.py

@@ -263,7 +263,8 @@ def get_token(self, oauth_parameters):
             resp = requests.post(token_url.geturl(), 
                                  data=url_encoded_token_request, 
                                  headers=post_options['headers'],
-                                 verify=self._call_context.get('verify_ssl', None))
+                                 verify=self._call_context.get('verify_ssl', None),
+                                 timeout=self._call_context.get('timeout', None))
 
             util.log_return_correlation_id(self._log, operation, resp)
         except Exception:
@@ -294,7 +295,8 @@ def get_user_code_info(self, oauth_parameters):
             resp = requests.post(device_code_url.geturl(), 
                                  data=url_encoded_code_request, 
                                  headers=post_options['headers'],
-                                 verify=self._call_context.get('verify_ssl', None))
+                                 verify=self._call_context.get('verify_ssl', None),
+                                 timeout=self._call_context.get('timeout', None))
             util.log_return_correlation_id(self._log, operation, resp)
         except Exception:
             self._log.exception("%(operation)s request failed", {"operation": operation})