tests.py 2.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  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. def test_redirects_after_POST(self):
  23. response = self.client.post('/', data={'movie-title': 'X-Men'})
  24. self.assertEqual(response.status_code, 302)
  25. self.assertEqual(response['location'], '/libraries/benoit-hubert/')
  26. def test_uses_list_template(self):
  27. response = self.client.get('/libraries/benoit-hubert/')
  28. self.assertTemplateUsed(response, 'library.html')
  29. def test_displays_all_list_items(self):
  30. Movie.objects.create(title='itemey 1')
  31. Movie.objects.create(title='itemey 2')
  32. response = self.client.get('/libraries/benoit-hubert/')
  33. self.assertIn('itemey 1', response.content.decode())
  34. self.assertIn('itemey 2', response.content.decode())
  35. class MovieModelTest(TestCase):
  36. def test_saving_and_retrieving_movies(self):
  37. first_movie = Movie()
  38. first_movie.title = 'The first (ever) movie'
  39. first_movie.save()
  40. second_movie = Movie()
  41. second_movie.title = 'The second movie'
  42. second_movie.save()
  43. saved_movies = Movie.objects.all()
  44. self.assertEqual(saved_movies.count(), 2)
  45. first_saved_movie = saved_movies[0]
  46. second_saved_movie = saved_movies[1]
  47. self.assertEqual(first_saved_movie.title, 'The first (ever) movie')
  48. self.assertEqual(second_saved_movie.title, 'The second movie')