# encoding: utf-8
 
import requests
import urllib3
# from urllib3.exceptions import InsecureRequestWarning
 
urllib3.disable_warnings()
# 加這句不會(huì)報(bào)錯(cuò)(requests證書警告)
# requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
 
 
class HTTPRequests(object):
    def __init__(self, url):
        self.url = url
        self.req = requests.session()
        # 依據(jù)自己公司的請(qǐng)求頭默認(rèn)值配置
        self.head = {
            'User-Agent': 'Mozilla/5.0 (Windows NT 6.2; WOW64; Trident/7.0; rv:11.0) like Gecko',
            'Accept': 'image/gif, image/jpeg, image/pjpeg, application/x-ms-application, application/xaml+xml, '
                      'application/x-ms-xbap, application/vnd.ms-excel, application/vnd.ms-powerpoint, '
                      'application/msword, */*',
            'Accept-Language': 'zh-CN'}
 
    # 封裝自己的get請(qǐng)求，獲取資源
    def get(self, uri='', params='', data='', headers=None, cookies=None, verify=False):
        if headers is None:
            headers = self.head
        # print("請(qǐng)求頭是:{}".format(headers))
        url = self.url + uri
        response = self.req.get(url, params=params, data=data, headers=headers, cookies=cookies, verify=verify)
        return response
 
    # 封裝自己的post請(qǐng)求，獲取資源
    def post(self, uri='', params='', data='', headers=None, cookies=None, verify=False):
        if headers is None:
            headers = self.head
        url = self.url + uri
        response = self.req.post(url, params=params, data=data, headers=headers, cookies=cookies, verify=verify)
        return response
 
    # 封裝自己的put請(qǐng)求，獲取資源
    def put(self, uri='', params='', data='', headers=None, cookies=None, verify=False):
        if headers is None:
            headers = self.head
        url = self.url + uri
        response = self.req.put(url, params=params, data=data, headers=headers, cookies=cookies, verify=verify)
        return response
 
    # 封裝自己的delete請(qǐng)求，獲取資源
    def delete(self, uri='', params='', data='', headers=None, cookies=None, verify=False):
        if headers is None:
            headers = self.head
        url = self.url + uri
        response = self.req.delete(url, params=params, data=data, headers=headers, cookies=cookies, verify=verify)
        return response

# encoding: utf-8
 
import enum
 
class URLConf(enum.Enum):
    """環(huán)境配置信息"""
    url_mapping = {
        'dev': 'https://www.dev.com',
        'test': 'https://www.test.com',
        'prod': 'https://www.prod.com'
    }
 
    # token固定的用戶名密碼，固定用"/"分割用戶名和密碼
    email_user = {
        'dev': 'dev@qq.com',
        'test': 'zidonghua@qq.com/96e79218965eb72c92a549dd5a330112',
        'prod': 'prod@qq.com'
    }
 
    login_uri = r'/api/auth/login/account/v1'

# encoding: utf-8
# 代碼來源：全棧測(cè)開訓(xùn)練營(yíng)
 
import logging
import os
 
import pytest
 
from common.http_request import HTTPRequests
from config.url_config import URLConf
 
 
datadir = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "test_data")
logger = logging.getLogger('conftest日志')
 
 
def pytest_addoption(parser):
    # choices 只允許輸入的值的范圍
    parser.addoption(
        "--env", action="store", default='test', choices=['dev', 'test', 'prod'], help="set env"
    )
 
 
# 獲取命令行參數(shù)的fixture
@pytest.fixture(scope='session')
def get_env(request):
    # print("fixutre..................")
    return request.config.getoption('--env')
 
# 聲明一個(gè)返回http請(qǐng)求對(duì)象的fixture，所有用例在一個(gè)session中
# @pytest.fixture(scope='module', autouse=True)
@pytest.fixture(autouse=True)
def http(request):
    env = request.getfixturevalue("get_env")
    url_mapping = URLConf.url_mapping.value
    url = url_mapping.get(f'{env}')
    http = HTTPRequests(url)
 
    return http
 
 
@pytest.fixture(scope='session')
def get_token_head(request):
    env = request.getfixturevalue("get_env")
    url_mapping = URLConf.url_mapping.value
    url = url_mapping.get(f'{env}')
    http = HTTPRequests(url)
 
    user = URLConf.email_user.value
    user_list = user.get(f'{env}').split("/")
    username = user_list[0]
    password = user_list[1]
 
    param = {'clientType': 2,
             'language': 'en',
             'loginId': username,
             'loginPassword': password}
 
    logger.info("請(qǐng)求的url=={}".format(url))
    response = http.post(uri=r'/api/auth/login/account/v1', data=param)
    logger.info("獲取的返回值是:".format(response.text))
    token = None
    if response.status_code == 200:
        token = response.json().get('result')['token']
    else:
        token = 'get token fail'
 
    head = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 6.2; WOW64; Trident/7.0; rv:11.0) like Gecko',
        'Accept': 'image/gif, image/jpeg, image/pjpeg, application/x-ms-application, application/xaml+xml, '
                  'application/x-ms-xbap, application/vnd.ms-excel, application/vnd.ms-powerpoint, '
                  'application/msword, */*',
        'Accept-Language': 'zh-CN',
        'Authorization': token}
 
    yield head

# 創(chuàng)建解析excel的方法
import logging
 
from openpyxl import load_workbook
 
logger = logging.getLogger("讀取excel")
 
 
class ParseExcel(object):
    def __init__(self, excelPath, sheetName):
        self.wb = load_workbook(excelPath)
        self.sheet = self.wb[sheetName]
        self.maxRowNum = self.sheet.max_row
 
 
    # 依據(jù)傳入的數(shù)字，決定獲取幾列excel數(shù)據(jù)
    def getDataFromSheet(self, num):
        dataList = []
        for line in self.sheet.rows:
            tmplist = []
            for i in range(num):
                tmplist.append(line[i].value)
            dataList.append(tmplist)
            print("dadddddd:{}".format(dataList))
 
        return dataList[2:]

# encoding: utf-8
 
"""
 
Account Api模塊
"""
import logging
import os
 
import allure
import pytest
 
from common.get_data_url import get_data_url
from common.parse_excel import ParseExcel
 
datadir = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "test_data")
data_url = get_data_url()
# 獲取到test_data\test的目錄，如果是prod環(huán)境，那么就是獲取test_data\prod目錄
authBaseDir = os.path.join(datadir, data_url)
 
logger = logging.getLogger("Account Api模塊日志")
 
 
@allure.feature("AccountApi模塊")
@pytest.mark.webtest
class TestAccountApi(object):
    """Query Related Achievements: /api/auth/account/achievement/related/query/v1
 
    """
    Query_Related_Achievements_dir = os.path.join(authBaseDir, 'Query_Related_Achievements.xlsx')
    logger.info("Query_Related_Achievements測(cè)試數(shù)據(jù)的路徑是:{}".format(Query_Related_Achievements_dir))
    parse = ParseExcel(Query_Related_Achievements_dir, 'Sheet1')
    Query_Related_Achievements_params = parse.getDataFromSheet(5)
 
    @allure.story("Query Related Achievements(查詢用戶成就信息)")
    @pytest.mark.parametrize("clientType,language,retCode,istoken,result", Query_Related_Achievements_params)
    def test_001_Query_Related_Achievements(self, get_token_head, http, clientType, language, retCode, istoken, result):
        uri = '/api/auth/account/achievement/related/query/v1'
        params = {"clientType": clientType, "language": language}
 
        if istoken == 'yes':
            header = get_token_head
            response = http.get(uri=uri, params=params, headers=header)
            json_req = response.json()
            logger.info("Query_Related_Achievements有token的返回值是:{}".format(json_req))
 
            assert json_req.get('retCode') == 200
            assert json_req.get('result')[0]['smallImg'] == result
        else:
            response = http.get(uri=uri, params=params)
            json_req = response.json()
            logger.info("Query_Related_Achievements沒有token的返回值是:{}".format(json_req))
 
            assert json_req.get('retCode') == 401
            assert json_req.get('message') == result

# encoding: utf-8
 
 
"""
所有案例執(zhí)行并生成allure測(cè)試報(bào)告的執(zhí)行策略
"""
 
import os
import sys
import threading
import pytest
 
 
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/../')
from common.report import Report
 
 
project_root = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
report_dir = os.path.join(project_root, 'report')
# 存放測(cè)試結(jié)果的目錄，會(huì)生成一推json文件
result_dir = os.path.join(report_dir, 'allure_result')
allure_report = os.path.join(report_dir, 'allure_report')
 
report = Report()
 
#  定義搜索條件，搜索所有以test開頭的用例
tag = 'test'
 
 
def run_pytest():
    # --clean-alluredir
    # pytest.main(['-vv', '-s', '-m', 'webtest', f'--alluredir={result_dir}', '--clean-alluredir'])
    # 執(zhí)行前清除allure_result數(shù)據(jù)，避免生成報(bào)告時(shí)，會(huì)把上次執(zhí)行的數(shù)據(jù)帶進(jìn)去
    pytest.main(['-vv', '-s', '-k', f'{tag}', f'--alluredir={result_dir}', '--clean-alluredir'])
 
 
def general_report():
    # 調(diào)用cmd方法 report.allure，根據(jù)windows或linux環(huán)境判斷
    # 然后執(zhí)行生成報(bào)告的方法generate
    # --clean 覆蓋路徑，將上次的結(jié)果覆蓋掉
    cmd = "{} generate {} -o {} --clean".format(report.allure, result_dir, allure_report)
    # 執(zhí)行命令行命令，并通過read()方法將命令的結(jié)果返回
    print(os.popen(cmd).read())
 
 
if __name__ == '__main__':
    # 創(chuàng)建兩個(gè)線程，分別執(zhí)行兩個(gè)方法
    run = threading.Thread(target=run_pytest)
    gen = threading.Thread(target=general_report)
    run.start()
    # 先執(zhí)行第一個(gè)線程，這個(gè)線程執(zhí)行完才會(huì)執(zhí)行下面的線程和主線程
    run.join()
    gen.start()
    gen.join()