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)