Improve mocking logic

1 files changed, 14 insertions, 1 deletions

diff --git a/opendc/api/v2/users/test_endpoint.py b/opendc/api/v2/users/test_endpoint.py
index ffe2ce02..a5073c08 100644
--- a/opendc/api/v2/users/test_endpoint.py
+++ b/opendc/api/v2/users/test_endpoint.py
@@ -1,2 +1,15 @@
+from opendc.util.database import DB
+
+
+def test_get_user(client, mocker):
+    mocker.patch.object(DB, 'fetch_one', return_value={'email': 'test@test.com'})
+    assert '200' in client.get('/api/v2/users?email=test@test.com').status
+
+
+def test_get_user_non_existing(client, mocker):
+    mocker.patch.object(DB, 'fetch_one', return_value=None)
+    assert '404' in client.get('/api/v2/users?email=test@test.com').status
+
+
 def test_get_user_missing_parameter(client):
-    print(client.get('/api/v2/users'))
+    assert '400' in client.get('/api/v2/users').status