Add some tests

2 files changed, 54 insertions, 0 deletions

diff --git a/tests/test_pandora/test_client.py b/tests/test_pandora/test_client.py
index 835dc0d..62b8440 100644
--- a/tests/test_pandora/test_client.py
+++ b/tests/test_pandora/test_client.py
@@ -5,6 +5,43 @@ from pandora.errors import InvalidAuthToken
 from pandora.py2compat import Mock, MagicMock, call, patch
 
 
+class TestAPIClientLogin(TestCase):
+
+    class StubTransport(object):
+
+        API_VERSION = None
+
+        partner = None
+        user = None
+
+        FAKE_PARTNER = object()
+        FAKE_USER = object()
+
+        def __call__(self, method, **params):
+            if method == "auth.partnerLogin":
+                return self.FAKE_PARTNER
+            elif method == "auth.userLogin":
+                return self.FAKE_USER
+            else:
+                raise AssertionError("Invalid call")
+
+        def set_partner(self, partner):
+            self.partner = partner
+
+        def set_user(self, user):
+            self.user = user
+
+    def test_login(self):
+        transport = self.StubTransport()
+        client = BaseAPIClient(transport, None, None, None)
+        client.login("foobear", "secret")
+
+        self.assertEqual("foobear", client.username)
+        self.assertEqual("secret", client.password)
+        self.assertIs(self.StubTransport.FAKE_USER, transport.user)
+        self.assertIs(self.StubTransport.FAKE_PARTNER, transport.partner)
+
+
 class TestCallingAPIClient(TestCase):
 
     def test_call_should_retry_on_token_error(self):
diff --git a/tests/test_pandora/test_errors.py b/tests/test_pandora/test_errors.py
new file mode 100644
index 0000000..778771a
--- /dev/null
+++ b/tests/test_pandora/test_errors.py
@@ -0,0 +1,17 @@
+from unittest import TestCase
+
+from pandora.errors import InternalServerError, PandoraException
+
+
+class TestPandoraExceptionConstructionFromErrorCode(TestCase):
+
+    def test_it_returns_specific_error_class_if_possible(self):
+        error = PandoraException.from_code(0, "Test Message")
+        self.assertIsInstance(error, InternalServerError)
+        self.assertEqual("Test Message", error.extended_message)
+        self.assertEqual(0, error.code)
+
+    def test_it_returns_generic_error_if_unknown(self):
+        error = PandoraException.from_code(-99, "Test Message")
+        self.assertIsInstance(error, PandoraException)
+        self.assertEqual("Test Message", error.extended_message)