Login as partner when there is a token error

Fixes #14

1 files changed, 21 insertions, 0 deletions

diff --git a/tests/test_pandora/test_client.py b/tests/test_pandora/test_client.py
new file mode 100644
index 0000000..cfeebb7
--- /dev/null
+++ b/tests/test_pandora/test_client.py
@@ -0,0 +1,21 @@
+from unittest import TestCase
+
+from pandora.client import BaseAPIClient
+from pandora.transport import APITransport
+from pandora.errors import InvalidAuthToken
+from pandora.py2compat import Mock, MagicMock, call
+
+
+class TestCallingAPIClient(TestCase):
+
+    def test_call_should_retry_on_token_error(self):
+        transport = Mock(side_effect=[InvalidAuthToken(), None])
+
+        client = BaseAPIClient(transport, None, None, None)
+        client._authenticate = Mock()
+
+        client.login("foo", "bar")
+        client("method")
+
+        client._authenticate.assert_called_with()
+        transport.assert_has_calls([call("method"), call("method")])