"""
@Title   : app多設備并發(fā)-appium+pytest
@Author  : 檸檬班-小簡
@Email   : lemonban_simple@qq.com
"""
 
class ManageDevices:
    """
       1、重啟adb服務。
       2、通過adb devices命令獲取當前平臺中,已連接的設備個數(shù)，和設備uuid.
       3、通過adb -P 5037 -s 設備uuid shell getprop ro.build.version.release獲取每一個設備的版本號。
       4、將所有已連接設備的設備名稱、設備版本號存儲在一個列表當中。
       5、通過調用get_devices_info函數(shù)，即可獲得4中的列表。
    """
 
    def __init__(self):
        self.__devices_info = []
        # 重啟adb服務
        self.__run_command_and_get_stout("adb kill-server")
        self.__run_command_and_get_stout("adb start-server")
 
    def get_devices_info(self):
        """
        獲取已連接設備的uuid,和版本號。
        :return: 所有已連接設備的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ù)和對應的端口號
    :param port: appium服務的端口號。每一個設備對應一個。
    :param system_port: appium服務指定的本地端口，用來轉發(fā)數(shù)據給安卓設備。每一個設備對應一個。
    :return: 所有已連接設備的啟動參數(shù)和appium端口號。
    """
    desired_template = __get_yaml_data()
    devs_pool = []
    # 獲取當前連接的所有設備信息
    m = ManageDevices()
    all_devices_info = m.get_devices_info()
    # 補充每一個設備的啟動信息，以及配置對應的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服務。
    不同平臺上安裝的appium，默認的appium服務路徑不一樣。
    初始化時，設置appium服務啟動路徑
    再根據給定的端口號啟動appium
    """
 
    def __init__(self,appium_server_apth):
        self.server_apth = appium_server_apth
 
    # 啟動appium server服務
    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()
 
    # 關閉appium服務
    @classmethod
    def stop_appium(cls,pc,post_num=4723):
        '''關閉appium服務'''
        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}')  # 結束進程
                print('appium server已結束')
        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}')  # 結束進程
                print('appium server已結束')

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("開始與設備 {} 進行會話，并執(zhí)行測試用例 ??！".format(device["caps"]["deviceName"]))
    driver = start_appium_session(device)
    yield driver
    driver.close_app()
    driver.quit()

# 根據設備啟動信息，通過pytest.main來收集并執(zhí)行用例。
def run_cases(device):
  """
  參數(shù)：device為設備啟動參數(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)]
                )

# 第一步：從設備池當中，獲取當前連接的設備。若設備池為空，則無設備連接。
devices = devices_pool()
 
# 第二步：若設備池不為空，啟動appium server.與設備個數(shù)對應。起始server端口為4723，每多一個設備，端口號默認+4
if devices and platform_name and appium_server_path:
    # 創(chuàng)建線程池
    T = ThreadPoolExecutor()
    # 實例化appium服務管理類。
    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)
 
    # 第三步：若設備池不為空，在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)
 
    # 等待自動化任務執(zhí)行完成
    for future in as_completed(obj_list):
        data = future.result()
        print(f"sub_thread: {data}")
 
    # kill 掉appium server服務,釋放端口。
    for device in devices:
        ManageAppiumServer.stop_appium(platform_name, device["port"])