如何捕获有关Python单元测试的测试用例失败的屏幕截图

Question

我正在将Python 3.6.5与以下库一起使用：

• Appium的Python客户端== 0.26
• unittest2 == 1.1.0
• 硒== 3.5.0
• pytest == 3.6.3

现在，在测试失败的情况下，我需要捕获屏幕快照，因此，我故意放置了一条错误的语句self.driver.find_element_by_css_selector('test') 。

我正在使用sys.exc_info() 。 但是，当我使用以下命令执行以下代码时： py.test untitled.py或python3 -m unittest untitled.py它无法捕获它。

码：

import sys, time, unittest2
from selenium import webdriver

class FB360(unittest2.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()

    def test_user_can_login(self):
            self.driver.get('https://www.google.co.in')
            self.driver.find_element_by_css_selector('test')

    def tearDown(self):
        print(sys.exc_info())
        if sys.exc_info()[0]:
            test_method_name = self._testMethodName
            self.driver.save_screenshot(test_method_name + str(time.time()) + '.png')
        self.driver.quit()

if __name__ == '__main__':
    unittest2.main()

O / P：

（无，无，无）

Ë

================================================== ===================错误：test_user_can_login（untitled.FB360）

---

追溯（最近一次通话）：文件“ /Volumes/Harry/Projects/pythonScreenshots/untitled.py”，第11行，位于test_user_can_login self.driver.find_element_by_css_selector（'test'）文件中，“ / Library / Frameworks / Python.framework /在find_element_by_css_selector中的版本/3.6/lib/python3.6/site-packages/selenium/webdriver/remote/webdriver.py“中，行498返回self.find_element（by = By.CSS_SELECTOR，value = css_selector）文件” / Library / Frameworks / Python.framework / Versions / 3.6 / lib / python3.6 / site-packages / selenium / webdriver / remote / webdriver.py“，第832行，位于find_element'value'：value}）['value'] File” /Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/site-packages/selenium/webdriver/remote/webdriver.py“，第297行，在执行self.error_handler.check_response（response）File” / Library / Frameworks / Python.framework / Versions / 3.6 / lib / python3.6 / site-packages / selenium / webdriver / remote / errorhandler.py“，第194行，在check_response中引发exception_class（消息，屏幕，stacktrace）selenium.common 。 exceptions.NoSuchElementException：消息：没有这样的元素：无法找到元素：{“方法”：“ css选择器”，“选择器”：“测试”}（会话信息：chrome = 67.0.3396.99）（驱动程序信息：chromedriver = 2.40 .565386（45a059dc425e08165f9a10324bd1380cc13ca363），平台= Mac OS X 10.13.5 x86_64）

我的创立：

• print(sys.exc_info())打印：（无，无，无）。 这意味着即使我收到以下异常，我的测试用例也已通过：

selenium.common.exceptions.NoSuchElementException：消息：没有这样的元素：无法找到元素：{“方法”：“ css选择器”，“选择器”：“测试”}

• 由于sys.exc_info())返回（None，None，None）在if条件下不会执行

因此，我主要担心的是，即使我收到NoSuchElementException，为什么sys.exc_info()) ）也会返回（无，无，无）？

Answer 1

我用它来拍摄失败的测试的屏幕截图

@pytest.fixture()
def browser(request):
    options = Options()
    options.add_argument("--headless")  # Runs Chrome in headless mode.
    _chrome = webdriver.Chrome(options=options)
    failed_before = request.session.testsfailed
    yield _chrome
    if request.session.testsfailed != failed_before:
        test_name = request.node.name
        take_screenshot(_chrome, test_name)
    _chrome.close()


def take_screenshot(browser, test_name):
    screenshots_dir = "path_to/functional_tests/failure_screenshots"
    screenshot_file_path = "{}/{}.png".format(screenshots_dir, test_name)
    browser.save_screenshot(
        screenshot_file_path
    )