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)
