python-tdd/apps/dashboard/tests.py

from django.test import TestCase
from .models import Item

class HomePageTest(TestCase):
    def test_uses_home_template(self):
        response = self.client.get('/')
        self.assertTemplateUsed(response, 'apps/dashboard/home.html')

    def test_renders_input_form(self):
        response = self.client.get('/')
        self.assertContains(response, '<form method="POST">')
        self.assertContains(response, '<input name="item-text"')

    def test_can_save_a_POST_request(self):
        response = self.client.post('/', data={'item_text': 'A new dashboard item'})
        self.assertEqual(Item.objects.count(), 1)
        new_item = Item.objects.first()
        self.assertEqual(new_item.text, 'A new dashboard item')
        self.assertContains(response, 'A new dashboard item')
        self.assertTemplateUsed(response, 'apps/dashboard/home.html')


class ItemModelTest(TestCase):
    def test_saving_and_retrieving_items(self):
        first_item = Item()
        first_item.text = "The first (ever) dashboard item"
        first_item.save()

        second_item = Item()
        second_item.text = "A sequel somehow better than the first"
        second_item.save()

        saved_items = Item.objects.all()
        self.assertEqual(saved_items.count(), 2)

        first_saved_item = saved_items[0]
        second_saved_item = saved_items[1]
        self.assertEqual(first_saved_item.text, "The first (ever) dashboard item")
        self.assertEqual(second_saved_item.text, "A sequel somehow better than the first")