"""
@Title   : app多設(shè)備并發(fā)-appium+pytest
@Author  : 檸檬班-小簡
@Email   : lemonban_simple@qq.com
"""
 
class ManageDevices:
    """
       1、重啟adb服務(wù)。
       2、通過adb devices命令獲取當前平臺中,已連接的設(shè)備個數(shù)，和設(shè)備uuid.
       3、通過adb -P 5037 -s 設(shè)備uuid shell getprop ro.build.version.release獲取每一個設(shè)備的版本號。
       4、將所有已連接設(shè)備的設(shè)備名稱、設(shè)備版本號存儲在一個列表當中。
       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,和版本號。
        :return: 所有已連接設(shè)備的uuid,和版本號。
        """
        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è)備啟動參數(shù)管理池。含啟動參數(shù)和對應(yīng)的端口號
    :param port: appium服務(wù)的端口號。每一個設(shè)備對應(yīng)一個。
    :param system_port: appium服務(wù)指定的本地端口，用來轉(zhuǎn)發(fā)數(shù)據(jù)給安卓設(shè)備。每一個設(shè)備對應(yīng)一個。
    :return: 所有已連接設(shè)備的啟動參數(shù)和appium端口號。
    """
    desired_template = __get_yaml_data()
    devs_pool = []
    # 獲取當前連接的所有設(shè)備信息
    m = ManageDevices()
    all_devices_info = m.get_devices_info()
    # 補充每一個設(shè)備的啟動信息，以及配置對應(yīng)的appium server端口號
    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通過命令行啟動appium服務(wù)。
    不同平臺上安裝的appium，默認的appium服務(wù)路徑不一樣。
    初始化時，設(shè)置appium服務(wù)啟動路徑
    再根據(jù)給定的端口號啟動appium
    """
 
    def __init__(self,appium_server_apth):
        self.server_apth = appium_server_apth
 
    # 啟動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])  # 獲取進程號
                os.popen(f'taskkill /F /PID {p1}')  # 結(jié)束進程
                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])  # 獲取進程號
                os.popen(f'kill {p1}')  # 結(jié)束進程
                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è)備 {} 進行會話，并執(zhí)行測試用例 ?。?.format(device["caps"]["deviceName"]))
    driver = start_appium_session(device)
    yield driver
    driver.close_app()
    driver.quit()

# 根據(jù)設(shè)備啟動信息，通過pytest.main來收集并執(zhí)行用例。
def run_cases(device):
  """
  參數(shù)：device為設(shè)備啟動參數(shù)。在pytest.main當中，傳遞給--cmdopt選項。
  """
    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è)備池當中，獲取當前連接的設(shè)備。若設(shè)備池為空，則無設(shè)備連接。
devices = devices_pool()
 
# 第二步：若設(shè)備池不為空，啟動appium server.與設(shè)備個數(shù)對應(yīng)。起始server端口為4723，每多一個設(shè)備，端口號默認+4
if devices and platform_name and appium_server_path:
    # 創(chuàng)建線程池
    T = ThreadPoolExecutor()
    # 實例化appium服務(wù)管理類。
    mas = ManageAppiumServer(appium_server_path)
    for device in devices:
        # kill 端口，以免占用
        mas.stop_appium(platform_name,device["port"])
        # 啟動appium server
        task = T.submit(mas.start_appium_server,device["port"])
        time.sleep(1)
 
    # 第三步：若設(shè)備池不為空，在appium server啟動的情況下，執(zhí)行app自動化測試。
    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)
 
    # 等待自動化任務(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"])