From 1b0c426984f4beee2129ef854e2537c18288122c Mon Sep 17 00:00:00 2001
From: mattrost <71509176+mattrost@users.noreply.github.com>
Date: Sat, 18 Jun 2022 15:28:08 -0700
Subject: [PATCH] Addition of Test Cases (#818)

* Added test_categories_country to use optional country selection for categories method.

* Added test_categories_locale for testing locale method of categories.

* Added test_categories_limit_low and test_categories_limit_high to test the limit method of categories for edge cases.

* Rearranged for flow

* Added test_category_playlists_limit_low and test_category_playlists_limit_high for testing the limit method.
---
 tests/integration/test_user_endpoints.py | 35 ++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

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)