在pytest框架中,利用recwarn这个内置的fixture,可以对测试用例产生的所有告警进行记录,并可以在最后进行统一的解析处理,比如如下代码,这里人工产生两条不同类型的告警,UserWarning和SyntaxWarning类型的告警。在测试代码最后,对当前测试用例中产生的告警条数进行了断言,并且打印了告警的信息,包括类型、告警信息,产生告警的文件以及产生告警的代码行数。
import warnings
def test_demo(recwarn):
warnings.warn(UserWarning("user warning test ..."))
warnings.warn(SyntaxWarning("syntax warning demo ..."))
assert len(recwarn) == 2
w = recwarn.pop()
print("---------------------------")
print(w.category)
print(w.message)
print(w.filename)
print(w.lineno)
print("---------------------------")
w = recwarn.pop()
print(w.category)
print(w.message)
print(w.filename)
print(w.lineno)
print("---------------------------")
执行结果如下,可以看出,解析的时候是按照从前往后的顺序解析的,即可以理解为告警是存放在一个先进先出的队列中。
(demo-HCIhX0Hq) E:\demo>pytest -s
=================== test session starts ===================
platform win32 -- Python 3.7.9, pytest-7.2.0, pluggy-1.0.0
rootdir: E:\demo, configfile: pytest.ini
plugins: assume-2.4.3, rerunfailures-10.2
collected 1 item
test_demo.py ---------------------------
<class 'UserWarning'>
user warning test ...
E:\demo\test_demo.py
5
---------------------------
<class 'SyntaxWarning'>
syntax warning demo ...
E:\demo\test_demo.py
6
---------------------------
.
==================== 1 passed in 0.02s ====================
(demo-HCIhX0Hq) E:\demo>
有了这个功能其实就可以做很多事情了,比如在用例执行后对产生的告警进行处理,比如可以根据告警类型或者告警信息关键字提取等,对特定的告警进行特殊的处理,比如报错等。