Capture screenshot of failed test cases

Question

How can I capture a screenshot of the failed test cases in Selenium + Python? I am able to get screenshots of the page but is facing difficulty in capturing screenshots of faile

Answer 1

You can do like:

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

Hope it helps you!