接续前文，在上篇博客中我们编写了demo.py代码，里面代码过多冗余，更新代码：
#!/usr/bin/env python
# coding=utf-8 import requests class RunMain:
def __init__(self):
pass @staticmethod
def send_post(url,cookies,headers,params=None):
try:
res = requests.post(url=url,cookies=cookies,headers=headers,data=params)return res.status_code
except Exception as msg:
return msg @staticmethod
def send_get(url,cookies,headers,params=None,):
try:
res = requests.get(url=url,cookies=cookies,headers=headers,params=params,)return res.status_code
except Exception as msg:
return msg def run_main(self, url, method,cookies=None,headers=None,params=None):
if method == 'GET':
res = self.send_get(url,cookies,headers,params)
return res
elif method == 'POST':
res = self.send_post(url,cookies,headers,params)
return res
else:
print('不支持的请求方式！')
接下来我们采用Unittest框架进行搭建，使用ddt注解进行参数化。
#!/usr/bin/env python
# -*- coding:utf-8
from ddt import ddt,data,unpack
from Demo.demo import RunMain
import unittest
from urllib import parse
@ddt
class excelddt(unittest.TestCase):
@data(
[1, 'http://apis.juhe.cn/mobile/get', 'GET','{"phone":"13800000000","dtype":"json","key":"密匙"}', '', '', 200, '','Yes'],
[2, 'http://apis.juhe.cn/simpleWeather/query', 'POST','{"city": "济南","key": "密匙"}', '', '', 200, '', 'Yes']
)
@unpack
def testURL(self,id,url,method,params,headers,cookies,code,returns,run):
t=RunMain().run_main(url=url, method=method,headers=headers, params=params)
print(t)
self.assertEqual(t,code) if __name__ == '__main__':
unittest.main()
使用ddt参数化是为了进行excel取值，我们运行一下。
出现错误了，POST请求返回值异常。这个问题困扰了两天，无奈放弃，到最后也没分析出来结果。
更换pytest，用@pytest.mark.parametrize注解试试
import pytest
from Demo.demo import RunMain
@pytest.mark.parametrize(
"ID,URL,METHOD,PARAMS,HEADERS,COOKIES,RETURNCODE,RETURNTXT,YN",
[(1,"http://apis.juhe.cn/mobile/get",'GET',{"phone": "","dtype": "json","key": "x"},{"Content-Type":"application/json"},'',200,'','Yes'),
(2,"http://apis.juhe.cn/simpleWeather/query",'POST',{"city": "济南","key": "x"},{"Content-Type":"application/json"},'',200,'','No'),
(3,"http://apis.juhe.cn/simpleWeather/query",'POST',{"city": "济南","key": "x"},{"Content-Type":"application/json"},'',200,'','Yes'),
(4,"http://www.3wyuming.xyz:8080/BBSProject/userServlet.do",'GET',{"account": "x","pwd": "x"},{"Content-Type":"application/json"},'',200,'','Yes'),
(5,"http://www.3wyuming.xyz:8080/BBSProject/textServlet.do",'GET',{"md":"read","pid":""},{"Content-Type":"application/json"},'',200,'','Yes')],
ids=["case1","case2","case3","case4","case5"]
)
def test_url(ID,URL,METHOD,PARAMS,HEADERS,COOKIES,RETURNCODE,RETURNTXT,YN):
t = RunMain().run_main(url=URL, method=METHOD, headers=HEADERS, params=PARAMS)
assert t == RETURNCODE
if __name__ == '__main__':
pytest.main()
运行代码：
运行成功。
总结一下用参数化遇到的坑：
1.Unittest，ddt参数化，一直报错字符异常，返回值返回的为异常错误，导致断言失败。事后发现应该是中文问题。
2.Pytest，参数化中也是报错，具体错误是str没有item属性，我在Test.py里面发送请求一直正常状态，为什么使用参数化就会异常，转换思路，考虑到数据问题。打入断点，Debug寻找问题，发现Test.py里json格式数据类型都是dict，
但是参数化中属于str类型，于是知道，将数据中的{}括号放置于引号外面，发送请求，成功。