Add config builders and tests

7 files changed, 238 insertions, 0 deletions

diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/__init__.py
diff --git a/tests/pandora/__init__.py b/tests/pandora/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/pandora/__init__.py
diff --git a/tests/pandora/clientbuilder.py b/tests/pandora/clientbuilder.py
new file mode 100644
index 0000000..b953f9e
--- /dev/null
+++ b/tests/pandora/clientbuilder.py
@@ -0,0 +1,214 @@
+import os
+from unittest import TestCase
+
+try:
+    from unittest.mock import Mock
+except ImportError:
+    from mock import Mock
+
+import pandora.clientbuilder as cb
+from pandora import APIClient, DEFAULT_API_HOST
+
+
+class TestTranslatingDict(TestCase):
+
+    class TestDict(cb.TranslatingDict):
+
+        KEY_TRANSLATIONS = { "FOO": "BAR" }
+        VALUE_TRANSLATIONS = { "BAZ": lambda v: v + 1 }
+        callback_value = None
+
+        def was_translated(self, from_key, to_key):
+            self.callback_value = (from_key, to_key)
+
+    def setUp(self):
+        self.dct = self.TestDict()
+
+    def test_construction_with_dict(self):
+        dct = self.TestDict({ "BIZ": 1, "BUZ": 2 })
+
+        self.assertEqual(1, dct["BIZ"])
+        self.assertEqual(2, dct["BUZ"])
+
+    def test_construction_with_list(self):
+        dct = self.TestDict([("key", "value")])
+
+        self.assertEqual("value", dct["KEY"])
+
+    def test_key_translation(self):
+        self.dct.put(" TEST ", "value")
+        self.dct.put("MoRe", 1)
+        self.dct.put("foo", True)
+
+        self.assertEqual("value", self.dct["TEST"])
+        self.assertEqual(1, self.dct["MORE"])
+        self.assertEqual(True, self.dct["BAR"])
+
+    def test_value_translation(self):
+        dct = self.TestDict({ " Baz": 41 })
+
+        self.assertEqual(42, dct["BAZ"])
+
+    def test_setitem(self):
+        self.dct["Foo"] = "bar"
+
+        self.assertEqual("bar", self.dct["BAR"])
+
+    def test_put(self):
+        self.dct.put("Foo", "bar")
+
+        self.assertEqual("bar", self.dct["BAR"])
+
+    def test_key_translation_hook(self):
+        self.dct.put("Foo", "bar")
+        self.assertEqual(("FOO", "BAR"), self.dct.callback_value)
+
+
+class TestSettingsDictBuilder(TestCase):
+
+    def _build_minimal(self):
+        return cb.SettingsDictBuilder({
+            "DECRYPTION_KEY": "dec",
+            "ENCRYPTION_KEY": "enc",
+            "PARTNER_USER": "user",
+            "PARTNER_PASSWORD": "pass",
+            "DEVICE": "dev",
+        }).build()
+
+    def _build_maximal(self):
+        return cb.SettingsDictBuilder({
+            "DECRYPTION_KEY": "dec",
+            "ENCRYPTION_KEY": "enc",
+            "PARTNER_USER": "user",
+            "PARTNER_PASSWORD": "pass",
+            "DEVICE": "dev",
+            "PROXY": "proxy.example.com",
+            "AUDIO_QUALITY": "high",
+            "API_HOST": "example.com",
+        }).build()
+
+    def test_building(self):
+        client = self._build_minimal()
+
+        self.assertTrue(isinstance(client, APIClient))
+
+    def test_default_values(self):
+        client = self._build_minimal()
+
+        self.assertIsNone(client.transport.proxy)
+        self.assertEqual(DEFAULT_API_HOST, client.transport.api_host)
+        self.assertEqual(APIClient.MED_AUDIO_QUALITY,
+                client.default_audio_quality)
+
+    def test_validate_client(self):
+        client = self._build_maximal()
+
+        self.assertIsNotNone(client.transport.cryptor.bf_in)
+        self.assertIsNotNone(client.transport.cryptor.bf_out)
+
+        self.assertEqual("user", client.partner_user)
+        self.assertEqual("pass", client.partner_password)
+        self.assertEqual("dev", client.device)
+
+        self.assertEqual("proxy.example.com", client.transport.proxy)
+        self.assertEqual("example.com", client.transport.api_host)
+        self.assertEqual("high", client.default_audio_quality)
+
+
+class TestFileBasedBuilder(TestCase):
+
+    class StubBuilder(cb.FileBasedClientBuilder):
+
+        DEFAULT_CONFIG_FILE = "foo"
+
+        def parse_config(self):
+            return { "USER": { "USERNAME": "U", "PASSWORD": "P" }}
+
+        def build_from_settings_dict(self, config):
+            mock = Mock()
+            mock.login = Mock()
+            return mock
+
+    def test_default_config(self):
+        builder = self.StubBuilder()
+
+        self.assertEqual("foo", builder.path)
+
+    def test_setting_valid_path(self):
+        builder = cb.FileBasedClientBuilder(__file__)
+
+        self.assertTrue(builder.file_exists)
+        self.assertEqual(__file__, builder.path)
+
+    def test_setting_invalid_path(self):
+        builder = cb.FileBasedClientBuilder("nowhere")
+
+        with self.assertRaises(IOError):
+            builder.build()
+
+        self.assertFalse(builder.file_exists)
+
+    def test_setting_user_path(self):
+        builder = cb.FileBasedClientBuilder("~/")
+
+        self.assertEqual(os.path.expanduser("~/"), builder.path)
+
+    def test_logging_in(self):
+        client = self.StubBuilder(__file__, True).build()
+        client.login.assert_called_once_with("U", "P")
+
+    def test_not_logging_in(self):
+        client = self.StubBuilder(__file__, False).build()
+        self.assertFalse(client.login.called)
+
+
+class TestPydoraConfigFileBuilder(TestCase):
+
+    def test_cfg_to_dict(self):
+        cfg = Mock()
+        cfg.items = Mock(return_value=[("a", "b"), ("c", "d")])
+
+        dct = cb.PydoraConfigFileBuilder.cfg_to_dict(cfg, "foo")
+
+        self.assertEqual("b", dct["A"])
+        self.assertEqual("d", dct["C"])
+
+    def test_integration(self):
+        path = os.path.join(os.path.dirname(__file__), "pydora.cfg")
+        cfg = cb.PydoraConfigFileBuilder(path).parse_config()
+
+        self.assertDictEqual(cfg, {
+            "AUDIO_QUALITY": "test_quality",
+            "DECRYPTION_KEY": "test_decryption_key",
+            "DEVICE": "test_device",
+            "ENCRYPTION_KEY": "test_encryption_key",
+            "PARTNER_PASSWORD": "test_partner_password",
+            "PARTNER_USER": "test_partner_username",
+            "API_HOST": "test_host",
+            "USER": {
+                "USERNAME": "test_username",
+                "PASSWORD": "test_password",
+                }
+            })
+
+
+class TestPianobarConfigFileBuilder(TestCase):
+
+    def test_integration(self):
+        path = os.path.join(os.path.dirname(__file__), "pianobar.cfg")
+        cfg = cb.PianobarConfigFileBuilder(path).parse_config()
+
+        self.assertDictEqual(cfg, {
+            "AUDIO_QUALITY": "test_qualityQuality",
+            "DECRYPTION_KEY": "test_decryption_key",
+            "DEVICE": "test_device",
+            "ENCRYPTION_KEY": "test_encryption_key",
+            "PARTNER_PASSWORD": "test_partner_password",
+            "PARTNER_USER": "test_partner_username",
+            "API_HOST": "test_host/services/json/",
+            "PROXY": "test_proxy",
+            "USER": {
+                "USERNAME": "test_username",
+                "PASSWORD": "test_password",
+                }
+            })
diff --git a/tests/pandora/models/__init__.py b/tests/pandora/models/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/pandora/models/__init__.py
diff --git a/tests/pandora/pianobar.cfg b/tests/pandora/pianobar.cfg
new file mode 100644
index 0000000..7a915d8
--- /dev/null
+++ b/tests/pandora/pianobar.cfg
@@ -0,0 +1,12 @@
+# high-quality audio (192k mp3, for Pandora One subscribers only!)
+audio_quality = test_quality
+rpc_host = test_host
+partner_user = test_partner_username
+partner_password = test_partner_password
+device = test_device
+encrypt_password = test_encryption_key
+decrypt_password = test_decryption_key
+control_proxy = test_proxy
+
+user = test_username
+password = test_password
diff --git a/tests/pandora/pydora.cfg b/tests/pandora/pydora.cfg
new file mode 100644
index 0000000..854e377
--- /dev/null
+++ b/tests/pandora/pydora.cfg
@@ -0,0 +1,12 @@
+[api]
+api_host = test_host
+encryption_key = test_encryption_key
+decryption_key = test_decryption_key
+username = test_partner_username
+password = test_partner_password
+device = test_device
+default_audio_quality = test_quality
+
+[user]
+username = test_username
+password = test_password
diff --git a/tests/pydora/__init__.py b/tests/pydora/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/pydora/__init__.py