Skip to content

Commit a59aa2d

Browse files
authored
Add test covering Update view without queryset attribute (#9528)
1 parent 61e3376 commit a59aa2d

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

tests/test_prefetch_related.py

+14
Original file line numberDiff line numberDiff line change
@@ -56,3 +56,17 @@ def test_prefetch_related_excluding_instance_from_original_queryset(self):
5656
'email': 'tom@example.com'
5757
}
5858
assert response.data == expected
59+
60+
def test_can_update_without_queryset_on_class_view(self):
61+
class UserUpdateWithoutQuerySet(generics.UpdateAPIView):
62+
serializer_class = UserSerializer
63+
64+
def get_object(self):
65+
return User.objects.get(pk=self.kwargs['pk'])
66+
67+
request = factory.patch('/', {'username': 'new'})
68+
response = UserUpdateWithoutQuerySet.as_view()(request, pk=self.user.pk)
69+
assert response.data['id'] == self.user.id
70+
assert response.data['username'] == 'new'
71+
self.user.refresh_from_db()
72+
assert self.user.username == 'new'

0 commit comments

Comments
 (0)