tests.py 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. from django.urls import resolve
  2. from django.test import TestCase
  3. from django.http import HttpRequest
  4. from django.template.loader import render_to_string
  5. from lists.views import home_page
  6. from lists.models import Item
  7. class HomePageTest(TestCase):
  8. def test_root_url_resolves_to_home_page_view(self):
  9. found = resolve('/')
  10. self.assertEqual(found.func, home_page)
  11. def test_uses_home_template(self):
  12. response = self.client.get('/')
  13. # The Manual Way of doing it:
  14. html = response.content.decode('utf8')
  15. self.assertTrue(html.startswith('<html>'))
  16. self.assertIn('<title>To-Do lists</title>', html)
  17. self.assertTrue(html.strip().endswith('</html>'))
  18. # Using Django's Builtin Testing Client:
  19. self.assertTemplateUsed(response, 'home.html')
  20. def test_can_save_POST_request(self):
  21. response = self.client.post('/', data={'item_text': 'A new list item'})
  22. self.assertIn('A new list item', response.content.decode())
  23. self.assertTemplateUsed(response, 'home.html')
  24. class ItemModelTest(TestCase):
  25. def test_save_and_retrieve_items(self):
  26. item1 = Item()
  27. item1.text = 'The first (ever) list item'
  28. item1.save()
  29. item2 = Item()
  30. item2.text = 'Item the 2nd'
  31. item2.save()
  32. saved = Item.objects.all()
  33. self.assertEqual(saved.count(), 2)
  34. first_item = saved[0]
  35. second_item = saved[1]
  36. self.assertEqual(first_item.text, item1.text)
  37. self.assertEqual(second_item.text, item2.text)