Test reset: clean reset function.

master
Tom Hacohen 4 years ago
parent 151bec0d9e
commit fa0979dce1

@ -2,31 +2,33 @@ from django.conf import settings
from django.contrib.auth import get_user_model from django.contrib.auth import get_user_model
from django.db import transaction from django.db import transaction
from django.shortcuts import get_object_or_404 from django.shortcuts import get_object_or_404
from fastapi import APIRouter, Request, Response, status from fastapi import APIRouter, Request, status
from django_etebase.utils import get_user_queryset, CallbackContext from django_etebase.utils import get_user_queryset, CallbackContext
from etebase_fastapi.authentication import SignupIn, signup_save from etebase_fastapi.authentication import SignupIn, signup_save
from etebase_fastapi.msgpack import MsgpackRoute from etebase_fastapi.msgpack import MsgpackRoute
from etebase_fastapi.exceptions import HttpError
test_reset_view_router = APIRouter(route_class=MsgpackRoute, tags=["test helpers"]) test_reset_view_router = APIRouter(route_class=MsgpackRoute, tags=["test helpers"])
User = get_user_model() User = get_user_model()
@test_reset_view_router.post("/reset/") @test_reset_view_router.post("/reset/", status_code=status.HTTP_204_NO_CONTENT)
def reset(data: SignupIn, request: Request): def reset(data: SignupIn, request: Request):
# Only run when in DEBUG mode! It's only used for tests # Only run when in DEBUG mode! It's only used for tests
if not settings.DEBUG: if not settings.DEBUG:
return Response("Only allowed in debug mode.", status_code=status.HTTP_400_BAD_REQUEST) raise HttpError(code="generic", detail="Only allowed in debug mode.")
with transaction.atomic(): with transaction.atomic():
user_queryset = get_user_queryset(User.objects.all(), CallbackContext(request.path_params)) user_queryset = get_user_queryset(User.objects.all(), CallbackContext(request.path_params))
user = get_object_or_404(user_queryset, username=data.user.username) user = get_object_or_404(user_queryset, username=data.user.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 Response("Endpoint not allowed for user.", status_code=status.HTTP_400_BAD_REQUEST) raise HttpError(code="generic", detail="Endpoint not allowed for user.")
if hasattr(user, "userinfo"): if hasattr(user, "userinfo"):
user.userinfo.delete() user.userinfo.delete()
signup_save(data, request) signup_save(data, request)
# 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()
@ -34,5 +36,3 @@ def reset(data: SignupIn, request: Request):
user.incoming_invitations.all().delete() user.incoming_invitations.all().delete()
# FIXME: also delete chunk files!!! # FIXME: also delete chunk files!!!
return Response(status_code=status.HTTP_204_NO_CONTENT)

Loading…
Cancel
Save