Align with using Pandora's ParameterMissing exception for ValueErrors.

2 files changed, 4 insertions, 3 deletions

diff --git a/tests/test_pandora/test_client.py b/tests/test_pandora/test_client.py
index 24d9734..69fb632 100644
--- a/tests/test_pandora/test_client.py
+++ b/tests/test_pandora/test_client.py
@@ -1,7 +1,7 @@
 from unittest import TestCase
 
 from pandora.client import APIClient, BaseAPIClient
-from pandora.errors import InvalidAuthToken
+from pandora.errors import InvalidAuthToken, ParameterMissing
 from pandora.py2compat import Mock, call, patch
 from tests.test_pandora.test_models import TestAdItem
 
@@ -117,4 +117,4 @@ class TestGettingAds(TestCase):
         client = APIClient(transport, None, None, None, None)
         client.get_ad_metadata = Mock()
 
-        self.assertRaises(ValueError, client.get_ad_item, '', 'mock_token')
+        self.assertRaises(ParameterMissing, client.get_ad_item, '', 'mock_token')
diff --git a/tests/test_pandora/test_models.py b/tests/test_pandora/test_models.py
index 06c988c..538cee5 100644
--- a/tests/test_pandora/test_models.py
+++ b/tests/test_pandora/test_models.py
@@ -3,6 +3,7 @@ from datetime import datetime
 from pandora.py2compat import Mock, patch
 from pandora import APIClient
 from pandora.models.pandora import AdItem, PlaylistModel
+from pandora.errors import ParameterMissing
 
 import pandora.models as m
 
@@ -235,7 +236,7 @@ class TestAdItem(TestCase):
         assert self.result._api_client.register_ad.called
 
     def test_register_ad_raises_exception_if_no_tracking_tokens_available(self):
-        with self.assertRaises(ValueError):
+        with self.assertRaises(ParameterMissing):
             self.result.tracking_tokens = []
             self.result._api_client.register_ad = Mock(spec=AdItem)