Debug reset: put the whole request in a transaction.

master
Tom Hacohen 4 years ago
parent cc163d27af
commit ee23707fff

@ -778,23 +778,24 @@ class TestAuthenticationViewSet(viewsets.ViewSet):
if not settings.DEBUG: if not settings.DEBUG:
return HttpResponseBadRequest("Only allowed in debug mode.") return HttpResponseBadRequest("Only allowed in debug mode.")
user = get_object_or_404(User.objects.all(), username=request.data.get('user').get('username')) with transaction.atomic():
user = get_object_or_404(User.objects.all(), username=request.data.get('user').get('username'))
# Only allow test users for extra safety # Only allow test users for extra safety
if not getattr(user, User.USERNAME_FIELD).startswith('test_user'): if not getattr(user, User.USERNAME_FIELD).startswith('test_user'):
return HttpResponseBadRequest("Endpoint not allowed for user.") return HttpResponseBadRequest("Endpoint not allowed for user.")
if hasattr(user, 'userinfo'): if hasattr(user, 'userinfo'):
user.userinfo.delete() user.userinfo.delete()
serializer = AuthenticationSignupSerializer(data=request.data) serializer = AuthenticationSignupSerializer(data=request.data)
serializer.is_valid(raise_exception=True) serializer.is_valid(raise_exception=True)
serializer.save() serializer.save()
# Delete all of the journal data for this user for a clear test env # Delete all of the journal data for this user for a clear test env
user.collection_set.all().delete() user.collection_set.all().delete()
user.incoming_invitations.all().delete() user.incoming_invitations.all().delete()
# FIXME: also delete chunk files!!! # FIXME: also delete chunk files!!!
return HttpResponse() return HttpResponse()

Loading…
Cancel
Save