Kaynağa Gözat

make UT pass for Movie creation

Benoît Hubert 7 yıl önce
ebeveyn
işleme
91c1bf1f39
2 değiştirilmiş dosya ile 12 ekleme ve 2 silme
  1. 6 0
      movies/tests.py
  2. 6 2
      movies/views.py

+ 6 - 0
movies/tests.py

@@ -25,7 +25,13 @@ class HomePageTest(TestCase):
 
     def test_can_save_a_POST_request(self):
         response = self.client.post('/', data={'movie-title': 'X-Men'})
+
+        self.assertEqual(Movie.objects.count(), 1)
+        new_movie = Movie.objects.first()
+        self.assertEqual(new_movie.title, 'X-Men')
+
         self.assertIn('X-Men', response.content.decode())
+        self.assertTemplateUsed(response, 'home.html')
 
 class MovieModelTest(TestCase):
 

+ 6 - 2
movies/views.py

@@ -1,8 +1,12 @@
 from django.shortcuts import render
-from django.http import HttpResponse
+from movies.models import Movie
 
 # Create your views here.
 def home_page(request):
+    movie = Movie()
+    movie.title = request.POST.get('movie-title', '')
+    movie.save()
+
     return render(request, 'home.html', {
-        'new_movie_title': request.POST.get('movie-title', ''),
+        'new_movie_title': movie.title
     })