diff options
Diffstat (limited to 'tests/test_pydora/test_utils.py')
| -rw-r--r-- | tests/test_pydora/test_utils.py | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/tests/test_pydora/test_utils.py b/tests/test_pydora/test_utils.py index 475a7b5..fc33a40 100644 --- a/tests/test_pydora/test_utils.py +++ b/tests/test_pydora/test_utils.py | |||
| @@ -40,12 +40,12 @@ class TestIterateForever(TestCase): | |||
| 40 | side_effect=ParameterMissing("ParameterMissing")) | 40 | side_effect=ParameterMissing("ParameterMissing")) |
| 41 | 41 | ||
| 42 | with plmock as get_playlist_mock: | 42 | with plmock as get_playlist_mock: |
| 43 | with self.assertRaises(ParameterMissing): | 43 | with self.assertRaises(ParameterMissing): |
| 44 | station = Station.from_json( | 44 | station = Station.from_json( |
| 45 | self.client, {'stationToken': 'token_mock'}) | 45 | self.client, {'stationToken': 'token_mock'}) |
| 46 | track_mock = PlaylistItem.from_json( | 46 | track_mock = PlaylistItem.from_json( |
| 47 | self.client, {'token': 'token_mock'}) | 47 | self.client, {'token': 'token_mock'}) |
| 48 | get_playlist_mock.return_value = iter([track_mock]) | 48 | get_playlist_mock.return_value = iter([track_mock]) |
| 49 | 49 | ||
| 50 | station_iter = iterate_forever(station.get_playlist) | 50 | station_iter = iterate_forever(station.get_playlist) |
| 51 | next(station_iter) | 51 | next(station_iter) |