python自動化學習筆記11-自動化測試UTP框架

 

前面基本的unittest及ddt已經學過了，現在我們系統把這些知識結合起來，寫一個簡單的UTP自動化測試框架；

我們先來建基礎目錄，首先新建一個項目，項目下建父目錄UTP，conf目錄，用來存放配置文件信息；data目錄，用來存放測試數據；cases目錄用來存放測試用例，common用來存放通用方法，report用來存放生成的測試報告，log用來存放生成的日志；bin目錄用來存放執行文件；基本的目錄結構就建好了；如下：

 

 下面我們來完善具體代碼。

先來分析一下思路：

1、編寫配置文件信息

2、編寫測試數據，yml文件

3、生成測試用例（調用接口的方法），Python文件

4、獲取測試用例

5、通過unittest執行測試用例

6、生成測試報告，發送郵件

首先在conf目錄下新建setting.py文件，用來存放配置信息

import os
#當前目錄的父目錄
BASE_PATH=os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
#測試用例路徑
CASE_PATH=os.path.join(BASE_PATH,'cases')
#測試數據路徑
DATA_PATH=os.path.join(BASE_PATH,'data')
#日志路徑
LOG_PATH=os.path.join(BASE_PATH,'log','utp.log')
#測試報告路徑
REPORT_PATH=os.path.join(BASE_PATH,'report')
#測試用例示例路徑
EG_PATH=os.path.join(BASE_PATH,'conf')

USER_MAIL='發件箱'
MAIL_PASSWD='密碼，QQ為授權碼'
MAIL_HOST='smtp.qq.com'
RECV_MAIL='收件箱'

在data目錄下新建yml格式的測試數據；示例如下：

-
 url : http://www.xxxx.com/user/loginAction
 method : post
 data:
 username : pei
 password : 123456
 check : '請輸入正確的用戶名和密碼'
 header:
 Connection: keep-alive
-
 url : http://www.xxxx.com/user/loginAction
 method : post
 data:
 password : 123456
 check : '請輸入正確的用戶名和密碼'
 header:
 Connection: keep-alive

在cases目錄下新建testcase.py，

import os
import ddt
import requests
import unittest


@ddt.ddt
class Test_login(unittest.TestCase):
 @ddt.file_data(r'C:\besttest\untitled1\UTP\data\login.yml')
 @ddt.unpack
 def test_login(self,**kwargs):
 url=kwargs.get('url')
 method=kwargs.get('method')
 data=kwargs.get('data',{})
 header=kwargs.get('header',{})
 is_json=kwargs.get('is_json',0)
 cookie=kwargs.get('cookie',{})
 check=kwargs.get('check')
 if method.upper=='POST':
 if is_json:
 req=requests.post(url=url,json=data,headers=header,cookies=cookie).text
 else:
 req=requests.post(url=url,data=data,headers=header,cookies=cookie).text
 else:
 req=requests.get(url=url,params=data,headers=header,cookies=cookie).text
 self.assertIn(check,req)

在寫測試用例的過程中我們發現，由於測試用例基本格式一致，只是調用的測試數據不同，生成的測試用例名稱不同，，所以我們可以通過寫代碼來實現，根據測試用例模板，自動創建測試用例py文件，這個時候我們把testcase.py文件改名為testcase.eg,作為測試用例的一個示例；把class名參數化，把yml文件參數化，同時把測試用例模板放到conf目錄下；

import os
import ddt
import requests
import unittest
@ddt.ddt
class %s(unittest.TestCase):
    def setup(self):
        print('初始化')
    def tearDown(self):
        print('執行結束')
 @ddt.file_data(r'%s')
 @ddt.unpack
 def test_login(self,**kwargs):
 url=kwargs.get('url')
 method=kwargs.get('method')
 data=kwargs.get('data',{})
 header=kwargs.get('header',{})
 is_json=kwargs.get('is_json',0)
 cookie=kwargs.get('cookie',{})
 check=kwargs.get('check')
 if method.upper=='POST':
 if is_json:
 req=requests.post(url=url,json=data,headers=header,cookies=cookie).text
 else:
 req=requests.post(url=url,data=data,headers=header,cookies=cookie).text
 else:
 req=requests.get(url=url,params=data,headers=header,cookies=cookie).text
 self.assertIn(check,req)

然后編寫生成測試用例的方法；

import os
from conf.setting import EG_PATH,CASE_PATH,DATA_PATH

class GetCase(object):
 def creat_case(self):
 #打開示例文件
 with open(os.path.join(EG_PATH,'testcase.eg'),encoding='utf-8') as fw:
 #讀取出文件內容
 test_case_str=fw.read()
 #獲取出yml測試數據
 file_list=os.listdir(DATA_PATH)
 for file_name in file_list:
 #yml文件的絕對路徑
 file_path=os.path.join(DATA_PATH,file_name)
 if file_name.endswith('yml') or file_name.endswith('yaml'):
 #生成的測試用例的文件名
 abs_case_path=os.path.join(CASE_PATH,'test_'+file_name.split('.')[0]+'.py')
 #生成的測試用例中的方法名
 class_name='Test_'+file_name.split('.')[0]
 #生成py文件
 with open(abs_case_path,'w+',encoding='utf-8') as f:
 f.write((test_case_str)%(class_name,file_path))


if __name__ == '__main__':
 getcase=GetCase()
 getcase.creat_case()

測試通過，可以正常根據示例生成測試用例的Python文件；

接下來編寫執行用例的文件；

 import os
import sys
#添加到環境變量里
BASE_PATH=os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0,BASE_PATH)

import unittest
import BeautifulReport
from conf.setting import CASE_PATH,REPORT_PATH,USER_MAIL,MAIL_PASSWD,MAIL_HOST,RECV_MAIL
from common.getTestCase import GetCase
import time
from common.sendemail import SendMail

def Run_Case():
#實例化生成獲取測試用例的方法
cases=GetCase()
#生成py的測試用例
cases.creat_case()
#創建測試套件
suite=unittest.TestSuite()
#獲取所有的測試用例
all_case=unittest.defaultTestLoader.discover(CASE_PATH,'test*py')
#將測試用例添加到測試套件里執行
[suite.addTests(case) for case in all_case]
#生成測試結果
result=BeautifulReport.BeautifulReport(suite)
#生成測試報告名稱
report_name=time.strftime('%Y%m%d%H%M%S')+'_report.html'
#生成測試報告
result.report(description='接口自動化測試',filename=report_name,log_path=REPORT_PATH)
#測試報告的絕對路徑
abs_report_path=os.path.join(REPORT_PATH,report_name)
#實例化郵件
mail=SendMail(
username=USER_MAIL,
passwd=MAIL_PASSWD,
recv=RECV_MAIL,
title='接口自動化測試報告',
content=' 各位，此次是本次自動化測試結果，具體結果請查看測試報告',
file=abs_report_path, ssl=True,
email_host=MAIL_HOST)
#發送郵件
mail.send_mail()

if __name__ == '__main__':
Run_Case()

 執行查看結果：

在report目錄下生成了測試測試報告，打開測試報告顯示此次測試結果：

再附上發送郵件的類方法：

import smtplib,os
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
import base64

class SendMail(object):
 def __init__(self,username,passwd,recv,title,content,
 file=None,ssl=False,
 email_host='smtp.163.com',port=25,ssl_port=465):
 '''
 :param username: 用戶名
 :param passwd: 密碼
 :param recv: 收件人，多個要傳list ['[e-mail]','[e-mail]]
 :param title: 郵件標題
 :param content: 郵件正文
 :param file: 附件路徑，如果不在當前目錄下，要寫絕對路徑，默認沒有附件
 :param ssl: 是否安全鏈接，默認為普通
 :param email_host: smtp服務器地址，默認為163服務器
 :param port: 非安全鏈接端口，默認為25
 :param ssl_port: 安全鏈接端口，默認為465
 '''
 self.username = username #用戶名
 self.passwd = passwd #密碼
 self.recv = recv #收件人，多個要傳list ['[e-mail]','[e-mail]]
 self.title = title #郵件標題
 self.content = content #郵件正文
 self.file = file #附件路徑，如果不在當前目錄下，要寫絕對路徑
 self.email_host = email_host #smtp服務器地址
 self.port = port #普通端口
 self.ssl = ssl #是否安全鏈接
 self.ssl_port = ssl_port #安全鏈接端口
 def send_mail(self):
 msg = MIMEMultipart()
 #發送內容的對象
 if self.file:#處理附件的
 file_name = os.path.split(self.file)[-1]#只取文件名，不取路徑
 try:
 f = open(self.file, 'rb').read()
 except Exception as e:
 raise Exception('附件打不開！！！！')
 else:
 att = MIMEText(f,"base64", "utf-8")
 att["Content-Type"] = 'application/octet-stream'
 #base64.b64encode(file_name.encode()).decode()
 new_file_name='=?utf-8?b?' + base64.b64encode(file_name.encode()).decode() + '?='
 #這里是處理文件名為中文名的，必須這么寫
 att["Content-Disposition"] = 'attachment; filename="%s"'%(new_file_name)
 msg.attach(att)
 msg.attach(MIMEText(self.content))#郵件正文的內容
 msg['Subject'] = self.title # 郵件主題
 msg['From'] = self.username # 發送者賬號
 msg['To'] = ','.join(self.recv) # 接收者賬號列表
 if self.ssl:
 self.smtp = smtplib.SMTP_SSL(self.email_host,port=self.ssl_port)
 else:
 self.smtp = smtplib.SMTP(self.email_host,port=self.port)
 #發送郵件服務器的對象
 self.smtp.login(self.username,self.passwd)
 try:
 self.smtp.sendmail(self.username,self.recv,msg.as_string())
 pass
 except Exception as e:
 print('出錯了。。',e)
 else: print('發送成功！') self.smtp.quit()

該方法可以通用於郵件發送，以下就是整個框架的完整結構了，一個簡單的自動化測試框架就完成了，大家也來試試吧。

上述是基於數據驅動的自動化測試，后面我們繼續學代碼驅動，用來編寫有關聯關系的測試用例