1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
|
#!/usr/bin/python
# Copyright (c) 2010 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
try:
import json
except ImportError: # < 2.6
import simplejson as json
import os
import optparse
import platform
import string
import subprocess
import sys
import time
import types
import unittest
import urllib2
from selenium.remote.webdriver import WebDriver
from selenium.common.exceptions import ErrorInResponseException
from urlparse import urlparse
if sys.version_info < (2,6):
# Subprocess.Popen.kill is not available prior to 2.6.
if platform.system() == 'Windows':
import win32api
else:
import signal
WEBDRIVER_EXE = os.path.abspath(os.path.join('.', 'chromedriver'))
if platform.system() == 'Windows':
WEBDRIVER_EXE = '%s.exe' % WEBDRIVER_EXE
WEBDRIVER_PORT = 8080
WEBDRIVER_SERVER_URL = None
WEBDRIVER_PROCESS = None
if not WEBDRIVER_SERVER_URL:
WEBDRIVER_SERVER_URL = 'http://localhost:%d' % WEBDRIVER_PORT
class RemoteWebDriverTest(unittest.TestCase):
"""A new instance of chrome driver is started for every test case"""
def setUp(self):
global WEBDRIVER_SERVER_URL
global WEBDRIVER_PROCESS
WEBDRIVER_PROCESS = subprocess.Popen([WEBDRIVER_EXE,
'--port=%d' % WEBDRIVER_PORT])
if WEBDRIVER_PROCESS == None:
print "Chromium executable not found. The path used was: "
print WEBDRIVER_EXE
sys.exit(-1)
time.sleep(5)
self.driver = WebDriver.WebDriver(WEBDRIVER_SERVER_URL, "chrome", "ANY")
self.assertTrue(self.driver)
def tearDown(self):
global WEBDRIVER_PROCESS
self.driver.quit()
if WEBDRIVER_PROCESS:
if sys.version_info < (2,6):
# From http://stackoverflow.com/questions/1064335
if platform.system() == 'Windows':
PROCESS_TERMINATE = 1
handle = win32api.OpenProcess(PROCESS_TERMINATE, False,
WEBDRIVER_PROCESS.pid)
win32api.TerminateProcess(handle, -1)
win32api.CloseHandle(handle)
else:
os.kill(WEBDRIVER_PROCESS.pid, signal.SIGKILL)
else:
WEBDRIVER_PROCESS.kill()
WEBDRIVER_PROCESS = None
"""Preforms a string search ignoring case"""
def assertFind(self, text, search):
text = string.lower(text)
search = string.lower(search)
self.assertNotEqual(-1, string.find(text, search))
class TestJavaScriptExecution(RemoteWebDriverTest):
""" Test the execute javascript ability of the remote driver"""
def testNoModification(self):
self.driver.get("http://www.google.com")
title = self.driver.execute_script("return document.title")
self.assertEqual(title, self.driver.get_title())
def testModification(self):
self.driver.get("http://www.google.com")
title = self.driver.get_title()
self.assertFind(title, "google")
r = self.driver.execute_script("return document.Foo")
self.assertTrue(r == None)
self.driver.execute_script("document.Foo = \"Hello\"")
r = self.driver.execute_script("return document.Foo")
self.assertTrue(r == "Hello")
def testComplexObjectFetch(self):
self.driver.get("http://www.google.com")
loc = self.driver.execute_script("return window.location")
self.assertTrue(type(loc) is types.DictType)
self.assertTrue(loc.has_key("hostname"))
self.assertFind(loc["hostname"], "google")
def testJavascriptExeception(self):
self.driver.get("http://www.google.com")
self.assertRaises(ErrorInResponseException, self.driver.execute_script,
"return windows");
def testJavascriptWithNoReturn(self):
self.driver.get("http://www.google.com")
try:
ret = self.driver.execute_script("return window.foobar")
self.assertTrue(type(ret) is types.NoneType)
except:
self.assertTrue(False)
class TestNavigation(RemoteWebDriverTest):
SEARCH = "http://www.google.com/webhp?hl=en"
NEWS = "http://www.google.com/news?hl=en"
"""Verifies that navigation to a specific page is correct by asserting on the
the reported URL. The function will not work with pages that redirect."""
def navigate(self, url):
self.driver.get(url)
self.assertURL(url)
def assertURL(self, url):
u = self.driver.get_current_url()
self.assertEqual(u, url)
def testNavigateToURL(self):
# No redirects are allowed on the google home page.
self.navigate(self.SEARCH)
def testGoBackWithNoHistory(self):
# Go back one page with nothing to go back to.
self.assertRaises(ErrorInResponseException, self.driver.back)
def testGoForwardWithNoHistory(self):
# Go forward with nothing to move forward to.
self.assertRaises(ErrorInResponseException, self.driver.forward)
def testNavigation(self):
# Loads two pages into chrome's navigation history.
self.navigate(self.NEWS)
self.navigate(self.SEARCH)
# Go back to news.
self.driver.back()
self.assertURL(self.NEWS)
# Move forward to search.
self.driver.forward()
self.assertURL(self.SEARCH)
# Verify refresh keeps us on the current page.
self.driver.refresh()
self.assertURL(self.SEARCH)
# Go back to the previous URL and refresh making sure
# we dont change the page.
self.driver.back()
self.assertURL(self.NEWS)
self.driver.refresh()
self.assertURL(self.NEWS)
def testGetTitle(self):
self.navigate(self.SEARCH)
title = self.driver.get_title()
# The google name must always be in the search title.
self.assertFind(title, u"google")
def testGetSource(self):
self.navigate(self.SEARCH)
source = self.driver.get_page_source()
self.assertFind(source, u"document.body.style") # Basic javascript.
self.assertFind(source, u"feeling lucky") # Lucky button.
self.assertFind(source, u"google search") # Searh button.
if __name__ == '__main__':
parser = optparse.OptionParser('%prog [options]')
parser.add_option('-u', '--url', dest='url', action='store',
type='string', default=None,
help=('Specifies the URL of a remote WebDriver server to '
'test against. If not specified, a server will be '
'started on localhost according to the --exe and '
'--port flags'))
parser.add_option('-e', '--exe', dest='exe', action='store',
type='string', default=None,
help=('Path to the WebDriver server executable that should '
'be started for testing; This flag is ignored if '
'--url is provided for a remote server.'))
parser.add_option('-p', '--port', dest='port', action='store',
type='int', default=8080,
help=('The port to start the WebDriver server executable '
'on; This flag is ignored if --url is provided for a '
'remote server.'))
(options, args) = parser.parse_args()
# Strip out our flags so unittest.main() correct parses the remaining.
sys.argv = sys.argv[:1]
sys.argv.extend(args)
if options.url:
WEBDRIVER_SERVER_URL = options.url
else:
if options.port:
WEBDRIVER_PORT = options.port
if options.exe:
WEBDRIVER_EXE = options.exe
if not os.path.exists(WEBDRIVER_EXE):
parser.error('WebDriver server executable not found:\n\t%s\n'
'Please specify a valid path with the --exe flag.'
% WEBDRIVER_EXE)
unittest.main()
|
