#****************************************************************# TestFrame.py# Author   : Vince# Version  : 1.1.2# Date    : 2011-3-14# Description: 自動化測試平臺#****************************************************************import os,sys, urllib, httplib, profile, datetime, timefrom xml2dict import XML2Dictimport win32com.clientfrom win32com.client import Dispatchimport xml.etree.ElementTree as et#import MySQLdb#Excel表格中測試結(jié)果底色OK_COLOR=0xffffffNG_COLOR=0xff#NT_COLOR=0xffffNT_COLOR=0xC0C0C0#Excel表格中測試結(jié)果匯總顯示位置TESTTIME=[1, 14]TESTRESULT=[2, 14]#Excel模版設(shè)置#self.titleindex=3    #Excel中測試用例標題行索引#self.casebegin =4    #Excel中測試用例開始行索引#self.argbegin  =3    #Excel中參數(shù)開始列索引#self.argcount =8    #Excel中支持的參數(shù)個數(shù)class create_excel:  def __init__(self, sFile, dtitleindex=3, dcasebegin=4, dargbegin=3, dargcount=8):    self.xlApp = win32com.client.Dispatch('et.Application')  #MS:Excel WPS:et    try:      self.book = self.xlApp.Workbooks.Open(sFile)    except:      print_error_info()      print "打開文件失敗"      exit()    self.file=sFile    self.titleindex=dtitleindex    self.casebegin=dcasebegin    self.argbegin=dargbegin    self.argcount=dargcount    self.allresult=[]    self.retCol=self.argbegin+self.argcount    self.xmlCol=self.retCol+1    self.resultCol=self.xmlCol+1  def close(self):    #self.book.Close(SaveChanges=0)    self.book.Save()    self.book.Close()    #self.xlApp.Quit()    del self.xlApp  def read_data(self, iSheet, iRow, iCol):    try:      sht = self.book.Worksheets(iSheet)      sValue=str(sht.Cells(iRow, iCol).Value)    except:      self.close()      print('讀取數(shù)據(jù)失敗')      exit()    #去除'.0'    if sValue[-2:]=='.0':      sValue = sValue[0:-2]    return sValue  def write_data(self, iSheet, iRow, iCol, sData, color=OK_COLOR):    try:      sht = self.book.Worksheets(iSheet)      sht.Cells(iRow, iCol).Value = sData.decode("utf-8")      sht.Cells(iRow, iCol).Interior.Color=color      self.book.Save()    except:      self.close()      print('寫入數(shù)據(jù)失敗')      exit()  #獲取用例個數(shù)  def get_ncase(self, iSheet):    try:      return self.get_nrows(iSheet)-self.casebegin+1    except:      self.close()      print('獲取Case個數(shù)失敗')      exit()  def get_nrows(self, iSheet):    try:      sht = self.book.Worksheets(iSheet)      return sht.UsedRange.Rows.Count    except:      self.close()      print('獲取nrows失敗')      exit()  def get_ncols(self, iSheet):    try:      sht = self.book.Worksheets(iSheet)      return sht.UsedRange.Columns.Count    except:      self.close()      print('獲取ncols失敗')      exit()  def del_testrecord(self, suiteid):    try:      #為提升性能特別從For循環(huán)提取出來      nrows=self.get_nrows(suiteid)+1      ncols=self.get_ncols(suiteid)+1      begincol=self.argbegin+self.argcount      #提升性能      sht = self.book.Worksheets(suiteid)      for row in range(self.casebegin, nrows):        for col in range(begincol, ncols):          str=self.read_data(suiteid, row, col)          #清除實際結(jié)果[]          startpos = str.find('[')          if startpos>0:            str = str[0:startpos].strip()            self.write_data(suiteid, row, col, str, OK_COLOR)          else:            #提升性能            sht.Cells(row, col).Interior.Color = OK_COLOR        #清除TestResul列中的測試結(jié)果，設(shè)置為NT        self.write_data(suiteid, row, self.argbegin+self.argcount+1, ' ', OK_COLOR)        self.write_data(suiteid, row, self.resultCol, 'NT', NT_COLOR)    except:      self.close()      print('清除數(shù)據(jù)失敗')      exit()#執(zhí)行調(diào)用def HTTPInvoke(IPPort, url):  conn = httplib.HTTPConnection(IPPort)  conn.request("GET", url)  rsps = conn.getresponse()  data = rsps.read()  conn.close()  return data#獲取用例基本信息[Interface,argcount,[ArgNameList]]def get_caseinfo(Data, SuiteID):  caseinfolist=[]  sInterface=Data.read_data(SuiteID, 1, 2)  argcount=int(Data.read_data(SuiteID, 2, 2))  #獲取參數(shù)名存入ArgNameList  ArgNameList=[]  for i in range(0, argcount):    ArgNameList.append(Data.read_data(SuiteID, Data.titleindex, Data.argbegin+i))  caseinfolist.append(sInterface)  caseinfolist.append(argcount)  caseinfolist.append(ArgNameList)  return caseinfolist#獲取輸入def get_input(Data, SuiteID, CaseID, caseinfolist):  sArge=''  #參數(shù)組合  for j in range(0, caseinfolist[1]):    if Data.read_data(SuiteID, Data.casebegin+CaseID, Data.argbegin+j) != "None":      sArge=sArge+caseinfolist[2][j]+'='+Data.read_data(SuiteID, Data.casebegin+CaseID, Data.argbegin+j)+'&'  #去掉結(jié)尾的&字符  if sArge[-1:]=='&':    sArge = sArge[0:-1]  sInput=caseinfolist[0]+sArge  #組合全部參數(shù)  return sInput#結(jié)果判斷def assert_result(sReal, sExpect):  sReal=str(sReal)  sExpect=str(sExpect)  if sReal==sExpect:    return 'OK'  else:    return 'NG'#將測試結(jié)果寫入文件def write_result(Data, SuiteId, CaseId, resultcol, *result):  if len(result)>1:    ret='OK'    for i in range(0, len(result)):      if result[i]=='NG':        ret='NG'        break    if ret=='NG':      Data.write_data(SuiteId, Data.casebegin+CaseId, resultcol,ret, NG_COLOR)    else:      Data.write_data(SuiteId, Data.casebegin+CaseId, resultcol,ret, OK_COLOR)    Data.allresult.append(ret)  else:    if result[0]=='NG':      Data.write_data(SuiteId, Data.casebegin+CaseId, resultcol,result[0], NG_COLOR)    elif result[0]=='OK':      Data.write_data(SuiteId, Data.casebegin+CaseId, resultcol,result[0], OK_COLOR)    else: #NT      Data.write_data(SuiteId, Data.casebegin+CaseId, resultcol,result[0], NT_COLOR)    Data.allresult.append(result[0])  #將當前結(jié)果立即打印  print 'case'+str(CaseId+1)+':', Data.allresult[-1]#打印測試結(jié)果def statisticresult(excelobj):  allresultlist=excelobj.allresult  count=[0, 0, 0]  for i in range(0, len(allresultlist)):    #print 'case'+str(i+1)+':', allresultlist[i]    count=countflag(allresultlist[i],count[0], count[1], count[2])  print 'Statistic result as follow:'  print 'OK:', count[0]  print 'NG:', count[1]  print 'NT:', count[2]#解析XmlString返回Dictdef get_xmlstring_dict(xml_string):  xml = XML2Dict()  return xml.fromstring(xml_string)#解析XmlFile返回Dictdef get_xmlfile_dict(xml_file):  xml = XML2Dict()  return xml.parse(xml_file)#去除歷史數(shù)據(jù)expect[real]def delcomment(excelobj, suiteid, iRow, iCol, str):  startpos = str.find('[')  if startpos>0:    str = str[0:startpos].strip()    excelobj.write_data(suiteid, iRow, iCol, str, OK_COLOR)  return str#檢查每個item （非結(jié)構(gòu)體）def check_item(excelobj, suiteid, caseid,real_dict, checklist, begincol):  ret='OK'  for checkid in range(0, len(checklist)):    real=real_dict[checklist[checkid]]['value']    expect=excelobj.read_data(suiteid, excelobj.casebegin+caseid, begincol+checkid)    #如果檢查不一致測將實際結(jié)果寫入expect字段，格式：expect[real]    #將return NG    result=assert_result(real, expect)    if result=='NG':      writestr=expect+'['+real+']'      excelobj.write_data(suiteid, excelobj.casebegin+caseid, begincol+checkid, writestr, NG_COLOR)      ret='NG'  return ret#檢查結(jié)構(gòu)體類型def check_struct_item(excelobj, suiteid, caseid,real_struct_dict, structlist, structbegin, structcount):  ret='OK'  if structcount>1: #傳入的是List    for structid in range(0, structcount):      structdict=real_struct_dict[structid]      temp=check_item(excelobj, suiteid, caseid,structdict, structlist, structbegin+structid*len(structlist))      if temp=='NG':        ret='NG'  else: #傳入的是Dict    temp=check_item(excelobj, suiteid, caseid,real_struct_dict, structlist, structbegin)    if temp=='NG':      ret='NG'  return ret#獲取異常函數(shù)及行號def print_error_info():  """Return the frame object for the caller's stack frame."""  try:    raise Exception  except:    f = sys.exc_info()[2].tb_frame.f_back  print (f.f_code.co_name, f.f_lineno)#測試結(jié)果計數(shù)器，類似Switch語句實現(xiàn)def countflag(flag,ok, ng, nt):  calculation = {'OK':lambda:[ok+1, ng, nt],             'NG':lambda:[ok, ng+1, nt],             'NT':lambda:[ok, ng, nt+1]}  return calculation[flag]()

# -*- coding: utf-8 -*-#****************************************************************# xxx_server_case.py# Author   : Vince# Version  : 1.0# Date    : 2011-3-10# Description: 內(nèi)容服務(wù)系統(tǒng)測試代碼#****************************************************************from testframe import *from common_lib import *httpString='http://xxx.com/xxx_product/test/'expectXmldir=os.getcwd()+'/TestDir/expect/'realXmldir=os.getcwd()+'/TestDir/real/'def run(interface_name, suiteid):  print '【'+interface_name+'】' + ' Test Begin,please waiting...'  global expectXmldir, realXmldir  #根據(jù)接口名分別創(chuàng)建預(yù)期結(jié)果目錄和實際結(jié)果目錄  expectDir=expectXmldir+interface_name  realDir=realXmldir+interface_name  if os.path.exists(expectDir) == 0:    os.makedirs(expectDir)  if os.path.exists(realDir) == 0:    os.makedirs(realDir)  excelobj.del_testrecord(suiteid) #清除歷史測試數(shù)據(jù)  casecount=excelobj.get_ncase(suiteid) #獲取case個數(shù)  caseinfolist=get_caseinfo(excelobj, suiteid) #獲取Case基本信息  #遍歷執(zhí)行case  for caseid in range(0, casecount):    #檢查是否執(zhí)行該Case    if excelobj.read_data(suiteid,excelobj.casebegin+caseid, 2)=='N':      write_result(excelobj, suiteid, caseid, excelobj.resultCol, 'NT')      continue #當前Case結(jié)束，繼續(xù)執(zhí)行下一個Case    #獲取測試數(shù)據(jù)    sInput=httpString+get_input(excelobj, suiteid, caseid, caseinfolist)    XmlString=HTTPInvoke(com_ipport, sInput)   #執(zhí)行調(diào)用    #獲取返回碼并比較    result_code=et.fromstring(XmlString).find("result_code").text    ret1=check_result(excelobj, suiteid, caseid,result_code, excelobj.retCol)    #保存預(yù)期結(jié)果文件    expectPath=expectDir+'/'+str(caseid+1)+'.xml'    #saveXmlfile(expectPath, XmlString)    #保存實際結(jié)果文件    realPath=realDir+'/'+str(caseid+1)+'.xml'    saveXmlfile(realPath, XmlString)    #比較預(yù)期結(jié)果和實際結(jié)果    ret2= check_xmlfile(excelobj, suiteid, caseid,expectPath, realPath)    #寫測試結(jié)果    write_result(excelobj, suiteid, caseid, excelobj.resultCol, ret1, ret2)  print '【'+interface_name+'】' + ' Test End!'

# -*- coding: utf-8 -*-#****************************************************************# main.py# Author   : Vince# Version  : 1.0# Date    : 2011-3-16# Description: 測試組裝，用例執(zhí)行入口#****************************************************************from testframe import *from xxx_server_case import *import xxx_server_case#產(chǎn)品系統(tǒng)接口測試#設(shè)置測試環(huán)境xxx_server_case.excelobj=create_excel(os.getcwd()+'/TestDir/xxx_Testcase.xls')xxx_server_case.com_ipport=xxx.com'#Add testsuite beginrun("xxx_book_list", 4)#Add other suite from here#Add testsuite endstatisticresult(xxx_server_case.excelobj)xxx_server_case.excelobj.close()