Convert test_forms to pytest

Author: shepherdjay

lists/tests/test_forms.py

@@ -1,49 +1,51 @@
-from django.test import TestCase
+import pytest
 from lists.forms import ItemForm, EMPTY_ITEM_ERROR, DUPLICATE_ITEM_ERROR, ExistingListItemForm
 from lists.models import Item, List
 
+pytestmark = pytest.mark.django_db
 
-class ItemFormTest(TestCase):
+
+class TestItemForm:
     def test_form_renders_item_text_input(self):
         form = ItemForm()
-        self.assertIn('placeholder="Enter a to-do item"', form.as_p())
-        self.assertIn('class="form-control input-lg"', form.as_p())
+        assert 'placeholder="Enter a to-do item"' in form.as_p()
+        assert 'class="form-control input-lg"' in form.as_p()
 
     def test_form_validation_for_blank_items(self):
         form = ItemForm(data={'text': ''})
-        self.assertFalse(form.is_valid())
-        self.assertEqual(form.errors['text'], [EMPTY_ITEM_ERROR])
+        assert not form.is_valid()
+        assert form.errors['text'] == [EMPTY_ITEM_ERROR]
 
     def test_form_save_handles_saving_to_a_list(self):
         list_ = List.objects.create()
         form = ItemForm(data={'text': 'do me'})
         new_item = form.save(for_list=list_)
-        self.assertEqual(new_item, Item.objects.first())
-        self.assertEqual(new_item.text, 'do me')
-        self.assertEqual(new_item.list, list_)
+        assert new_item == Item.objects.first()
+        assert new_item.text == 'do me'
+        assert new_item.list == list_
 
 
-class ExistingListItemFormTest(TestCase):
+class TestExistingListItemForm:
     def test_form_renders_item_text_input(self):
         list_ = List.objects.create()
         form = ExistingListItemForm(for_list=list_)
-        self.assertIn('placeholder="Enter a to-do item"', form.as_p())
+        assert 'placeholder="Enter a to-do item"' in form.as_p()
 
     def test_form_validation_for_blank_items(self):
         list_ = List.objects.create()
         form = ExistingListItemForm(for_list=list_, data={'text': ''})
-        self.assertFalse(form.is_valid())
-        self.assertEqual(form.errors['text'], [EMPTY_ITEM_ERROR])
+        assert not form.is_valid()
+        assert form.errors['text'] == [EMPTY_ITEM_ERROR]
 
     def test_form_validation_for_duplicate_items(self):
         list_ = List.objects.create()
         Item.objects.create(list=list_, text='no twins!')
         form = ExistingListItemForm(for_list=list_, data={'text': 'no twins!'})
-        self.assertFalse(form.is_valid())
-        self.assertEqual(form.errors['text'], [DUPLICATE_ITEM_ERROR])
+        assert not form.is_valid()
+        assert form.errors['text'] == [DUPLICATE_ITEM_ERROR]
 
-    def test_firm_save(self):
+    def test_form_save(self):
         list_ = List.objects.create()
         form = ExistingListItemForm(for_list=list_, data={'text': 'hi'})
         new_item = form.save()
-        self.assertEqual(new_item, Item.objects.all()[0])
+        assert new_item == Item.objects.all()[0]