1 files changed, 22 insertions, 0 deletions
diff --git a/src/tests/test_api.py b/src/tests/test_api.py
new file mode 100644
index 0000000..063f32f
--- /dev/null
+++ b/src/tests/test_api.py
@@ -0,0 +1,22 @@
+import pytest, re
+
+from flask import session
+from bson.json_util import dumps
+
+from pprint import pprint
+
+def test_api_get_all_users(client, app):
+    response = client.get('/api/users')
+    assert response.status_code == 200
+    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"}')
+))
+def test_api_register(client, app, username, message):
+    response = client.post('/api/user/register', data='{"username": "'+username+'"}')
+    print(response.data)
+    assert message in response.data