adding more unit tests

Author: GMCtree

spotify_bot.py

@@ -85,6 +85,9 @@ def search(query, query_type, auth_token):
 def is_empty_query(query):
     return True if query == '' else False
 
+def check_no_results(results):
+    return True if len(results) == 0 else False
+
 # main function to handle all inline queries
 def inlinequery(bot, update):
 
@@ -108,7 +111,7 @@ def inlinequery(bot, update):
                         input_message_content=InputTextMessageContent(response)))
 
     # if there are no results, tell user
-    if len(results) == 0:
+    if check_no_results(results):
         results.append(InlineQueryResultArticle(id=uuid4(),
                 title="No results found",
                 input_message_content=InputTextMessageContent("No results found")))

tests.py

@@ -4,12 +4,20 @@ class TestInlineQueryHandling(unittest.TestCase):
     def setUp(self):
         self.emptyQuery = ''
         self.nonEmptyQuery = 'Coldplay'
+        self.emptyResults = []
+        self.nonEmptyResults = [{"artist": "Coldplay"}]
 
     def test_empty_query(self):
         self.assertTrue(spotify_bot.is_empty_query(self.emptyQuery))
 
     def test_nonempty_query(self):
         self.assertFalse(spotify_bot.is_empty_query(self.nonEmptyQuery))
 
+    def test_empty_results(self):
+        self.assertTrue(spotify_bot.check_no_results(self.emptyResults))
+
+    def test_nonempty_results(self):
+        self.assertFalse(spotify_bot.check_no_results(self.nonEmptyResults))
+
 if __name__ == '__main__':
     unittest.main()