Merge pull request #117 from tylergould/default-timeout

Default timeout

Author: aiham

opentok/opentok.py

@@ -59,10 +59,11 @@ class OpenTok(object):
     TOKEN_SENTINEL = 'T1=='
     """For internal use."""
 
-    def __init__(self, api_key, api_secret, api_url='https://api.opentok.com'):
+    def __init__(self, api_key, api_secret, api_url='https://api.opentok.com', timeout=None):
         self.api_key = str(api_key)
         self.api_secret = api_secret
         self.api_url = api_url
+        self.timeout = timeout
         self._proxies = None
 
     @property
@@ -263,7 +264,7 @@ def create_session(self, location=None, media_mode=MediaModes.relayed, archive_m
             options[u('location')] = location
 
         try:
-            response = requests.post(self.session_url(), data=options, headers=self.headers(), proxies=self.proxies)
+            response = requests.post(self.session_url(), data=options, headers=self.headers(), proxies=self.proxies, timeout=self.timeout)
             response.encoding = 'utf-8'
 
             if response.status_code == 403:
@@ -351,7 +352,7 @@ def start_archive(self, session_id, has_audio=True, has_video=True, name=None, o
                    'outputMode': output_mode.value
         }
 
-        response = requests.post(self.archive_url(), data=json.dumps(payload), headers=self.archive_headers(), proxies=self.proxies)
+        response = requests.post(self.archive_url(), data=json.dumps(payload), headers=self.archive_headers(), proxies=self.proxies, timeout=self.timeout)
 
         if response.status_code < 300:
             return Archive(self, response.json())
@@ -377,7 +378,7 @@ def stop_archive(self, archive_id):
 
         :rtype: The Archive object corresponding to the archive being stopped.
         """
-        response = requests.post(self.archive_url(archive_id) + '/stop', headers=self.archive_headers(), proxies=self.proxies)
+        response = requests.post(self.archive_url(archive_id) + '/stop', headers=self.archive_headers(), proxies=self.proxies, timeout=self.timeout)
 
         if response.status_code < 300:
             return Archive(self, response.json())
@@ -400,7 +401,7 @@ def delete_archive(self, archive_id):
 
         :param String archive_id: The archive ID of the archive to be deleted.
         """
-        response = requests.delete(self.archive_url(archive_id), headers=self.archive_headers(), proxies=self.proxies)
+        response = requests.delete(self.archive_url(archive_id), headers=self.archive_headers(), proxies=self.proxies, timeout=self.timeout)
 
         if response.status_code < 300:
             pass
@@ -418,7 +419,7 @@ def get_archive(self, archive_id):
 
         :rtype: The Archive object.
         """
-        response = requests.get(self.archive_url(archive_id), headers=self.archive_headers(), proxies=self.proxies)
+        response = requests.get(self.archive_url(archive_id), headers=self.archive_headers(), proxies=self.proxies, timeout=self.timeout)
 
         if response.status_code < 300:
             return Archive(self, response.json())
@@ -447,7 +448,7 @@ def get_archives(self, offset=None, count=None):
         if count is not None:
             params['count'] = count
 
-        response = requests.get(self.archive_url() + "?" + urlencode(params), headers=self.archive_headers(), proxies=self.proxies)
+        response = requests.get(self.archive_url() + "?" + urlencode(params), headers=self.archive_headers(), proxies=self.proxies, timeout=self.timeout)
 
         if response.status_code < 300:
             return ArchiveList(self, response.json())

tests/test_http_options.py

@@ -0,0 +1,28 @@
+import unittest
+from six import u
+from nose.tools import raises
+import httpretty
+import requests
+
+from opentok import OpenTok, OpenTokException
+
+def _raise_timeout(*args):
+    raise requests.Timeout('Timeout occurred')
+
+class OpenTokSessionCreationTest(unittest.TestCase):
+    def setUp(self):
+        self.api_key = u('123456')
+        self.api_secret = u('1234567890abcdef1234567890abcdef1234567890')
+        httpretty.enable()
+        httpretty.register_uri(httpretty.POST, u('https://api.opentok.com/session/create'),
+                               body=_raise_timeout,
+                               status=200,
+                               content_type=u('text/xml'))
+
+    def tearDown(self):
+        httpretty.disable()
+
+    @raises(OpenTokException)
+    def test_timeout(self):
+        opentok = OpenTok(self.api_key, self.api_secret, timeout=1)
+        opentok.create_session()

tests/test_initialization.py

@@ -31,7 +31,11 @@ def test_initialization_with_api_url(self):
         assert isinstance(opentok, OpenTok)
 
     def test_initialization_with_numeric_api_key(self):
-        opentok = OpenTok(123456, self.api_secret, self.api_url)
+        opentok = OpenTok(123456, self.api_secret)
+        assert isinstance(opentok, OpenTok)
+
+    def test_initialization_with_timeout(self):
+        opentok = OpenTok(self.api_key, self.api_secret, timeout=5)
         assert isinstance(opentok, OpenTok)
 
 if __name__ == '__main__':