Use status code instead of simple message for testingHEADmaster

1 files changed, 3 insertions, 3 deletions

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