From 3de319435697e65a3b5d8570adc3ddaf88ee857f Mon Sep 17 00:00:00 2001 From: Mike Crute Date: Sat, 1 Sep 2018 04:17:23 +0000 Subject: Add missing tests for PandoraType --- tests/test_pandora/test_models.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) 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) -- cgit v1.2.3