Add test for transport.__call__ decorator.

2 files changed, 31 insertions, 3 deletions

diff --git a/tests/test_pandora/test_clientbuilder.py b/tests/test_pandora/test_clientbuilder.py
index 4dc5b5e..27117b0 100644
--- a/tests/test_pandora/test_clientbuilder.py
+++ b/tests/test_pandora/test_clientbuilder.py
@@ -62,6 +62,7 @@ class TestTranslatingDict(TestCase):
 
 class TestSettingsDictBuilder(TestCase):
 
+    @classmethod
     def _build_minimal(self):
         return cb.SettingsDictBuilder({
             "DECRYPTION_KEY": "dec",
@@ -71,6 +72,7 @@ class TestSettingsDictBuilder(TestCase):
             "DEVICE": "dev",
         }).build()
 
+    @classmethod
     def _build_maximal(self):
         return cb.SettingsDictBuilder({
             "DECRYPTION_KEY": "dec",
@@ -84,12 +86,12 @@ class TestSettingsDictBuilder(TestCase):
         }).build()
 
     def test_building(self):
-        client = self._build_minimal()
+        client = TestSettingsDictBuilder._build_minimal()
 
         self.assertTrue(isinstance(client, APIClient))
 
     def test_default_values(self):
-        client = self._build_minimal()
+        client = TestSettingsDictBuilder._build_minimal()
 
         self.assertEqual({}, client.transport._http.proxies)
         self.assertEqual(DEFAULT_API_HOST, client.transport.api_host)
@@ -97,7 +99,7 @@ class TestSettingsDictBuilder(TestCase):
                 client.default_audio_quality)
 
     def test_validate_client(self):
-        client = self._build_maximal()
+        client = TestSettingsDictBuilder._build_maximal()
         expected_proxies = {
                 "http": "proxy.example.com",
                 "https": "proxy.example.com"
diff --git a/tests/test_pandora/test_transport.py b/tests/test_pandora/test_transport.py
new file mode 100644
index 0000000..a9261bd
--- /dev/null
+++ b/tests/test_pandora/test_transport.py
@@ -0,0 +1,26 @@
+import time
+from unittest import TestCase
+
+from OpenSSL.SSL import SysCallError
+
+from pandora.py2compat import Mock, call
+
+from tests.test_pandora.test_clientbuilder import TestSettingsDictBuilder
+
+
+class TestTransport(TestCase):
+
+    def test_call_should_retry_max_times_on_sys_call_error(self):
+        with self.assertRaises(SysCallError) as context:
+
+            client = TestSettingsDictBuilder._build_minimal()
+
+            time.sleep = Mock()
+            client.transport._make_http_request = Mock(side_effect=SysCallError("mock_error"))
+            client.transport._start_request = Mock()
+
+            client("method")
+
+        client.transport._start_request.assert_has_calls([call("method")])
+        assert client.transport._start_request.call_count == 5
+        self.assertTrue('mock_error' in context.exception)