Add working test setup

3 files changed, 6 insertions, 4 deletions

diff --git a/opendc/api/v2/tiles/tileId/rack/machines/position/endpoint.py b/opendc/api/v2/tiles/tileId/rack/machines/position/endpoint.py
index e769a049..72fd44d5 100644
--- a/opendc/api/v2/tiles/tileId/rack/machines/position/endpoint.py
+++ b/opendc/api/v2/tiles/tileId/rack/machines/position/endpoint.py
@@ -96,7 +96,7 @@ def PUT(request):
         return Response(409, 'Rack position occupied.')
 
     except Exception as e:
-        print e
+        print(e)
         return Response(400, 'Invalid Machine.')
 
     # Return this Machine
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))
diff --git a/opendc/api/v2/users/test_endpoint.py b/opendc/api/v2/users/test_endpoint.py
new file mode 100644
index 00000000..ffe2ce02
--- /dev/null
+++ b/opendc/api/v2/users/test_endpoint.py
@@ -0,0 +1,2 @@
+def test_get_user_missing_parameter(client):
+    print(client.get('/api/v2/users'))