目录
一、日志封装及应用（理解）
二、认证开户接口脚本编写
1、代码编写
1️⃣api目录
2️⃣script目录
2、BeautifulSoup库
1️⃣简介及例子
2️⃣提取html数据工具封装
3、认证开户参数化
一、日志封装及应用（理解）
🔴日志的作用：
- 记录程序运行的步骤和错误。
🔴日志的场景：
- 1、调试bug
- 2、查看程序运行轨迹
🔴日志基本应用：
# 1、导包
import logging
# 2、调用日志⼊口
logging.error("出错啦，错误原因：{}".format(e))import logging
# 设置日志级别 及保存⽂件名
logging.basicConfig(level=logging.DEBUG, filename="../log/p2p.log")
# 调用日志
logging.debug("调试信息")
logging.info("信息级别")
logging.warning("警告")
logging.error("断⾔错误！")
logging.critical("严重错误")
🔴测试人员使用的日志的入口：
- info：记录运行步骤
- error：记录运行错误
🔴日志底层组成介绍：了解底层是为了修改log打印出的信息进行美化封装
- 说明：logging库底层有4大组件（日志器、处理器、格式器、过滤器）
- 1、日志器：接受日志信息，设置日志显示级别
- 2、处理器：控制日志显示位置或文件
- 3、格式器：控制日志输出的显示样式
- 关系：
- 格式器必须关联处理器
- 处理器必须关联日志器
🔴日志封装应用：
重组封装的目的：解决日志显示的样式、存储方式
①日志工具的封装
在util.py中添加日志工具的封装：
下面这个日志工具类可以当模板用，最多改一下保存日志的.log文件名称
import logging.handlers
# 日志工具
class GetLog:
@classmethod
def get_log(cls):
cls.log = None
if cls.log is None:
# 1、获取日志器
cls.log = logging.getLogger()
# 设置日志级别 info
cls.log.setLevel(logging.INFO)
filepath = DIR_PATH + os.sep + "log" + os.sep + "p2p.log"
# 2、获取处理器 TimedRotatingFileHandler：日志保存到文件且根据时间去分割
tf = logging.handlers.TimedRotatingFileHandler(filename=filepath,
when="midnight",
interval=1,
backupCount=3,
encoding="utf-8")
# 3、获取格式器
fmt = "%(asctime)s %(levelname)s [%(filename)s(%(funcName)s:%(lineno)d)] - %(message)s"
fm = logging.Formatter(fmt)
# 4、将格式器添加到处理器中
tf.setFormatter(fm)
# 5、将处理器添加到日志器中
cls.log.addHandler(tf)
# 返回日志器
return cls.log
# 下面只是测试上面的工具能不能用的，如果运行成功，则会在log目录下产生p2p.log
# p2p.log文件内容类似为：2025-01-14 22:30:41,013 INFO [util.py(<module>:50)] - 信息级别测试
if __name__ == '__main__':
GetLog.get_log().info("信息级别测试")②工具使用
（为什么这样用： Python中的`__init__.py`文件 -CSDN博客）
应用的级别：info、error
- info：i记录程序运行的步骤
- error：记录程序错误
标注地方：api和script
api层：记录程序步骤
script:记录程序执行结果、断言结果、错误原因
可以在api和script中每个方法中标注log，下面仅为示例：
# script目录中文件标注的示例
#2、获取短信验证码接口 测试
@parameterized.expand(read_json("register_login.json", "phone_code"))
def test02_phone_code(self,phone,imgVerifyCode,expec_text):
try:
# 1、调用获取图片验证码接口 -- 目的:让session对象记录cookie
# 调用接口后session会自动记录cookie
self.reg.api_img_code(234)
# 2、调用短信验证码接口
r = self.reg.api_phone_code(phone=phone,imgVerifyCode=imgVerifyCode)
# 3、查看响应结果
log.info("执行接口结果为：{}".format(r.text))
self.assertIn(expec_text,r.text) # 使用text提取结果是更方便，json还要根据键找值
log.info("执行断言通过!")
except Exception as err:
# 日志
# print(err)
log.error("断言失败，原因：{}".format(err))
# 抛异常
raise# api目录中的示例
# 2、获取短信验证码接⼝ 封装
def api_phone_code(self,phone,imgVerifyCode):
data = {
"phone": phone,
"imgVerifyCode": imgVerifyCode,
"type": "reg"
}
log.info("正在调用获取短信验证码接口，请求方法：{}, 请求url:{} 请求参数：{}".format("post", self.__url_phone_code,data))
return self.session.post(url=self.__url_phone_code,data=data)二、认证开户接口脚本编写
1、代码编写
1️⃣api目录
①结构
from config import HOST
class ApiApproveTrust:
# 初始化
def __init__(self, session):
# 获取session对象
self.session = session
# 认证url
self.__url_approve = HOST + "/member/realname/approverealname"
# 查询认证状态url
self.__url_approve_status = HOST + "/member/member/getapprove"
# 开户url
self.__url_trust = HOST + "/trust/trust/register"
# 图片验证码url
self.__url_img_code = HOST + "/common/public/verifycode/{}"
# 充值url
self.__url_recharge = HOST + "/trust/trust/recharge"
# 1、认证接口 封装
def api_approve(self):
pass
# 2、查询认证状态接口 封装
def api_approve_status(self):
pass
# 3、开户接口 封装
def api_trust(self):
pass
# 4、获取图片验证码接口 封装
def api_img_code(self, random):
pass
# 5、充值接口封装
def api_recharge(self,valicode):
pass
②实现
from config import HOST
class ApiApproveTrust:
# 初始化
def __init__(self, session):
# 获取session对象
self.session = session
# 认证url
self.__url_approve = HOST + "/member/realname/approverealname"
# 查询认证状态url
self.__url_approve_status = HOST + "/member/member/getapprove"
# 开户url
self.__url_trust = HOST + "/trust/trust/register"
# 图片验证码url
self.__url_img_code = HOST + "/common/public/verifycode/{}"
# 充值url
self.__url_recharge = HOST + "/trust/trust/recharge"
# 1、认证接口 封装
def api_approve(self):
# 1、请求参数
data = {
"realname":"华仔",
"card_id":"350102199003072237"
}
# 2、调用请求方法 难题：multipart/form-data使用：data+files来实现多消息体类型
return self.session.post(url=self.__url_approve, data=data, files={"x": "y"})
# 2、查询认证状态接口 封装
def api_approve_status(self):
return self.session.post(url=self.__url_approve_status)
# 3、开户接口 封装
def api_trust(self):
return self.session.post(url=self.__url_trust)
# 4、获取图片验证码接口 封装
def api_img_code(self, random):
return self.session.get(url=self.__url_img_code.format(random))
# 5、充值接口封装
def api_recharge(self,valicode):
# 1、请求参数
data = {
"paymentType": "chinapnrTrust",
"amount": "1000",
"formStr":"reForm",
"valicode":valicode
}
# 2、调用请求方法
return self.session.post(url=self.__url_recharge, data=data)
难点：认证接口请求参数类型为：multipart/form-data多消息类型，如何实现？
- 解决：请求使用data+files两种参数格式，消息头会自动切换到multipart即可。
- 示例：self.session.post(url=self.__url_approve, data=data, files={"x": "y"})
- files={"x": "y"}只是起到占位作用，表明有文件而已。
2️⃣script目录
①结构
import unittest
import requests
from api.api_register_login import ApiRegisterLogin
class TestApproveTrust(unittest.TestCase):
# 初始化
def setUp(self) -> None:
#1、获取session对象
self.session = requests.Session()
#2、实例化ApiRegisterLogin类方法
self.approve = ApiRegisterLogin(self.session)
#3、调用登录接口
ApiRegisterLogin(self.session).api_login()
# 结束
def tearDown(self) -> None:
self.session.close()
#1、认证接口 测试
def test01_approve(self):
pass
#2、查询认证状态接口 测试
def test02_approve_status(self):
pass
#3、开户接口 测试
def test03_trust(self):
pass
#4、获取图片验证码接口 测试
def test04_img_code(self):
pass
#5、充值接口 测试
def test05_recharge(self):
pass②实现
请求第三方开户接口 和 请求第三方充值接口 的代码需要后文的BeautifulSoup库知识。
此处是加了参数化的代码，参数化的其他内容看后文参数化小节。
import unittest
import requests
from parameterized import parameterized
from api import log
from api.api_approve_trust import ApiApproveTrust
from api.api_register_login import ApiRegisterLogin
from util import parser_html, read_json
class TestApproveTrust(unittest.TestCase):
# 初始化
def setUp(self) -> None:
#1、获取session对象
self.session = requests.Session()
#2、实例化ApiRegisterLogin类方法
self.approve = ApiApproveTrust(self.session)
#3、调用登录接口
ApiRegisterLogin(self.session).api_login()
# 结束
def tearDown(self) -> None:
r = self.session.close()
# 1、认证接口 测试
# 认证接口不需要参数化的原因是：只有一个可测的用例，其他两个测试用例是bug，不适合拿来做实战
def test01_approve(self,expect_test="提交成功"):
try:
r = self.approve.api_approve()
# print(r.json())
log.info("正在执行认证接口响应结果为：{}".format(r.text))
self.assertIn(expect_test,r.text)
log.info("认证接口断言成功！")
except Exception as err:
# 日志
log.error("断言失败，原因：{}".format(err))
# 抛异常
raise
#2、查询认证状态接口 测试
# 不需要参数化的原因是：只有一个可测的用例，
def test02_approve_status(self,expect_test="华"):
try:
r = self.approve.api_approve_status()
# print(r.json())
log.info("正在执行查询认证状态接口响应结果为：{}".format(r.text))
self.assertIn(expect_test, r.text)
log.info("查询认证状态接口断言成功！")
except Exception as err:
# 日志
log.error("断言失败，原因：{}".format(err))
# 抛异常
raise
#3、开户接口 测试
def test03_trust(self,expect_test="form"):
try:
# 1、请求后台开户接口
r = self.approve.api_trust()
# print(r.json())
log.info("正在执行开户接口响应结果为：{}".format(r.json()))
self.assertIn(expect_test, r.text)
log.info("认证开户断言成功！")
# 2、请求第三方开户接口
result = parser_html(r) # 结果为：（’http://xxxx‘，‘{'xxx':'xxx'},……’）
r = self.session.post(url=result[0],data=result[1])
# print(r.text) # 结果为：UserRegister OK
self.assertIn("OK",r.text)
log.info("请求第三方开户断言成功！")
except Exception as err:
# 日志
log.error("断言失败，原因：{}".format(err))
# 抛异常
raise
#4、获取图片验证码接口 测试
@parameterized.expand(read_json("approve_trust.json","img_code"))
def test04_img_code(self,random,expect_code):
try:
r = self.approve.api_img_code(random)
# print(r.status_code)
log.info("正在执行获取图片验证码接口响应结果为：{}".format(r.status_code))
self.assertEqual(expect_code, r.status_code)
log.info("断言获取图片验证码接口成功！")
except Exception as err:
# 日志
log.error("断言失败，原因：{}".format(err))
# 抛异常
raise
#5、充值接口 测试
@parameterized.expand(read_json("approve_trust.json","recharge"))
def test05_recharge(self,valicode,expect_text):
try:
# 1、调用图片验证码接口
self.approve.api_img_code(123)
# 2、充值接口
r = self.approve.api_recharge(valicode)
# print("充值接口响应结果：",r.text)
log.info("正在执行充值接口响应结果为：{}".format(r.json()))
if valicode == 8888:
# # 断言
self.assertIn("form", r.text)
log.info("断言充值接口成功！")
# 3、三方充值
result = parser_html(r) # 结果为：（’http://xxxx‘，‘{'xxx':'xxx'},……’）
r = self.session.post(url=result[0], data=result[1])
self.assertIn(expect_text, r.text)
log.info("请求第三方充值断言成功！")
else:
# print("验证码错误的响应结果：",r.text)
self.assertIn(expect_text,r.text)
except Exception as err:
# 日志
log.error("断言失败，原因：{}".format(err))
# 抛异常
raise
2、BeautifulSoup库
1️⃣简介及例子
说明：⼀个python解析html/xml的三方库
安装： pip install beautifulsoup4 -i https://mirrors.tuna.tsinghua.edu.cn/
基本用法：
应用步骤：
- 1、导包
- 2、实例化
- 3、调用方法
例子：
重点:
1、查找所有标签 bs.find_all("标签名") == 元素的集合 == ["元素1","元素2"]
2、查找属性 元素.get("属性名")
其他方法：
2️⃣提取html数据工具封装
①思路：
②实现
在工具类py文件末尾添加下面的封装方法
from bs4 import BeautifulSoup
def parser_html(result):
# 1、提取html
html = result.json().get("description").get("form")
# 2、获取bs对象
bs = BeautifulSoup(html,"html.parser")
# 3、提取url
url = bs.form.get("action")
data = {}
# 4、查找所有的input标签
for input in bs.find_all("input"):
data[input.get("name")]=input.get("value")
return url, data使用BeautifulSoup库后就可以实现三方开户和三方充值了，代码已经在前文2️⃣script目录展示过。
3、认证开户参数化
实现难点:
{
"img_code": [
{
"desc": "获取图片验证码成功（随机小数）",
"random": 0.123,
"expect_code": 200
},
{
"desc": "获取图片验证码成功（随机整数）",
"random": 123,
"expect_code": 200
},
{
"desc": "获取图片验证码失败（随机数为空）",
"random": "",
"expect_code": 404
},
{
"desc": "获取图片验证码失败（随机数为字符串）",
"random": "123hello",
"expect_code": 400
}
],
"recharge": [
{
"desc": "后台充值响应成功",
"valicode": 8888,
"expect_text": "OK"
},
{
"desc": "后台充值响应成功",
"valicode": 8889,
"expect_text": "验证码错误"
}
]
}添加了参数化的完整认证开户script文件已在前文展示：2️⃣script目录
![【BUUCTF】[GXYCTF2019]BabySQli](https://i-blog.csdnimg.cn/direct/91cc7946a56f41fea8fa26322d812cac.png)
