RunStfTestMain.py

'''
  @Description      
  @auther         leizi
  @create          2020-03-20 18:39
'''
from config.config import Test_mobile_type, Test_plan_num, TestappPackage, TestAppActivity
from untils.StfTestPhoneUntil import StfPhoneOpear
import os, unittest, datetime, random,sys
from multiprocessing import Pool
from testcase.regcasetest import regtest
from untils.Parmeris import Parmer
from untils.log import LOG
from untils.makecase import makecasefile
from untils.pyreport_excel import create
from untils.AppiumServer import AppiumServer

test_ophone = []


def connectmobile(testnum) -> list:
    LOG.info("---开始链接stf平台-----")
    stf = StfPhoneOpear()
    all_list_phone = stf.getstflist()
    if len(all_list_phone) < testnum:
        return []
    LOG.info("---获取可用设备列表-----")
    for item in all_list_phone:
        if item['platform'] == Test_mobile_type and item['use'] is False:
            test_ophone.append(item)
    if len(test_ophone) < testnum:
        return []
    LOG.info("---准备申请设备-----")
    all_connect_phone = test_ophone[:testnum]
    connect_adb_device = []
    for i in range(len(all_connect_phone)):
        #TODO     设备申请后链接需要对链接的设备进行区分，
        ## 先找到已知设备列表，链接后，找到新增的那台设备，现在有bug
        LOG.info("---申请设备：%s-----" % all_connect_phone[i]['serial'])
        stf.oparyonephone(all_connect_phone[i]['serial'])
        LOG.info("---获取设备：%s 远程地址-----" % all_connect_phone[i]['serial'])
        connect = stf.getoneconnecturl(all_connect_phone[i]['serial'])
        all_detail = {}
        LOG.info("---链接设备：%s -----" % all_connect_phone[i]['serial'])
        os.popen("/Users/lileilei/Downloads/android-sdk-macosx/platform-tools/adb connect %s" % connect)
        reslut = os.popen("/Users/lileilei/Downloads/android-sdk-macosx/platform-tools/adb devices").read()
        index = i + 1
        devices = str(reslut).split("\n")[index].split("\t")[0]
        all_detail['devices'] = devices
        all_detail['serial'] = all_connect_phone[i]['serial']
        all_detail['version'] = all_connect_phone[i]['version']
        connect_adb_device.append(all_detail)
    return connect_adb_device


l_devices = []


def runnerPool(getDevices):
    '''
       根据链接的设备生成不同的dict
       然后放到设备的list里面
       设备list的长度产生进程池大小
    '''
    devices_Pool = []
    for i in range(0, len(getDevices)):
        _pool = []
        _initApp = {}
        _initApp["deviceName"] = getDevices[i]['devices']["devices"]
        _initApp["udid"] = getDevices[i]['devices']["devices"]
        _initApp["platformVersion"] = getDevices[i]['devices']['version']
        _initApp["platformName"] = Test_mobile_type
        _initApp["port"] = getDevices[i]["port"]
        _initApp["appPackage"] = TestappPackage
        _initApp["appActivity"] = TestAppActivity
        _pool.append(_initApp)
        devices_Pool.append(_initApp)
    pool = Pool(len(devices_Pool))
    for dev in devices_Pool:
        pool.map(runnerCaseApp, dev)
    pool.close()
    pool.join()


def runnerCaseApp(devices):
    '''利用unittest的testsuite来组织测试用例'''
    test_suit = unittest.TestSuite()
    test_suit.addTest(Parmer.parametrize(regtest, param=devices))  # 扩展的其他的测试用例均这样添加
    unittest.TextTestRunner(verbosity=2).run(test_suit)


if __name__ == "__main__":
    try:
        testnum=sys.argv[1]
    except :
        testnum=Test_plan_num
    LOG.info("测试开始执行")
    start_time = datetime.datetime.now()
    makecasefile('reg', 'reg', 'reg')
    path = os.getcwd()
    report=os.path.join(path,'testreport')
    filenm = report + 'result.xls'
    devicess = connectmobile(testnum)
    listport = []
    if len(devicess) > 0:
        for dev in devicess:
            app = {}
            app["devices"] = dev
            port = str(random.randint(4593, 4598))
            app["port"] = port
            l_devices.append(app)
            listport.append(port)
        appium_server = AppiumServer(l_devices)
        appium_server.start_server()  # 启动服务
        runnerPool(l_devices)
        try:
            appium_server.stop_server(listport)
        except Exception as e:
            print("关闭服务失败，原因：%s" % e)
            LOG.info("关闭服务失败，原因：%s" % e)
        end_time = datetime.datetime.now()
        hour = end_time - start_time
        s = StfPhoneOpear()
        for i in devicess:
            LOG.info("---归还设备：%s -----" % i['serial'])
            s.removeroneophen(i["serial"])
        create(filename=filenm, devices_list=devicess, Test_version='2.0.1', testtime=str(hour))
        LOG.info("测试执行完毕，耗时：%s" % hour)
    else:
        LOG.info("没有可用的%s设备,请查看stf链接设备" % Test_mobile_type)
        print("没有可用的%s设备,请查看stf链接设备" % Test_mobile_type)