一学就会，Allure报告添加用例失败截图

lsekfe

前言
　　使用Selenium 做 Web 自动化的时候，很多小伙伴希望用例失败的时候能截图，把异常截图展示到allure报告里面。
　　Pytest 有个很好的钩子函数 pytest_runtest_makereport 可以获取到用例执行的结果，所以我们在这个钩子函数里面判断用例失败后截图就可以了。
　　Allure报告添加截图可以使用 allure.attach 方法
　　失败截图
　　关于 pytest_runtest_makereport 获取用例执行结果的钩子函数介绍可以看前面这篇https://www.cnblogs.com/yoyoketang/p/12609871.html
　　在 conftest.py 文件写用例执行的钩子函数:

1. <font size="3">import pytest
   
2. 
   
3. 　　from selenium import webdriver
   
4. 
   
5. 　　import os
   
6. 
   
7. 　　import allure
   
8. 
   
9. 　　# 作者-上海悠悠 QQ交流群:7172XXXX
   
10. 
    
11. 　　# blog地址 https://www.cnblogs.com/yoyoketang/
    
12. 
    
13. 　　_driver = None
    
14. 
    
15. 　　@pytest.hookimpl(tryfirst=True, hookwrapper=True)
    
16. 
    
17. 　　def pytest_runtest_makereport(item, call):
    
18. 
    
19. 　　    '''
    
20. 
    
21. 　　    获取每个用例状态的钩子函数
    
22. 
    
23. 　　    :param item:
    
24. 
    
25. 　　    :param call:
    
26. 
    
27. 　　    :return:
    
28. 
    
29. 　　    '''
    
30. 
    
31. 　　    # 获取钩子方法的调用结果
    
32. 
    
33. 　　    outcome = yield
    
34. 
    
35. 　　    rep = outcome.get_result()
    
36. 
    
37. 　　    # 仅仅获取用例call 执行结果是失败的情况, 不包含 setup/teardown
    
38. 
    
39. 　　    if rep.when == "call" and rep.failed:
    
40. 
    
41. 　　        mode = "a" if os.path.exists("failures") else "w"
    
42. 
    
43. 　　        with open("failures", mode) as f:
    
44. 
    
45. 　　            # let's also access a fixture for the fun of it
    
46. 
    
47. 　　            if "tmpdir" in item.fixturenames:
    
48. 
    
49. 　　                extra = " (%s)" % item.funcargs["tmpdir"]
    
50. 
    
51. 　　            else:
    
52. 
    
53. 　　                extra = ""
    
54. 
    
55. 　　            f.write(rep.nodeid + extra + "\n")
    
56. 
    
57. 　　        # 添加allure报告截图
    
58. 
    
59. 　　        if hasattr(_driver, "get_screenshot_as_png"):
    
60. 
    
61. 　　            with allure.step('添加失败截图...'):
    
62. 
    
63. 　　                allure.attach(_driver.get_screenshot_as_png(), "失败截图", allure.attachment_type.PNG)
    
64. 
    
65. 　　@pytest.fixture(scope='session')
    
66. 
    
67. 　　def browser():
    
68. 
    
69. 　　    global _driver
    
70. 
    
71. 　　    if _driver is None:
    
72. 
    
73. 　　        _driver =webdriver.Chrome()
    
74. 
    
75. 　　    yield _driver
    
76. 
    
77. 　　    print("1111111111")
    
78. 
    
79. 　　    _driver.quit()</font>

复制代码

　用例运行　　整个目录结构如下:

　　test_login.py文件内容如下:

1. import pytest
   
2. 
   
3. 　　from selenium import webdriver
   
4. 
   
5. 　　import allure
   
6. 
   
7. 　　# 作者-上海悠悠 QQ交流群:71722XXX
   
8. 
   
9. 　　# blog地址 https://www.cnblogs.com/yoyoketang/
   
10. 
    
11. 　　def test_login(browser):
    
12. 
    
13. 　　    with allure.step("step1：打开登录首页"):
    
14. 
    
15. 　　        browser.get("http://ip:6009/admin/login/?next=/admin/")
    
16. 
    
17. 　　    with allure.step("step2：输入账号：admin"):
    
18. 
    
19. 　　        browser.find_element_by_name("username").send_keys("admin")
    
20. 
    
21. 　　    with allure.step("step2：输入密码：123456"):
    
22. 
    
23. 　　        browser.find_element_by_name("password").send_keys("123456")
    
24. 
    
25. 　　    # 故意断言失败，看是否会截图
    
26. 
    
27. 　　    assert browser.title == "悠悠"

复制代码

　运行用例后，截图会存到./report 报告目录，Allure报告展示: