50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
from django.http import response
|
|
from django.test import TestCase
|
|
from django.urls import resolve
|
|
from django.http import HttpRequest
|
|
|
|
from lists.views import home_page
|
|
from lists.models import Item
|
|
|
|
class HomePageTest(TestCase):
|
|
|
|
def test_root_url_resolves_to_home_page_view(self):
|
|
found = resolve('/')
|
|
self.assertEqual(found.func, home_page)
|
|
|
|
def test_uses_home_template(self):
|
|
response = self.client.get('/')
|
|
self.assertTemplateUsed(response, 'home.html')
|
|
|
|
def test_can_save_a_POST_request(self):
|
|
response = self.client.post('/', data={'item_text': 'A new list item'})
|
|
|
|
self.assertEqual(Item.objects.count(), 1)
|
|
new_item = Item.objects.first()
|
|
self.assertEqual(new_item.text, 'A new list item')
|
|
|
|
self.assertEqual(response.status_code, 302)
|
|
self.assertEqual(response['location'], '/')
|
|
|
|
def test_only_saves_items_when_necessary(self):
|
|
self.client.get('/')
|
|
self.assertEqual(Item.objects.count(), 0)
|
|
|
|
class ItemModelTest(TestCase):
|
|
|
|
def test_saving_and_retrieving_items(self):
|
|
first_item = Item() # Create an object
|
|
first_item.text = 'The first (ever) list item' # Assign attributes
|
|
first_item.save() # Calling .save() function
|
|
|
|
second_item = Item()
|
|
second_item.text = 'Item the second'
|
|
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) list item')
|
|
self.assertEqual(second_saved_item.text, 'Item the second') |