예제 #1
0
class CategoryTest(unittest.TestCase):
    """ category test setup"""
    def setUp(self):
        self.title = Category("lunch")

    def test_addcategory_added(self):
        """ usertest if category is added"""
        self.assertEqual(self.title.add_category("dinner"), True)

    def test_addcategoryname_exists(self):
        """ usertest if category already exists"""
        self.title.add_category("dinner")
        self.assertEqual(self.title.add_category("dinner"), False)

    def test_editcategory_not_found(self):
        """ usertest if user is added"""
        self.assertEqual(self.title.edit_category("drinks", "others"), False)

    def test_editcategory_successful(self):
        """ usertest if category is added"""
        self.title.add_category("Snacks")
        self.assertEqual(self.title.edit_category("dissert", "veggies"), False)

    def test_deletecategory_not_found(self):
        """ usertest if category to delete is not found"""
        self.assertEqual(self.title.delete_category("deleted"), False)

    def test_deletecategory_deleted(self):
        """ usertest if category is deleted"""
        self.title.add_category("breakfast recipes")
        self.assertEqual(self.title.delete_category("breakfast recipes"), True)