update tests

Author: alexmochu

tests/test_business_category.py

@@ -53,5 +53,13 @@ def test_get_all_categories(self):
         msg = self.category_item_class.category_list
         value = self.category_item_class.get_all_categories()
         self.assertEqual(msg, value)
+
+    def test_owner(self):
+        """ Check for correct event creation """
+        self.category_item_class.category_list = [{'owner': 'alexmochu', 'id': '4873498080', 'category': 'Software'}]
+        user = "alexmochu"
+        msg = self.category_item_class.get_owner(user)
+        self.assertEqual(msg, [{'owner': 'alexmochu', 'id': '4873498080', 'category': 'Software'}])
+
 if __name__ == '__main__':
     unittest.main()

tests/test_views.py

@@ -51,5 +51,16 @@ def test_reset_passoword(self):
         self.assertEqual(result["message"], "Password changed successful")
         self.assertEqual(response.status_code, 200)
 
+    def test_getting_all_categories(self):
+        """
+        Test API can get all Businesses in the system (GET request).
+        """
+        # create a business by making a POST request
+        response = self.app.post('/api/v1/auth/login', data = json.dumps(self.data1) , content_type = 'application/json')
+        self.assertEqual(response.status_code, 200)
+        res = self.app.get('/api/v1/category', content_type='application/json')
+        # get all the business that belong to the test user by making a GET request
+        self.assertEqual(res.status_code, 200)
+
 if __name__ == '__main__':
     unittest.main()