KaneAI - World’s First E2E Software Testing Agent.Book a Demo Join TestMu Conference

How to use test_save_screenshot method in SeleniumBase

Best Python code snippet using SeleniumBase

screenshot_test.py

Source:screenshot_test.py

...31        try:32            self.assertNone(_get_filename("A", ".png"), "expected None")33        finally:34            screenshot_module.listdir = orig35    def test_save_screenshot(self):36        self.window = Window(37            width=12, height=34, visible=False,38            caption='test_save_screenshot',39        )40        import solescion.utils.screenshot as screenshot_module41        orig_get = screenshot_module._get_filename42        filename = join(expanduser("~"), "test_save_screenshot.png")43        screenshot_module._get_filename = lambda *_: filename44        try:45            try:46                save_screenshot(self.window)47            finally:48                screenshot_module._get_filename = orig_get49            png = image.load(filename)...

test_save_screenshots.py

Source:test_save_screenshots.py

1import os2from seleniumbase import BaseCase3class ScreenshotTests(BaseCase):4    def test_save_screenshot(self):5        self.open("https://seleniumbase.io/demo_page")6        # "./downloaded_files" is a special SeleniumBase folder for downloads7        self.save_screenshot("demo_page.png", folder="./downloaded_files")8        self.assert_downloaded_file("demo_page.png")9        print('\n"%s/%s" was saved!' % ("downloaded_files", "demo_page.png"))10    def test_save_screenshot_to_logs(self):11        self.open("https://seleniumbase.io/demo_page")12        self.save_screenshot_to_logs()13        # "self.log_path" is the absolute path to the "./latest_logs" folder.14        # Each test that generates log files will create a subfolder in there15        test_logpath = os.path.join(self.log_path, self.test_id)16        expected_screenshot = os.path.join(test_logpath, "_1_screenshot.png")17        self.assert_true(os.path.exists(expected_screenshot))18        print('\n"%s" was saved!' % (expected_screenshot))...

screenshots.py

Source:screenshots.py

2from selenium import webdriver3class SaveScreenshots(unittest.TestCase):4    def setUp(self):5        self.driver = webdriver.Chrome()6    def test_save_screenshot(self):7        self.driver.get('https://letskodeit.teachable.com/')8        self.driver.maximize_window()9        self.driver.implicitly_wait(3)10        self.driver.find_element_by_link_text('Login').click()11        self.driver.find_element_by_id('user_email').send_keys('abc@email.com')12        self.driver.find_element_by_id('user_password').send_keys('test')13        self.driver.find_element_by_name('commit').click()14        destination_file_name = 'D:\\pratima\\tutorials\\QA-testing\\test.png'15        try:16            self.driver.save_screenshot(destination_file_name)17            print('Screenshot saved to directory ' + destination_file_name)18        except NotADirectoryError:19            print('Not a directory issue')20    def tearDown(self):...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.