今天在使用global全局变量解决数据依赖的问题时,睬了一个大坑。unittest框架执行顺序的问题,unittest默认加载脚本的顺序是:根据ASCII码的顺序加载,数字与字母的顺序为:0-9,A-Z,a-z。所以以A开头的测试用例方法会优先执行,以a开头会后执行。
前面不知道这个规则,导致报错如下:
AttributeError: 'NoneType' object has no attribute 'group'
产生这个报错的原因是,我第二条用例需要用到登录的cookie,结果第二条用例却先执行,导致cookie传递为空。最后解决办法:调整运行顺序,改变一下用例名称,让登陆用例先执行。
看下图先运行的第二条用例,所以打印的cookie为空:
有下图代码可知:
定义了一个全局变量,COOKIE=None,然后将登录的cookies进行了一个替换操作。即COOKIE=login_res.cookies。然后在下一条用例,获取我的精品页面信息的时候,将cookies=COOKIE传递到了http请求里面。从而解决了下一个请求依赖于上一个请求的返回值的问题。
全部代码如下:
import unittest from API_AUTO.tools.http_request import HttpRequest import re COOKIE = None class TestHttp(unittest.TestCase): def setUp(self): pass # abcdefghijklmnopqrstuvwxyz def test_Normal_login(self): '''正常登录''' global COOKIE url = 'https://www.ketangpai.com/UserApi/login' data = { "email": "1489088761@qq.com", "password": "A156829", "remember": 0 } headers = { "User-Agent": " Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/78.0.3904.108 Safari/537.36", "Content-Type": "application/x-www-form-urlencoded", } login_res = HttpRequest().http_request('post', url, data, headers) if login_res.cookies: COOKIE = login_res.cookies print(COOKIE) try: self.assertEqual(1, login_res.json()['status'], '登录失败') except AssertionError as e: print('登录错误{}'.format(e)) raise e def test_mooc(self): '''我的精品页面''' global COOKIE # print(COOKIE) url1 = 'https://www.ketangpai.com/Mooc/Mooc/index.html' headers1 = { "Referer": "https: // www.ketangpai.com / Main / index.html", "User-Agent": " Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/78.0.3904.108 Safari/537.36", } print(COOKIE) res1 = HttpRequest().http_request('get', url=url1, headers=headers1, cookies=COOKIE) try: pattern = '<img class=.*?salt=(".*?").*?>' regular = re.search(pattern, res1.text, re.S) self.assertEqual('夏茂杰', eval(regular.group(1)), '进入我的界面失败') except Exception as e: print('错误是{}'.format(e)) raise e def tearDown(self): pass if __name__ == '__main__': unittest.main()