Operate Firefox with Selenium from python and save the screen capture

• Taking a screen capture with Selenium is convenient because you can also save a vertically long WEB screen.
• The selenium.py exported with Firefox's Selenium IDE on Mac worked the same even if I had it on windows.

procedure

Environmental preparation

python

sudo easy_install selenium

Export from Selenium IDE

1. Create a test case with Selenium IDE.

2. Select Python 2 / unitetest / WebDriver in Export

3. Save as selenium.py

Fixed selenium.py

Screen capture is not possible as it is.

ERROR: Caught exception [ERROR: Unsupported command [captureEntirePageScreenshot | /Users/owner/a.jpg | ]]

Add the following under the line

python

        driver.save_screenshot("a.jpg ")

Reference: Seleniumworks: Capture the Screen on Test failure --WebDriver

Run

python

python selenium.py

Confirm that Firefox works and a.jpg is generated.

Reference: Full text of selenium.py

selenium.py

# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re

class Aa(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = "http://auction.yahoo.co.jp/"
        self.verificationErrors = []
        self.accept_next_alert = True

    def test_aa(self):
        driver = self.driver
        driver.get(self.base_url + "/")
        driver.find_element_by_link_text(u"My auction").click()
        driver.find_element_by_link_text(u"help").click()
        driver.find_element_by_link_text(u""The ID or password is incorrect" is displayed").click()
        # ERROR: Caught exception [ERROR: Unsupported command [captureEntirePageScreenshot | /Users/owner/a.jpg | ]]
        driver.save_screenshot("a.jpg ")

    def is_element_present(self, how, what):
        try: self.driver.find_element(by=how, value=what)
        except NoSuchElementException, e: return False
        return True

    def is_alert_present(self):
        try: self.driver.switch_to_alert()
        except NoAlertPresentException, e: return False
        return True

    def close_alert_and_get_its_text(self):
        try:
            alert = self.driver.switch_to_alert()
            alert_text = alert.text
            if self.accept_next_alert:
                alert.accept()
            else:
                alert.dismiss()
            return alert_text
        finally: self.accept_next_alert = True

    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)

if __name__ == "__main__":
    unittest.main()

Support for basic authentication

selenium.py

- self.base_url = "http://secret.page.local/"
+ self.base_url = "http://username:[email protected]/"

xfce4 desktop icon support

google.selenium

#!/bin/sh
python google.py

Prepare something like the above and double-click to execute it.

Prevent firefox from quitting automatically

google.py

+   #def tearDown(self):
+       #self.driver.quit()
+       #self.assertEqual([], self.verificationErrors)