Exemplo n.º 1
0
    def test_book_detail_found(self):
        request = django.http.HttpRequest()
        test_book_id = 1000
        test_book_result = Book(id=test_book_id, title='All I know about Mock', page_length=2)

        # This time, we'll use patch to handle keeping track of the original get_object_or_404.
        with mock.patch('library.views.get_object_or_404') as mock_get_object_or_404:
            mock_get_object_or_404.return_value = test_book_result
            response = views.book_detail(request, test_book_id)
            mock_get_object_or_404.assert_called_with(Book, id=test_book_id)

        assert response.content.startswith('<h1>')
        assert response.content.endswith('</br>')
Exemplo n.º 2
0
    def test_book_detail_not_found(self):
        request = django.http.HttpRequest()
        test_book_id = 1

        # Keep a handle on the original get_object_or_404
        old_get_object_or_404 = views.get_object_or_404

        # Replace get_object_or_404 with a mock object.
        mock_get_object_or_404 = mock.Mock(side_effect=django.http.Http404('book not found'))
        views.get_object_or_404 = mock_get_object_or_404

        with pytest.raises(django.http.Http404):
            response = views.book_detail(request, test_book_id)

        # Reset in case we want to use the real function again.
        views.get_object_or_404 = old_get_object_or_404