class MinimalTestCategory(unittest.TestCase): def setUp(self): self._category = Category("Ice Tea") def test_create_category(self): self.assertIsInstance(self._category, Category, 'Failed to create category object') def test_add_category(self): self.assertEqual(self._category.add_category("Pineapple Punch"), "category added successfully") def test_add_category_failed(self): self._category.increment_categories_list("Foodie Category") self.assertEqual(self._category.add_category("Foodie Category"), "category exists") def test_edit_category(self): self._category.increment_categories_list("Pineapple Punch") self.assertEqual( self._category.edit_category("apple Punch", "Pineapple Punch"), "Category updated") def test_edit_category_failed(self): self._category.increment_categories_list("Pineapple Punch") self.assertEqual( self._category.edit_category("apple Punch", "Pine Punch"), "category doesn't exist") def test_delete_category(self): self._category.increment_categories_list("Fruits Salad") self.assertEqual(self._category.delete_category("Fruits Salad"), "category deleted") def test_delete_category_failed(self): self._category.increment_categories_list("Fruits Salad") self.assertEqual(self._category.delete_category("Mango"), "category does not exist", msg="Category not found") def test_get_category(self): self._category.increment_categories_list("My Category") self.assertEqual(self._category.get_category("My Category"), "My Category") def test_get_category_failed(self): self._category.increment_categories_list("Another One") self.assertEqual(self._category.get_category("XO"), "category does not exist", msg="Category not found")
def add_category(): if request.method == "POST": msg = '' _category = request.form["category"] _category_obj = Category() global db_category_list global db_category_name global CATEGORY if _category in CATEGORY: cms = 'Category already exists. use different name' return render_template('dashboard.html', cms=cms, category_list=CATEGORY, info=USERS, rec=RECIPES, det=db_logged_in_user) if _category: if _category_obj.add_category(_category): db_category_list = _category_obj.get_all_categories() if not CATEGORY: CATEGORY = db_category_list else: CATEGORY.append(db_category_list[0]) return render_template("dashboard.html", category_list=CATEGORY, info=USERS, rec=RECIPES, det=db_logged_in_user) else: msg = "Failed to add category" return render_template("dashboard.html", msg=msg, category_list=CATEGORY, info=USERS, rec=RECIPES, det=db_logged_in_user) else: msg = "Login" return render_template("dashboard.html", msg=msg, category_list=CATEGORY, info=USERS, rec=RECIPES, det=db_logged_in_user)