Захватить скриншот неудачных тестов

Question

Как сделать снимок экрана с ошибочными тестами в Selenium + Python?Я могу получить скриншоты страницы, но сталкиваюсь с трудностями при захвате скриншотов неудачных тестов.

Код:

def test_2new(self):

   driver = self.driver driver.get("google.com") 
   driver.find_element_by_name("q").send_keys("Automation") 
   driver.set_page_load_timeout(10) 
   driver.find_element_by_name("btnK").click() 
   self.assertTrue("something.com" in self.driver.current_url) 
   print("Google Search") 

@classmethod
def tearDownClass(cls): 

    # cls.driver.refresh() 
    time.sleep(5) 
    if sys.exc_info()[0]: 
         cls.driver.save_screenshot(r"C:\Users\ANCP2\sample.png") 

Answer 1

Вы можете сделать как:

def test_2new(self):

    driver = self.driver 
    driver.get("google.com")
    driver.set_page_load_timeout(10)
    driver.find_element_by_name("q").send_keys("Automation")  
    driver.find_element_by_name("btnK").click()

    try:

        self.assertTrue("something.com" in self.driver.current_url) 
        print("Google Search!")

    except AssertionError:

        driver.save_screenshot(r"C:\Users\ANCP2\sample.png")

Надеюсь, это поможет вам!