tests.py 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  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.assertEqual(Movie.objects.count(), 1)
  20. new_movie = Movie.objects.first()
  21. self.assertEqual(new_movie.title, 'X-Men')
  22. self.assertIn('X-Men', response.content.decode())
  23. self.assertTemplateUsed(response, 'home.html')
  24. class MovieModelTest(TestCase):
  25. def test_saving_and_retrieving_movies(self):
  26. first_movie = Movie()
  27. first_movie.title = 'The first (ever) movie'
  28. first_movie.save()
  29. second_movie = Movie()
  30. second_movie.title = 'The second movie'
  31. second_movie.save()
  32. saved_movies = Movie.objects.all()
  33. self.assertEqual(saved_movies.count(), 2)
  34. first_saved_movie = saved_movies[0]
  35. second_saved_movie = saved_movies[1]
  36. self.assertEqual(first_saved_movie.title, 'The first (ever) movie')
  37. self.assertEqual(second_saved_movie.title, 'The second movie')