From a1782de1037dc1f818c45257d978da888e87ff85 Mon Sep 17 00:00:00 2001 From: Georgios Andreadis Date: Wed, 24 Jun 2020 14:22:58 +0200 Subject: Improve mocking logic --- opendc/api/v2/users/test_endpoint.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) (limited to 'opendc/api/v2/users/test_endpoint.py') 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 -- cgit v1.2.3