tests.py 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. from django.urls import resolve
  2. from django.test import TestCase
  3. from django.http import HttpRequest
  4. from movies.views import home_page
  5. from movies.models import Movie
  6. class HomePageTest(TestCase):
  7. def test_root_url_resolves_to_home_page_view(self):
  8. found = resolve('/')
  9. self.assertEqual(found.func, home_page)
  10. def test_home_page_returns_correct_html(self):
  11. response = self.client.get('/')
  12. html = response.content.decode('utf8')
  13. self.assertTrue(html.startswith('<html>'))
  14. self.assertIn('<title>MovieLib</title>', html)
  15. self.assertTrue(html.strip().endswith('</html>'))
  16. self.assertTemplateUsed(response, 'home.html')
  17. def test_can_save_a_POST_request(self):
  18. response = self.client.post('/', data={'movie-title': 'X-Men'})
  19. self.assertIn('X-Men', response.content.decode())
  20. class MovieModelTest(TestCase):
  21. def test_saving_and_retrieving_movies(self):
  22. first_movie = Movie()
  23. first_movie.title = 'The first (ever) movie'
  24. first_movie.save()
  25. second_movie = Movie()
  26. second_movie.title = 'The second movie'
  27. second_movie.save()
  28. saved_movies = Movie.objects.all()
  29. self.assertEqual(saved_movies.count(), 2)
  30. first_saved_movie = saved_movies[0]
  31. second_saved_movie = saved_movies[1]
  32. self.assertEqual(first_saved_movie.title, 'The first (ever) movie')
  33. self.assertEqual(second_saved_movie.title, 'The second movie')