python+requests+unittest+HTMLTestRunner构建接口自动化测试框架（三）数据库、日志、http请求、邮件方法封装
- python+requests+unittest+HTMLTestRunner构建接口自动化测试框架（一）了解基础框架，读取配置文件
- python+requests+unittest+HTMLTestRunner构建接口自动化测试框架（二）commonMoudle模板详解
- python+requests+unittest+HTMLTestRunner构建接口自动化测试框架（三）数据库、日志、http请求、邮件方法封装
- python+requests+unittest+HTMLTestRunner构建接口自动化测试框架（四）测试用例执行
本篇主要针对数据库、日志、http请求、邮件方法封装的方法进行讲解
一、日志打印
封装日志打印的方法，可以使用不同等级的log打印想要的内容，Log.py文件完整代码如下：
# -*- coding:utf-8 -*-
#@Time : 2019/8/13 11:32
#@Author: csu
#@File : Log.py
import logging, threading, os
from datetime import datetime
import readConfig
localConfig = readConfig.ReadConfig()
class Log:
def __init__(self):
global logPath, resultPath, basePath
basePath = readConfig.basePath
resultPath = os.path.join(basePath, 'result')
if not os.path.exists(resultPath):
os.mkdir(resultPath)
logPath = os.path.join(resultPath, str(datetime.now().strftime("%Y%m%d%H%M%S")))
if not os.path.exists(logPath):
os.mkdir(logPath)
self.logger = logging.getLogger()
self.logger.setLevel(logging.INFO)
handler = logging.FileHandler(os.path.join(logPath, "output.log")) # logging.FileHandler -> 文件输出
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s') #设置日志格式
handler.setFormatter(formatter)
self.logger.addHandler(handler)
def get_logger(self):
"""
get logger
:return:
"""
return self.logger
def build_start_line(self, case_no):
"""
write start line
:return:
"""
self.logger.info("--------" + case_no + " START--------")
def build_end_line(self, case_no):
"""
write end line
:return:
"""
self.logger.info("--------" + case_no + " END--------")
def build_case_line(self, case_name, code, msg):
"""
write test case line
:param case_name:
:param code:
:param msg:
:return:
"""
self.logger.info(case_name+" - Code:"+code+" - msg:"+msg)
def get_report_path(self):
"""
get report file path
:return:
"""
report_path = os.path.join(logPath, "report.html")
return report_path
def get_result_path(self):
"""
get test result path
:return:
"""
return logPath
def write_result(self, result):
"""
:param result:
:return:
"""
result_path = os.path.join(logPath, "report.txt")
fb = open(result_path, "wb")
try:
fb.write(result)
except FileNotFoundError as ex:
self.logger.error(str(ex))
class MyLog:
log = None
mutex = threading.Lock() #创建锁
def __init__(self):
pass
@staticmethod
def get_log():
if MyLog.log is None:
MyLog.mutex.acquire() #锁定
MyLog.log = Log()
MyLog.mutex.release() #释放
return MyLog.log
二、邮件发送
因为公司邮箱是Outlook，所以选用邮件是Outlook发送，可能不适用于其他类型的邮箱发送，如果采用其他邮件方式发送的话，需要小伙伴自己研究啦
# -*- coding:utf-8 -*-
#@Time : 2019/4/11 17:18
#@Author: junni
#@File : configEmail.py
import os
import readConfig
import datetime
import win32com.client as win32
from common.Log import MyLog
localReadConfig = readConfig.ReadConfig()
proDir = readConfig.basePath
class send_email:
def __init__(self):
global subject, app, addressee, cc, mail_path
self.subject = localReadConfig.get_email("subject")
self.app = localReadConfig.get_email("app")
self.addressee = localReadConfig.get_email("addressee")
self.cc = localReadConfig.get_email("cc")
self.log = MyLog.get_log()
self.logger = self.log.get_logger()
#self.mail_path = os.path.join(proDir, 'result', 'report.html')
self.mail_path = MyLog.get_log().get_report_path()
def outlook(self):
olook = win32.Dispatch("%s.application" % self.app) #固定写法
mail = olook.CreateItem(0) #固定写法
mail.To = self.addressee #收件人
mail.CC = self.cc #抄送人
mail.Subject = str(datetime.datetime.now())[0:19] + '测试报告' # 邮件主题
mail.Attachments.Add(self.mail_path, 1, 1, "myFile")
read = open(self.mail_path, encoding='utf-8') # 打开需要发送的测试报告附件文件
content = read.read() # 读取测试报告文件中的内容
read.close()
#mail.Body = content # 将从报告中读取的内容，作为邮件正文中的内容
mail.Body = '接口测试结果'
mail.Send() # 发送
三、http请求
这里只是简单的封装post和get请求，需要结合具体请求进行设置
# -*- coding:utf-8 -*-
#@Time : 2019/8/12 19:57
#@Author: csu
#@File : configHttp.py
import requests
from common.Log import MyLog as Log
import readConfig
localReadConfig = readConfig.ReadConfig()
class ConfigHttp:
def __init__(self):
global protocol, ip, port
protocol = localReadConfig.get_http('protocol')
ip = localReadConfig.get_http('ip')
port = localReadConfig.get_http('port')
self.log = Log.get_log()
self.logger = self.log.get_logger()
self.data = None
self.url = None
def set_url(self, url):
self.url = protocol + '://' + ip + ':' + port + url
return self.url
def set_data(self, data):
self.data = data
return self.data
def get(self):
response = requests.get(self.url)
return response
def post(self):
response = requests.post(self.url, data = self.data)
return response
四、数据库
# -*- coding:utf-8 -*-
#@Time : 2019/8/12 17:02
#@Author: csu
#@File : configDB.py
import pymysql
import readConfig
from common.Log import MyLog as Log
localReadConfig = readConfig.ReadConfig()
class ConfigDB:
global host, username, password, port, database, config
host = localReadConfig.get_db('host')
username = localReadConfig.get_db('username')
password = localReadConfig.get_db('password')
port = localReadConfig.get_db('port')
database = localReadConfig.get_db('database')
config = {
'host':str(host),
'user':username,
'passwd':password,
'port':int(port),
'db':database
}
def __init__(self):
self.log = Log.get_log()
self.logger = self.log.get_logger()
self.db = None
self.cursor = None
def connectDB(self):
try:
self.db = pymysql.connect(**config)
self.cursor = self.db.cursor()
print("数据库连接成功")
except ConnectionError as ex:
self.logger.error(str(ex))
def executeSQL(self, sql, params):
self.connectDB()
self.cursor.execute(sql, params)
self.db.commit()
return self.cursor
def get_all(self, cursor):
value = cursor.fetchall()
return value
def get_one(self, cursor):
value = cursor.fetchone()
return value
def close_db(self):
self.db.close()
print("关闭数据库")