コード例 #1
0
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")
コード例 #2
0
ファイル: app.py プロジェクト: code-sleuth/yummy-recipes
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)