diff --git a/README.md b/README.md index e12f8cb1..055612e4 100644 --- a/README.md +++ b/README.md @@ -651,7 +651,8 @@ Output: if api key is correct, return a json with user name and whether user is { "name": "John Wick", "is_premium": false, - "email": "john@wick.com" + "email": "john@wick.com", + "in_trial": true } ``` diff --git a/app/api/views/user_info.py b/app/api/views/user_info.py index 318b418a..e53ab6f4 100644 --- a/app/api/views/user_info.py +++ b/app/api/views/user_info.py @@ -14,5 +14,10 @@ def user_info(): user = g.user return jsonify( - {"name": user.name, "is_premium": user.is_premium(), "email": user.email} + { + "name": user.name, + "is_premium": user.is_premium(), + "email": user.email, + "in_trial": user.in_trial(), + } ) diff --git a/tests/api/test_user_info.py b/tests/api/test_user_info.py index 5c4c404f..2ebcbd72 100644 --- a/tests/api/test_user_info.py +++ b/tests/api/test_user_info.py @@ -19,7 +19,12 @@ def test_user_in_trial(flask_client): ) assert r.status_code == 200 - assert r.json == {"is_premium": True, "name": "Test User", "email": "a@b.c"} + assert r.json == { + "is_premium": True, + "name": "Test User", + "email": "a@b.c", + "in_trial": True, + } def test_wrong_api_key(flask_client):