Add PUT method and tests

1 files changed, 9 insertions, 5 deletions

diff --git a/opendc/api/v2/users/test_endpoint.py b/opendc/api/v2/users/test_endpoint.py
index dd6680a8..d60429b3 100644
--- a/opendc/api/v2/users/test_endpoint.py
+++ b/opendc/api/v2/users/test_endpoint.py
@@ -1,18 +1,20 @@
 from opendc.util.database import DB
 
 
-def test_get_user_missing_parameter(client):
+def test_get_user_by_email_missing_parameter(client):
     assert '400' in client.get('/api/v2/users').status
 
 
-def test_get_user_non_existing(client, mocker):
+def test_get_user_by_email_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(client, mocker):
+def test_get_user_by_email(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
+    res = client.get('/api/v2/users?email=test@test.com')
+    assert 'email' in res.json['content']
+    assert '200' in res.status
 
 
 def test_add_user_missing_parameter(client):
@@ -27,4 +29,6 @@ def test_add_user_existing(client, mocker):
 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
+    res = client.post('/api/v2/users', json={'user': {'email': 'test@test.com'}})
+    assert 'email' in res.json['content']
+    assert '200' in res.status