"""
@Title   : app多設(shè)備并發(fā)-appium+pytest
@Author  : 檸檬班-小簡(jiǎn)
@Email   : lemonban_simple@qq.com
"""
 
class ManageDevices:
    """
       1、重啟adb服務(wù)。
       2、通過adb devices命令獲取當(dāng)前平臺(tái)中,已連接的設(shè)備個(gè)數(shù)，和設(shè)備uuid.
       3、通過adb -P 5037 -s 設(shè)備uuid shell getprop ro.build.version.release獲取每一個(gè)設(shè)備的版本號(hào)。
       4、將所有已連接設(shè)備的設(shè)備名稱、設(shè)備版本號(hào)存儲(chǔ)在一個(gè)列表當(dāng)中。
       5、通過調(diào)用get_devices_info函數(shù)，即可獲得4中的列表。
    """
 
    def __init__(self):
        self.__devices_info = []
        # 重啟adb服務(wù)
        self.__run_command_and_get_stout("adb kill-server")
        self.__run_command_and_get_stout("adb start-server")
 
    def get_devices_info(self):
        """
        獲取已連接設(shè)備的uuid,和版本號(hào)。
        :return: 所有已連接設(shè)備的uuid,和版本號(hào)。
        """
        self.__get_devices_uuid()
        print(self.__devices_info)
        self.__get_device_platform_vesion()
        return self.__devices_info

def devices_pool(port=4723,system_port=8200):
    """
    設(shè)備啟動(dòng)參數(shù)管理池。含啟動(dòng)參數(shù)和對(duì)應(yīng)的端口號(hào)
    :param port: appium服務(wù)的端口號(hào)。每一個(gè)設(shè)備對(duì)應(yīng)一個(gè)。
    :param system_port: appium服務(wù)指定的本地端口，用來轉(zhuǎn)發(fā)數(shù)據(jù)給安卓設(shè)備。每一個(gè)設(shè)備對(duì)應(yīng)一個(gè)。
    :return: 所有已連接設(shè)備的啟動(dòng)參數(shù)和appium端口號(hào)。
    """
    desired_template = __get_yaml_data()
    devs_pool = []
    # 獲取當(dāng)前連接的所有設(shè)備信息
    m = ManageDevices()
    all_devices_info = m.get_devices_info()
    # 補(bǔ)充每一個(gè)設(shè)備的啟動(dòng)信息，以及配置對(duì)應(yīng)的appium server端口號(hào)
    if all_devices_info:
        for dev_info in all_devices_info:
            dev_info.update(desired_template)
            dev_info["systemPort"] = system_port
            new_dict = {
                "caps": dev_info,
                "port": port
            }
            devs_pool.append(new_dict)
            port += 4
            system_port += 4
    return devs_pool

import subprocess
import os
 
from Common.handle_path import appium_logs_dir
 
class ManageAppiumServer:
    """
    appium desktop通過命令行啟動(dòng)appium服務(wù)。
    不同平臺(tái)上安裝的appium，默認(rèn)的appium服務(wù)路徑不一樣。
    初始化時(shí)，設(shè)置appium服務(wù)啟動(dòng)路徑
    再根據(jù)給定的端口號(hào)啟動(dòng)appium
    """
 
    def __init__(self,appium_server_apth):
        self.server_apth = appium_server_apth
 
    # 啟動(dòng)appium server服務(wù)
    def start_appium_server(self,port=4723):
        appium_log_path = os.path.join(appium_logs_dir,"appium_server_{0}.log".format(port))
        command = "node {0} -p {1} -g {2} " \
                  "--session-override " \
                  "--local-timezone " \
                  "--log-timestamp & ".format(self.server_apth, port, appium_log_path)
        subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE,shell=True).communicate()
 
    # 關(guān)閉appium服務(wù)
    @classmethod
    def stop_appium(cls,pc,post_num=4723):
        '''關(guān)閉appium服務(wù)'''
        if pc.upper() == 'WIN':
            p = os.popen(f'netstat  -aon|findstr {post_num}')
            p0 = p.read().strip()
            if p0 != '' and 'LISTENING' in p0:
                p1 = int(p0.split('LISTENING')[1].strip()[0:4])  # 獲取進(jìn)程號(hào)
                os.popen(f'taskkill /F /PID {p1}')  # 結(jié)束進(jìn)程
                print('appium server已結(jié)束')
        elif pc.upper() == 'MAC':
            p = os.popen(f'lsof -i tcp:{post_num}')
            p0 = p.read()
            if p0.strip() != '':
                p1 = int(p0.split('\n')[1].split()[1])  # 獲取進(jìn)程號(hào)
                os.popen(f'kill {p1}')  # 結(jié)束進(jìn)程
                print('appium server已結(jié)束')

def pytest_addoption(parser):
    parser.addoption(
        "--cmdopt", action="store", default="{platformName:'Android',platformVersion:'5.1.1'}",
        help="my devices info"
    )
 
 
@pytest.fixture(scope="session")
def cmdopt(request):
    return request.config.getoption("--cmdopt")
 
 
@pytest.fixture
def start_app(cmdopt):
    device = eval(cmdopt)
    print("開始與設(shè)備 {} 進(jìn)行會(huì)話，并執(zhí)行測(cè)試用例 ！！".format(device["caps"]["deviceName"]))
    driver = start_appium_session(device)
    yield driver
    driver.close_app()
    driver.quit()

# 根據(jù)設(shè)備啟動(dòng)信息，通過pytest.main來收集并執(zhí)行用例。
def run_cases(device):
  """
  參數(shù)：device為設(shè)備啟動(dòng)參數(shù)。在pytest.main當(dāng)中，傳遞給--cmdopt選項(xiàng)。
  """
    print(["-s", "-v", "--cmdopt={}".format(device)])
    reports_path = os.path.join(reports_dir,"test_result_{}_{}.html".format(device["caps"]["deviceName"], device["port"]))
    pytest.main(["-s", "-v",
                 "--cmdopt={}".format(device),
                 "--html={}".format(reports_path)]
                )

# 第一步：從設(shè)備池當(dāng)中，獲取當(dāng)前連接的設(shè)備。若設(shè)備池為空，則無設(shè)備連接。
devices = devices_pool()
 
# 第二步：若設(shè)備池不為空，啟動(dòng)appium server.與設(shè)備個(gè)數(shù)對(duì)應(yīng)。起始server端口為4723，每多一個(gè)設(shè)備，端口號(hào)默認(rèn)+4
if devices and platform_name and appium_server_path:
    # 創(chuàng)建線程池
    T = ThreadPoolExecutor()
    # 實(shí)例化appium服務(wù)管理類。
    mas = ManageAppiumServer(appium_server_path)
    for device in devices:
        # kill 端口，以免占用
        mas.stop_appium(platform_name,device["port"])
        # 啟動(dòng)appium server
        task = T.submit(mas.start_appium_server,device["port"])
        time.sleep(1)
 
    # 第三步：若設(shè)備池不為空，在appium server啟動(dòng)的情況下，執(zhí)行app自動(dòng)化測(cè)試。
    time.sleep(15)
    obj_list = []
    for device in devices:
        index = devices.index(device)
        task = T.submit(run_cases,device)
        obj_list.append(task)
        time.sleep(1)
 
    # 等待自動(dòng)化任務(wù)執(zhí)行完成
    for future in as_completed(obj_list):
        data = future.result()
        print(f"sub_thread: {data}")
 
    # kill 掉appium server服務(wù),釋放端口。
    for device in devices:
        ManageAppiumServer.stop_appium(platform_name, device["port"])