Use status code instead of simple message for testingHEADmaster

1 files changed, 6 insertions, 6 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