#**************************************************************** # TestFrame.py # Author : Vince # Version : 1.1.2 # Date : 2011-3-14 # Description: 自动化测试平台 #**************************************************************** import os,sys, urllib, httplib, profile, datetime, time from xml2dict import XML2Dict import win32com.client from win32com.client import Dispatch import xml.etree.ElementTree as et #import MySQLdb #Excel表格中测试结果底色 OK_COLOR = 0xffffff NG_COLOR = 0xff #NT_COLOR=0xffff NT_COLOR = 0xC0C0C0 #Excel表格中测试结果汇总显示位置 TESTTIME = [ 1 , 14 ] TESTRESULT = [ 2 , 14 ] #Excel模版设置 #self.titleindex=3 #Excel中测试用例标题行索引 #self.casebegin =4 #Excel中测试用例开始行索引 #self.argbegin =3 #Excel中参数开始列索引 #self.argcount =8 #Excel中支持的参数个数 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 ( '读取数据失败' ) 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 ( '写入数据失败' ) exit() #获取用例个数 def get_ncase( self , iSheet): try : return self .get_nrows(iSheet) - self .casebegin + 1 except : self .close() print ( '获取Case个数失败' ) 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循环提取出来 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) #清除实际结果[] 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列中的测试结果,设置为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 ( '清除数据失败' ) exit() #执行调用 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 )) #获取参数名存入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 = '' #参数组合 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) + '&' #去掉结尾的&字符 if sArge[ - 1 :] = = '&' : sArge = sArge[ 0 : - 1 ] sInput = caseinfolist[ 0 ] + sArge #组合全部参数 return sInput #结果判断 def assert_result(sReal, sExpect): sReal = str (sReal) sExpect = str (sExpect) if sReal = = sExpect: return 'OK' else : return 'NG' #将测试结果写入文件 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 ]) #将当前结果立即打印 print 'case' + str (CaseId + 1 ) + ':' , Data.allresult[ - 1 ] #打印测试结果 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返回Dict def get_xmlstring_dict(xml_string): xml = XML2Dict() return xml.fromstring(xml_string) #解析XmlFile返回Dict def get_xmlfile_dict(xml_file): xml = XML2Dict() return xml.parse(xml_file) #去除历史数据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 (非结构体) 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) #如果检查不一致测将实际结果写入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 #检查结构体类型 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 #获取异常函数及行号 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) #测试结果计数器,类似Switch语句实现 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]() |