예제 #1
0
    def test_delete(self):
        title = "Vodka based"
        category = Category(title=title)
        category.save()
        self.assertEqual(Category.objects.all().count(), 1)

        url = category.get_absolute_url()
        res = self.app.get(url)

        del_url = category.get_delete_url()
        res = self.app.get(del_url)
        form = res.forms["delete_form"]
        res = form.submit().follow()
        self.assertEqual(Category.objects.all().count(), 0)
예제 #2
0
    def test_delete_with_coctails(self):
        title = "Vodka based"
        category = Category(title=title)
        category.save()

        Coctail(title="Pure vodka", category=category).save()
        Coctail(title="Only vodka", category=category).save()

        self.assertEqual(Category.objects.all().count(), 1)
        self.assertEqual(Coctail.objects.all().count(), 2)

        del_url = category.get_delete_url()
        res = self.app.get(del_url)
        form = res.forms["delete_form"]
        res = form.submit().follow()
        
        self.assertEqual(Category.objects.all().count(), 0)
        self.assertEqual(Coctail.objects.all().count(), 0)