Add missing tests for PandoraType

1 files changed, 15 insertions, 0 deletions

diff --git a/tests/test_pandora/test_models.py b/tests/test_pandora/test_models.py
index 808efa4..ce71ef8 100644
--- a/tests/test_pandora/test_models.py
+++ b/tests/test_pandora/test_models.py
@@ -642,3 +642,18 @@ class TestBookmark(TestCase):
     def test_delete_artist_bookmark(self):
         pm.Bookmark.from_json(self.client, self.ARTIST_BOOKMARK).delete()
         self.client.delete_artist_bookmark.assert_called_with("token")
+
+
+class TestPandoraType(TestCase):
+
+    def test_it_can_be_built_from_a_model(self):
+        pt = pm.PandoraType.from_model(None, "TR")
+        self.assertIs(pm.PandoraType.TRACK, pt)
+
+    def test_it_can_be_built_from_string(self):
+        pt = pm.PandoraType.from_string("TR")
+        self.assertIs(pm.PandoraType.TRACK, pt)
+
+    def test_it_returns_genre_for_unknown_string(self):
+        pt = pm.PandoraType.from_string("FOO")
+        self.assertIs(pm.PandoraType.GENRE, pt)