diff options
Diffstat (limited to '')
| -rw-r--r-- | src/tests/test_api.py | 12 | ||||
| -rw-r--r-- | src/tests/test_user.py | 6 |
2 files changed, 9 insertions, 9 deletions
diff --git a/src/tests/test_api.py b/src/tests/test_api.py index 063f32f..741780e 100644 --- a/src/tests/test_api.py +++ b/src/tests/test_api.py @@ -11,12 +11,12 @@ def test_api_get_all_users(client, app): assert response.is_json assert re.search('username', str(response.data)) -@pytest.mark.parametrize(('username', 'message'), ( - ('', b'{"msg": "Username is required."}'), - ('neo', b'{"msg": "User neo is already registered."}'), - ('neodarz', b'{"msg": "done"}') +@pytest.mark.parametrize(('username', 'code'), ( + ('', 400), + ('neo', 409), + ('neodarz', 201) )) -def test_api_register(client, app, username, message): +def test_api_register(client, app, username, code): response = client.post('/api/user/register', data='{"username": "'+username+'"}') print(response.data) - assert message in response.data + assert code == response.status_code diff --git a/src/tests/test_user.py b/src/tests/test_user.py index f39279e..c6ddbd4 100644 --- a/src/tests/test_user.py +++ b/src/tests/test_user.py @@ -12,10 +12,10 @@ def test_get_all_users(app): def test_register_user(app): with app.app_context(): user = User(app) - assert user.register("neodarz") is None + assert user.register("neodarz")['code'] is 201 def test_register_and_remove_user(app): with app.app_context(): user = User(app) - assert user.register("neodarz") is None - assert user.remove("neodarz") is None + assert user.register("neodarz")['code'] is 201 + assert user.remove("neodarz")['code'] is 200 |