apiserver: add interface /api/token

This commit is contained in:
debugtalk
2017-06-28 22:19:01 +08:00
parent 3d9777ba57
commit 7f362385b6
3 changed files with 26 additions and 0 deletions

View File

@@ -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():

View File

@@ -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)

View File

@@ -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)