From 7f362385b6707acc3f7752991f48b4004627bbb0 Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Wed, 28 Jun 2017 22:19:01 +0800
Subject: [PATCH] apiserver: add interface /api/token

---
 test/api_server.py        | 11 +++++++++++
 test/test_apiserver.py    |  7 +++++++
 test/test_apiserver_v2.py |  8 ++++++++
 3 files changed, 26 insertions(+)

diff --git a/test/api_server.py b/test/api_server.py
index b0cdd87f..bb2b3690 100644
--- a/test/api_server.py
+++ b/test/api_server.py
@@ -65,6 +65,17 @@ def get_customized_response():
 
     return response
 
+@app.route('/api/token')
+@validate_request
+def get_token():
+    result = {
+        'success': True,
+        'token': utils.gen_random_string(8)
+    }
+    response = make_response(json.dumps(result))
+    response.headers["Content-Type"] = "application/json"
+    return response
+
 @app.route('/api/users')
 @validate_request
 def get_users():
diff --git a/test/test_apiserver.py b/test/test_apiserver.py
index 9daf0b86..29694f30 100644
--- a/test/test_apiserver.py
+++ b/test/test_apiserver.py
@@ -128,3 +128,10 @@ class TestApiServer(ApiServerUnittest):
         resp = self.api_client.post(url, json=expected_response)
         self.assertIn('abc', resp.headers)
         self.assertIn('123', resp.headers['abc'])
+
+    def test_get_token(self):
+        url = "%s/api/token" % self.host
+        resp = self.api_client.get(url)
+        resp_json = resp.json()
+        self.assertTrue(resp_json["success"])
+        self.assertEqual(len(resp_json["token"]), 8)
diff --git a/test/test_apiserver_v2.py b/test/test_apiserver_v2.py
index 57d47be2..c95ab935 100644
--- a/test/test_apiserver_v2.py
+++ b/test/test_apiserver_v2.py
@@ -148,3 +148,11 @@ class TestApiServerV2(ApiServerUnittest):
         )
         self.assertIn('abc', resp.headers)
         self.assertIn('123', resp.headers['abc'])
+
+    def test_get_token(self):
+        url = "%s/api/token" % self.host
+        headers = self.prepare_headers()
+        resp = self.api_client.get(url, headers=headers)
+        resp_json = resp.json()
+        self.assertTrue(resp_json["success"])
+        self.assertEqual(len(resp_json["token"]), 8)