Add more tests

2 files changed, 20 insertions, 8 deletions

diff --git a/opendc/api/v2/users/endpoint.py b/opendc/api/v2/users/endpoint.py
index 7b4158e3..6a99d2b3 100644
--- a/opendc/api/v2/users/endpoint.py
+++ b/opendc/api/v2/users/endpoint.py
@@ -1,6 +1,3 @@
-from werkzeug.exceptions import abort
-
-from opendc.models.user import User
 from opendc.util import exceptions
 from opendc.util.database import DB
 from opendc.util.rest import Response
diff --git a/opendc/api/v2/users/test_endpoint.py b/opendc/api/v2/users/test_endpoint.py
index a5073c08..dd6680a8 100644
--- a/opendc/api/v2/users/test_endpoint.py
+++ b/opendc/api/v2/users/test_endpoint.py
@@ -1,9 +1,8 @@
 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_missing_parameter(client):
+    assert '400' in client.get('/api/v2/users').status
 
 
 def test_get_user_non_existing(client, mocker):
@@ -11,5 +10,21 @@ def test_get_user_non_existing(client, mocker):
     assert '404' in client.get('/api/v2/users?email=test@test.com').status
 
 
-def test_get_user_missing_parameter(client):
-    assert '400' in client.get('/api/v2/users').status
+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_add_user_missing_parameter(client):
+    assert '400' in client.post('/api/v2/users').status
+
+
+def test_add_user_existing(client, mocker):
+    mocker.patch.object(DB, 'fetch_one', return_value={'email': 'test@test.com'})
+    assert '409' in client.post('/api/v2/users', json={'user': {'email': 'test@test.com'}}).status
+
+
+def test_add_user(client, mocker):
+    mocker.patch.object(DB, 'fetch_one', return_value=None)
+    mocker.patch.object(DB, 'insert', return_value={'email': 'test@test.com'})
+    assert '200' in client.post('/api/v2/users', json={'user': {'email': 'test@test.com'}}).status