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 delete_category(): if request.method == "POST": msg = '' _name = request.form["item"] _del = request.form['deletecategory'] global CATEGORY _obj_cat = Category() _obj_cat.set_categories(CATEGORY) if _name and _del: if _obj_cat.delete_category(_name): db = _obj_cat.get_all_categories() CATEGORY = db return render_template('dashboard.html', msg=msg, category_list=CATEGORY, info=USERS, rec=RECIPES, det=db_logged_in_user) else: 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)