import argparse
import os.path
import shutil
import sys

from xdevice.__main__ import main_process
from xdevice._core.report.result_reporter import ResultReporter
from xml.dom.minidom import parse
# import pandas as pd
from datetime import datetime

BASE_DIR = os.path.dirname(__file__)

TEST_CASE_MUST = [
    'Launcher',
    #'ProcessCheck',
    #'APLCheck',
    #'ACLCheck',
]

# 根据用例耗时分配的两台设备跑的用例，使两台设备跑的时间都差不多在2min
TEST_CASE_DEVICE1 = [
    'SettingsWifi',
    #'CrashCheck',
    #'Photos',
    #'Contacts',
    #'Mms',
    #'DistributedMusicPlayer'
]

TEST_CASE_DEVICE2 = [
    'Camera',
    #'NotificationBar',
    #'Note',
]


def get_test_result(report_path):
    try:
        if not os.path.exists(report_path):
            return False
        rst = ResultReporter.get_task_info_params(report_path)
        unsuccessful_params = rst.get('unsuccessful_params')
        for case, step in unsuccessful_params.items():
            if step:
                return False
        return True
    except:
        return False


# def collect_test_result(report_path):
#     xml_report = os.path.join(report_path, 'summary_report.xml')
#     if not os.path.exists(xml_report):
#         return
#     timestamp = datetime.fromtimestamp(os.path.getmtime(xml_report))
#     test_date = timestamp.strftime('%Y-%m-%d')
#
#     try:
#         dom = parse(xml_report)
#         data = dom.documentElement
#         test_result = {
#             '用例名': [],
#             '测试结果': [],
#             '耗时': [],
#             '报错信息': [],
#             '报告路径': [],
#         }
#         testcases = data.getElementsByTagName('testsuite')
#         testcase_result = []
#         for t in testcases:
#             module_name = t.getAttribute('modulename')
#             result_kind = t.getAttribute('result_kind')
#             time = t.getAttribute('time')
#             testcase = t.getElementsByTagName('testcase')
#             message = testcase[0].getAttribute('message')
#             line = (module_name, result_kind, time, message, xml_report, test_date)
#             if line not in testcase_result:
#                 testcase_result.append(line)
#                 # csv
#                 test_result['用例名'].append(module_name)
#                 test_result['测试结果'].append(result_kind)
#                 test_result['耗时'].append(time)
#                 test_result['报错信息'].append(message)
#                 test_result['报告路径'].append(xml_report)
#
#         df = pd.DataFrame(test_result)
#
#         with open('D:\\smoke_result_{}.csv'.format(test_date), 'a', newline='') as f:
#             df.to_csv(f, header=f.tell() == 0, index=False, mode='a')
#     except:
#         pass
#

if __name__ == '__main__':
    argv = sys.argv[1:]
    parser = argparse.ArgumentParser(description='manual to this scription')
    parser.add_argument('--config', type=str)
    parser.add_argument('--test_num', type=str, default='1/1')
    parser.add_argument('--tools_path', type=str)
    parser.add_argument('--anwser_path', type=str)
    parser.add_argument('--save_path', type=str)
    parser.add_argument('--device_num', type=str)
    parser.add_argument('--pr_url', type=str)
    args = parser.parse_args()

    new_cmd = 'run'
    # 指定设备sn
    if not args.device_num:
        print("SmokeTest: End of check, test failed!")
        sys.exit(98)
    new_cmd += ' -sn {}'.format(args.device_num)
    # 测试用例路径
    tcpath = args.tools_path
    new_cmd += ' -tcpath {}'.format(tcpath)
    # 测试的设备编号，1/1表示只有一台设备；1/2表示第一台设备；2/2表示第二台设备
    if args.test_num == '1/1':
        new_cmd += ' -l {}'.format(';'.join(TEST_CASE_MUST + TEST_CASE_DEVICE1 + TEST_CASE_DEVICE2))
    elif args.test_num == '1/2':
        new_cmd += ' -l {}'.format(';'.join(TEST_CASE_MUST + TEST_CASE_DEVICE1))
    elif args.test_num == '2/2':
        new_cmd += ' -l {}'.format(';'.join(TEST_CASE_MUST + TEST_CASE_DEVICE2))
    # 指定报告生成路径
    report_path = args.save_path
    new_cmd += ' -rp {} -ta screenshot:true'.format(report_path)
    # 测试资源路径
    # respath = args.anwser_path
    # new_cmd += ' -respath {}'.format(respath)
    # shutil.rmtree(os.path.join(BASE_DIR, 'reports'), ignore_errors=True)
    
    print('SmokeTest Begin >>>>>>>>>>>>')
    main_process(new_cmd)

    # print('SmokeTest collect test result >>>>>>>>>>>>')
    # collect_test_result(report_path)

    print('SmokeTest ending >>>>>>>>>>>>')
    smoke_rst = get_test_result(report_path)
    if smoke_rst:
        print("SmokeTest: End of check, test succeeded!")
        sys.exit(0)
    print("SmokeTest: End of check, test failed!")
    sys.exit(99)