Add ad_token to AdItem definition.

2 files changed, 3 insertions, 0 deletions

diff --git a/tests/test_pandora/test_client.py b/tests/test_pandora/test_client.py
index 58fca74..cf486e0 100644
--- a/tests/test_pandora/test_client.py
+++ b/tests/test_pandora/test_client.py
@@ -105,6 +105,7 @@ class TestGettingAds(TestCase):
 
             ad_item = client.get_ad_item('id_mock', 'token_mock')
             assert ad_item.station_id == 'id_mock'
+            assert ad_item.ad_token == 'token_mock'
 
             ad_metadata_mock.assert_has_calls([call("ad.getAdMetadata",
                                                       adToken='token_mock',
diff --git a/tests/test_pandora/test_models.py b/tests/test_pandora/test_models.py
index 538cee5..495bb8e 100644
--- a/tests/test_pandora/test_models.py
+++ b/tests/test_pandora/test_models.py
@@ -225,6 +225,8 @@ class TestAdItem(TestCase):
         api_client_mock = Mock(spec=APIClient)
         api_client_mock.default_audio_quality = APIClient.HIGH_AUDIO_QUALITY
         self.result = AdItem.from_json(api_client_mock, self.JSON_DATA)
+        self.result.station_id = 'station_id_mock'
+        self.result.ad_token = 'token_mock'
 
     def test_is_ad_is_true(self):
         assert self.result.is_ad is True