Wie Methode bei jedem py.test Assertion Fehler aufrufen?

Question

Hintergrund:

Ich benutze py.test zusammen mit Pytest-Selen, jetzt möchte ich einen Screenshot der Seite machen, wenn Assertion fehlschlägt.

Zur Zeit habe ich definiere kleine Helfer-Methode in meiner Basisseite Objektklasse:

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) 
     ) 

Das Problem ist, dass ich es anstelle von „manuell“ Nutzung automatisierten Mechanismus machen möchte: (Testklasse Beispiel) :

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') 

Wie bestimmte Methode für jeden Assertionsfehler ausgeführt wird?

Answer 1

Ich denke, dass screenShotInSelenium Seite Ihnen genug Informationen geben sollte, wie Sie einen Screenshot erstellen, wenn eine Assert-Bedingung erfüllt ist.

Was Ihnen fehlt, ist die Verwendung von @AfterMethod

Answer 2

ich persönlich nicht verwendet haben, aber dies könnte Ihre Lösung sein: https://pytest.org/latest/example/simple.html#writing-well-integrated-assertion-helpers

Auch könnte dies helfen: https://pytest.org/latest/assert.html#advanced-assertion-introspection