aboutsummaryrefslogtreecommitdiff
path: root/tests/test_pandora/test_client.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_pandora/test_client.py')
-rw-r--r--tests/test_pandora/test_client.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/test_pandora/test_client.py b/tests/test_pandora/test_client.py
index f88b755..6c61355 100644
--- a/tests/test_pandora/test_client.py
+++ b/tests/test_pandora/test_client.py
@@ -147,19 +147,19 @@ class TestGettingAds(TestCase):
147class TestCreatingStation(TestCase): 147class TestCreatingStation(TestCase):
148 148
149 def test_using_search_token(self): 149 def test_using_search_token(self):
150 client = APIClient(Mock(), None, None, None, None) 150 client = APIClient(Mock(return_value={}), None, None, None, None)
151 client.create_station(search_token="foo") 151 client.create_station(search_token="foo")
152 client.transport.assert_called_with( 152 client.transport.assert_called_with(
153 "station.createStation", musicToken="foo") 153 "station.createStation", musicToken="foo")
154 154
155 def test_using_artist_token(self): 155 def test_using_artist_token(self):
156 client = APIClient(Mock(), None, None, None, None) 156 client = APIClient(Mock(return_value={}), None, None, None, None)
157 client.create_station(artist_token="foo") 157 client.create_station(artist_token="foo")
158 client.transport.assert_called_with( 158 client.transport.assert_called_with(
159 "station.createStation", trackToken="foo", musicType="artist") 159 "station.createStation", trackToken="foo", musicType="artist")
160 160
161 def test_using_track_token(self): 161 def test_using_track_token(self):
162 client = APIClient(Mock(), None, None, None, None) 162 client = APIClient(Mock(return_value={}), None, None, None, None)
163 client.create_station(track_token="foo") 163 client.create_station(track_token="foo")
164 client.transport.assert_called_with( 164 client.transport.assert_called_with(
165 "station.createStation", trackToken="foo", musicType="song") 165 "station.createStation", trackToken="foo", musicType="song")