diff --git a/tests/integration/test_user_endpoints.py b/tests/integration/test_user_endpoints.py index 0b5c2eb3..09b57b35 100644 --- a/tests/integration/test_user_endpoints.py +++ b/tests/integration/test_user_endpoints.py @@ -359,6 +359,22 @@ def test_categories(self): response = self.spotify.categories() self.assertGreater(len(response['categories']), 0) + def test_categories_country(self): + response = self.spotify.categories(country='US') + self.assertGreater(len(response['categories']), 0) + + def test_categories_locale(self): + response = self.spotify.categories(locale='en_US') + self.assertGreater(len(response['categories']), 0) + + def test_categories_limit_low(self): + response = self.spotify.categories(limit=1) + self.assertEqual(len(response['categories']), 1) + + def test_categories_limit_high(self): + response = self.spotify.categories(limit=50) + self.assertLessEqual(len(response['categories']), 50) + def test_category_playlists(self): response = self.spotify.categories() category = 'rock' @@ -368,6 +384,25 @@ def test_category_playlists(self): response = self.spotify.category_playlists(category_id=cat_id) self.assertGreater(len(response['playlists']["items"]), 0) + def test_category_playlists_limit_low(self): + response = self.spotify.categories() + category = 'rock' + for cat in response['categories']['items']: + cat_id = cat['id'] + if cat_id == category: + response = self.spotify.category_playlists(category_id=cat_id, limit=1) + self.assertEqual(len(response['categories']['items']), 1) + + def test_category_playlists_limit_high(self): + response = self.spotify.categories() + category = 'rock' + for cat in response['categories']['items']: + cat_id = cat['id'] + if cat_id == category: + response = self.spotify.category_playlists(category_id=cat_id, limit=50) + self.assertLessEqual(len(response['categories']['items']), 50) + + def test_new_releases(self): response = self.spotify.new_releases() self.assertGreater(len(response['albums']), 0)