# -*- coding: UTF-8 -*-
import unittest,sys
from BeautifulReport import BeautifulReport as bf
from configuration import config, Encrypt
from configuration.logger import logger
from configuration.mysql import Mysql
from utest import datadriven


# 运行的相对路径
path = '.'
# 用例路径
casepath = ''
resultpath = ''

if __name__ == '__main__':
    # unittest.main()
    # suite = unittest.TestSuite()
    # suite.addTests(unittest.defaultTestLoader.loadTestsFromTestCase(testWeb))
    # suite = unittest.defaultTestLoader.discover(".", pattern="baidu.py", top_level_dir=None)
    # # 生成执行用例的对象
    # runner = bf(suite)
    # runner.report(filename='./test.html', description='这个描述参数是必填的')

    try:
        casepath = sys.argv[1]
    except:
        casepath = ''

    # 为空，则使用默认的
    if casepath == '':
        casepath = path + '/testCase/API用例1.xlsx'
        resultpath = path + '/testCase/result/result-API用例1.xlsx'
    else:
        # 如果是绝对路径，就使用绝对路径
        if casepath.find(':') >= 0:
            # 获取用例文件名
            resultpath = path + '/testCase/result-' + casepath[casepath.rfind('\\') + 1:]
        else:
            logger.error('非法用例路径')

    config.get_config(path + '/conf/conf.properties')
    # logger.info(config.config)
    mysql = Mysql()
    mysql.init_mysql(path + '/conf/initialize.sql')
    # 获取参数化的二维列表
    datadriven.getparams(casepath,resultpath)

    # unittest.main()
    # suite = unittest.TestSuite()
    # suite.addTests(unittest.defaultTestLoader.loadTestsFromTestCase(baidu))
    suite = unittest.defaultTestLoader.discover("./utest/", pattern="alltest.py", top_level_dir=None)
    # 生成执行用例的对象
    runner = bf(suite)
    runner.report(filename='./module.html', description='unitTest测试框架')