Additional error checking and test cases for retrieving ad metadata.

1 files changed, 44 insertions, 0 deletions

diff --git a/tests/test_pandora/test_client.py b/tests/test_pandora/test_client.py
index 71435a4..e7ef87f 100644
--- a/tests/test_pandora/test_client.py
+++ b/tests/test_pandora/test_client.py
@@ -91,3 +91,47 @@ class TestGettingQualities(TestCase):
                 BaseAPIClient.MED_AUDIO_QUALITY]
 
         self.assertEqual(expected, result)
+
+
+class TestGettingAds(TestCase):
+
+    mock_ad_metadata_result = {
+        'audioUrlMap': {
+            'mediumQuality': {
+                'audioUrl': 'mock_med_url', 'bitrate': '64', 'protocol': 'http', 'encoding': 'aacplus'
+            },
+            'highQuality': {
+                'audioUrl': 'mock_high_url', 'bitrate': '64', 'protocol': 'http', 'encoding': 'aacplus'
+            },
+            'lowQuality': {
+                'audioUrl': 'mock_low_url', 'bitrate': '32', 'protocol': 'http', 'encoding': 'aacplus'}},
+            'clickThroughUrl': 'mock_click_url',
+            'imageUrl': 'mock_img_url',
+            'companyName': '',
+            'title': '',
+            'trackGain': '0.0',
+            'adTrackingTokens': ['mock_token_1', 'mock_token_2']
+    }
+
+    def test_get_ad_item_(self):
+        with patch.object(APIClient, '__call__', return_value=self.mock_ad_metadata_result) as ad_metadata_mock:
+            transport = Mock(side_effect=[InvalidAuthToken(), None])
+
+            client = APIClient(transport, None, None, None, None)
+            client._authenticate = Mock()
+
+            ad_item = client.get_ad_item('mock_id', 'mock_token')
+            assert ad_item.station_id == 'mock_id'
+
+            ad_metadata_mock.assert_has_calls([call("ad.getAdMetadata",
+                                                      adToken='mock_token',
+                                                      returnAdTrackingTokens=True,
+                                                      supportAudioAds=True)])
+
+    def test_get_ad_item_with_no_station_id_specified_raises_exception(self):
+            transport = Mock(side_effect=[InvalidAuthToken(), None])
+
+            client = APIClient(transport, None, None, None, None)
+            client.get_ad_metadata = Mock()
+
+            self.assertRaises(ValueError, client.get_ad_item, '', 'mock_token')