From 5cefe60489e837c29f3d02beb1c3115ac006cba8 Mon Sep 17 00:00:00 2001
From: wangting0128 <ting.wang@zilliz.com>
Date: Mon, 20 Dec 2021 12:09:47 +0800
Subject: [PATCH] [skip e2e] Add new func of get_token on utils

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
---
 tests/benchmark/milvus_benchmark/utils.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/tests/benchmark/milvus_benchmark/utils.py b/tests/benchmark/milvus_benchmark/utils.py
index 6c24d35c4..ca72ac420 100644
--- a/tests/benchmark/milvus_benchmark/utils.py
+++ b/tests/benchmark/milvus_benchmark/utils.py
@@ -3,6 +3,7 @@ import time
 import logging
 import string
 import random
+import requests
 import json
 import os
 from yaml.representer import SafeRepresenter
@@ -263,3 +264,15 @@ def read_json_file(file_name):
     with open(file_name) as f:
         file_dict = json.load(f)
     return file_dict
+
+
+def get_token(url):
+    """ get the request token and return the value """
+    rep = requests.get(url)
+    data = json.loads(rep.text)
+    if 'token' in data:
+        token = data['token']
+    else:
+        token = ''
+        print("Can not get token.")
+    return token
-- 
GitLab