diff options
author | Mike Crute <mike@crute.us> | 2018-09-01 04:03:32 +0000 |
---|---|---|
committer | Mike Crute <mike@crute.us> | 2018-09-01 04:20:05 +0000 |
commit | 18b843f6fee09298ccfa3a85198f9194011b732f (patch) | |
tree | 01a80c4d157da0e015e919a9a6b7e71bafcd2251 /tests | |
parent | 3de319435697e65a3b5d8570adc3ddaf88ee857f (diff) | |
download | pydora-18b843f6fee09298ccfa3a85198f9194011b732f.tar.bz2 pydora-18b843f6fee09298ccfa3a85198f9194011b732f.tar.xz pydora-18b843f6fee09298ccfa3a85198f9194011b732f.zip |
create_station returns a Station model
related: #57
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_pandora/test_client.py | 6 |
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): | |||
147 | class TestCreatingStation(TestCase): | 147 | class 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") |