diff --git a/xvfbwrapper/test_xvfb_selenium.py b/xvfbwrapper/test_xvfb_selenium.py index e9400fc..f5e2ab8 100644 --- a/xvfbwrapper/test_xvfb_selenium.py +++ b/xvfbwrapper/test_xvfb_selenium.py @@ -1,19 +1,25 @@ +#!/usr/bin/env python + +from selenium import webdriver from xvfbwrapper import Xvfb import unittest -from selenium import webdriver -class TestUbuntuHomepage(unittest.TestCase): +class TestHomepages(unittest.TestCase): def setUp(self): self.vdisplay = Xvfb(width=1280, height=720) self.vdisplay.start() self.browser = webdriver.Firefox() - - def testTitle(self): + + 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()