# -*- coding: utf-8 -*-
import time
import os
import os.path
import re
import unittest
import HTMLTestRunner
import shutil
shutil.copyfile("setting.ini","../setting.ini")
casepaths = []
def createsuite(casepath):
  testunit = unittest.TestSuite()
  #discover方法定義
  discover = unittest.defaultTestLoader.discover(
  casepath,
  pattern = 'case*.py',
  top_level_dir= casepath
  )
  for test_suite in discover:
    for test_case in test_suite:
      testunit.addTest(test_case)
  print testunit
  return testunit
for parent,dirnames,filenames in os.walk('.'):
 
  for filename in filenames:
    #print "parent is:" + parent
    #print "filename is:" + filename
    path=os.path.join(parent,filename)
    #正則判斷是否為測試用例
    match = re.match('case', filename)
    if match:
      print u"獲取測試用例目錄:%s"%parent
      casepaths.append(parent)
      break
 
     
 
#定義報告存放目錄,支持相對路徑
now = time.strftime("%Y-%m-%M-%H-%M-%S",time.localtime(time.time()))
filename = now+'report.html'
fp = file(filename,'wb')
runner = HTMLTestRunner.HTMLTestRunner(
stream = fp,
title = u'自動化測試報告',
description = u'用例執(zhí)行情況'
)
 
for casepath in casepaths:
  print u"正在執(zhí)行 %s目錄下的測試用例"%casepath
  alltestnames = createsuite(casepath)
  runner.run(alltestnames)
  print u"執(zhí)行 %s目錄下的測試用例完成"%casepath
print u"完成所有測試用例執(zhí)行任務(wù)"