From 2079b16431805bcf818132d89734a8b8c0aa1911 Mon Sep 17 00:00:00 2001
From: Son NK <son@simplelogin.io>
Date: Wed, 18 Mar 2020 19:08:16 +0100
Subject: [PATCH] add "in_trial" to /user_info

---
 README.md                   | 3 ++-
 app/api/views/user_info.py  | 7 ++++++-
 tests/api/test_user_info.py | 7 ++++++-
 3 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index e952051c..67f2a098 100644
--- a/README.md
+++ b/README.md
@@ -634,7 +634,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):