From db979d36d0b9693cc81ffa0bdd29364c6218fc95 Mon Sep 17 00:00:00 2001 From: Georgios Andreadis Date: Wed, 24 Jun 2020 13:26:56 +0200 Subject: Add working test setup --- opendc/api/v2/users/endpoint.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'opendc/api/v2/users/endpoint.py') diff --git a/opendc/api/v2/users/endpoint.py b/opendc/api/v2/users/endpoint.py index 2712c625..dca509ed 100644 --- a/opendc/api/v2/users/endpoint.py +++ b/opendc/api/v2/users/endpoint.py @@ -2,7 +2,7 @@ from werkzeug.exceptions import abort from opendc.models.user import User from opendc.util import exceptions -from opendc.util.database import fetch_one, insert +from opendc.util.database import DB from opendc.util.rest import Response @@ -14,7 +14,7 @@ def GET(request): except exceptions.ParameterError as e: return Response(400, str(e)) - user = fetch_one({'email': request.params_query['email']}, 'users') + user = DB.fetch_one({'email': request.params_query['email']}, 'users') if user is not None: return Response(404, f'User with email {request.params_query["email"]} not found') @@ -32,7 +32,7 @@ def POST(request): request.params_body['user']['googleId'] = request.google_id user = request.params_body['user'] - existing_user = fetch_one({'googleId': user['googleId']}, 'users') + existing_user = DB.fetch_one({'googleId': user['googleId']}, 'users') if existing_user is not None: return Response(409, '{} already exists.'.format(existing_user)) -- cgit v1.2.3