headless Selenium WebDriver tests. Python unittest launching browser inside Xvfb.

test_xvfb_selenium.py

#!/usr/bin/env python
#
# Corey Goldberg - 2012
#
# requires:
#   * Xvfb
#   * X Windows
#   * xvfbwrapper (pip install xvfbwrapper)
#

from selenium import webdriver
from xvfbwrapper import Xvfb

import unittest



class TestHomepages(unittest.TestCase):

    def setUp(self):
        self.vdisplay = Xvfb(width=1280, height=720)
        self.vdisplay.start()
        self.browser = webdriver.Firefox()

    def testUbuntuHomepage(self):
        self.browser.get('http://www.ubuntu.com')
        self.assertIn('Ubuntu', self.browser.title)

    def testGoogleHomepage(self):
        self.browser.get('http://www.google.com')
        self.assertIn('Google', self.browser.title)

    def tearDown(self):
        self.browser.quit()
        self.vdisplay.stop()



if __name__ == '__main__':
    unittest.main(verbosity=2)