Merge branch 'master' of http://git.hikcreate.com/TestAuto/InterfaceAutoTest

common/tools.py

@@ -3,15 +3,15 @@
 import json
 import logging
 import requests
-from config import BaseConfig, BMCConfig, BmyConfig
+from config import *
 
 
 def request_main(url, headers, method, data):
     """封装requests的通用请求方法"""
     res = None
-    if headers == None or headers == {}:
-        # 如果传的headers为空，使用通用headers
-        headers = BaseConfig.headers
+    if headers == None or headers == {} or headers == "":
+        # 如果传的headers为空，使用各自产品的通用headers
+        headers = get_headers()
     header_content_type = headers["Content-Type"]
 
     try:
@@ -31,6 +31,18 @@ def request_main(url, headers, method, data):
     return res
 
 
+def get_headers():
+    name = BaseConfig.current_name
+    headers = BaseConfig.headers
+    if name == BMCConfig.name:
+        headers = BMCConfig.headers
+    elif name == BmyConfig.name:
+        headers = BmyConfig.headers
+    # elif name == SsoConfig.name:
+    #     headers = SsoConfig.headers
+    return headers
+
+
 def get_case_dir(product_name):
     """根据传入的产品名来运行对应产品的测试用例目录"""
     test_case_dir = BaseConfig.name

config.py

@@ -6,7 +6,7 @@ class BaseConfig():
     # 请求头
     headers = {'Content-Type': 'application/json'}
     # 当前运行的产品名
-    name = "auto"
+    current_name = "auto"
     test_case_dir = "test_case/"
 
     secs=0.1                    # 测试用例间隔运行时间

test_case/sso/conftest.py

@@ -5,6 +5,7 @@ from config import SSOConfig
 from common.utils.encryption import Encryption
 from common.tools import request_main
 
+
 @pytest.fixture(scope='module', autouse=True)
 def sso_login():
     """SSO登录获取token"""
@@ -12,8 +13,3 @@ def sso_login():
     req_data = {"loginName": SSOConfig.sso_username, "password": encrypted_password}
     res = request_main(SSOConfig.sso_url, headers=None, method='post', data=req_data)
     setattr(SSOConfig, 'sso_token', res['data']['token'])
-
-
-
-
-