如何在每个py.test断言失败时调用方法？

Question

背景：

我将py.test和pytest-selenium一起使用，现在我想在断言失败时拍摄页面的屏幕截图。

当前，我在基本页面对象类中定义了小助手方法：

class PageBase(object):
    def __init__(self,driver):
        self.driver = driver
        self.fake = Factory.create()

    def screenshot(self,name):
        self.driver.save_screenshot(datetime.now().strftime('%Y-%m-%d %H:%M:%S') + 'scr_'+name+'.png')

    @contextmanager
    def wait_for_page_load(self, timeout=45):
        old_page = self.driver.find_element_by_tag_name('html')
        yield
        WebDriverWait(self.driver, timeout).until(
            EC.staleness_of(old_page)
        )

问题是我想使其成为自动化的机制，而不是“手动”用法：（测试类示例）：

class TestLogin:
    @allure.feature('Ability to login into admin panel')
    def test_admin_login(self, prepare, page):

        print URLMap.admin('test')
        driver = prepare
        driver.get(URLMap.admin(page))

        login_page = LoginPage(driver)
        assert login_page.is_page_correct(),'Login page not loaded correctly'

        login_page.fill_login_data('testadmin','testadmin')
        login_page.click_login_button()
        assert login_page.is_user_logged_in(),'User cannot log in with provided credentials'
        login_page.screenshot(page+'_logged_in')

如何为每个断言失败运行某种方法？

Answer 1

我认为screenShotInSelenium页面应该为您提供有关在满足断言条件时如何创建屏幕截图的足够信息。

你缺少的是使用@AfterMethod