mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-22 09:36:49 +03:00
Add ValidationError to except in get_object_or_404 for django 1.11
This commit is contained in:
parent
69b0ac52c2
commit
cdb8a3c3c8
|
@ -3,6 +3,7 @@ Generic views that provide commonly needed behaviour.
|
|||
"""
|
||||
from __future__ import unicode_literals
|
||||
|
||||
from django.core.exceptions import ValidationError
|
||||
from django.db.models.query import QuerySet
|
||||
from django.http import Http404
|
||||
from django.shortcuts import get_object_or_404 as _get_object_or_404
|
||||
|
@ -18,7 +19,7 @@ def get_object_or_404(queryset, *filter_args, **filter_kwargs):
|
|||
"""
|
||||
try:
|
||||
return _get_object_or_404(queryset, *filter_args, **filter_kwargs)
|
||||
except (TypeError, ValueError):
|
||||
except (TypeError, ValueError, ValidationError):
|
||||
raise Http404
|
||||
|
||||
|
||||
|
|
|
@ -2,6 +2,7 @@ from __future__ import unicode_literals
|
|||
|
||||
import pytest
|
||||
from django.db import models
|
||||
from django.http import Http404
|
||||
from django.shortcuts import get_object_or_404
|
||||
from django.test import TestCase
|
||||
from django.utils import six
|
||||
|
@ -10,7 +11,8 @@ from rest_framework import generics, renderers, serializers, status
|
|||
from rest_framework.response import Response
|
||||
from rest_framework.test import APIRequestFactory
|
||||
from tests.models import (
|
||||
BasicModel, ForeignKeySource, ForeignKeyTarget, RESTFrameworkModel
|
||||
BasicModel, ForeignKeySource, ForeignKeyTarget, RESTFrameworkModel,
|
||||
UUIDForeignKeyTarget
|
||||
)
|
||||
|
||||
factory = APIRequestFactory()
|
||||
|
@ -647,3 +649,19 @@ class ApiViewsTests(TestCase):
|
|||
view.delete('test request', 'test arg', test_kwarg='test')
|
||||
assert view.called is True
|
||||
assert view.call_args == data
|
||||
|
||||
|
||||
class GetObjectOr404Tests(TestCase):
|
||||
def setUp(self):
|
||||
super(GetObjectOr404Tests, self).setUp()
|
||||
self.uuid_object = UUIDForeignKeyTarget.objects.create(name='bar')
|
||||
|
||||
def test_get_object_or_404_with_valid_uuid(self):
|
||||
obj = generics.get_object_or_404(
|
||||
UUIDForeignKeyTarget, pk=self.uuid_object.pk
|
||||
)
|
||||
assert obj == self.uuid_object
|
||||
|
||||
def test_get_object_or_404_with_invalid_string_for_uuid(self):
|
||||
with pytest.raises(Http404):
|
||||
generics.get_object_or_404(UUIDForeignKeyTarget, pk='not-a-uuid')
|
||||
|
|
Loading…
Reference in New Issue
Block a user