Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 """End to end tests for ChromeDriver.""" | 5 """End to end tests for ChromeDriver.""" |
| 6 | 6 |
| 7 import ctypes | 7 import ctypes |
| 8 import os | 8 import os |
| 9 import sys | 9 import sys |
| 10 import unittest | 10 import unittest |
| 11 | 11 |
| 12 import chromedriver | 12 import chromedriver |
| 13 import webserver | |
| 14 | |
| 15 _THIS_DIR = os.path.abspath(os.path.dirname(__file__)) | |
| 16 sys.path.insert(0, os.path.join(_THIS_DIR, os.pardir, 'pylib')) | |
| 17 | |
| 18 from common import chrome_paths | |
| 13 | 19 |
| 14 | 20 |
| 15 class ChromeDriverTest(unittest.TestCase): | 21 class ChromeDriverTest(unittest.TestCase): |
| 16 """End to end tests for ChromeDriver.""" | 22 """End to end tests for ChromeDriver.""" |
| 17 | 23 |
| 24 @classmethod | |
| 25 def setUpClass(cls): | |
| 26 cls._http_server = webserver.WebServer(chrome_paths.GetTestData()) | |
|
chrisgao (Use stgao instead)
2013/01/05 01:19:10
HTTP server is shared by all testcases.
| |
| 27 | |
| 28 @classmethod | |
| 29 def tearDownClass(cls): | |
| 30 cls._http_server.Shutdown() | |
| 31 | |
| 32 @staticmethod | |
| 33 def GetHttpUrlForFile(file_path): | |
| 34 return ChromeDriverTest._http_server.GetUrl() + file_path | |
| 35 | |
| 36 def setUp(self): | |
|
chrisgao (Use stgao instead)
2013/01/05 01:19:10
Add setUp and tearDown to ensure that chrome windo
| |
| 37 self._driver = chromedriver.ChromeDriver(_CHROMEDRIVER_LIB, _CHROME_BINARY) | |
| 38 | |
| 39 def tearDown(self): | |
| 40 self._driver.Quit() | |
| 41 | |
| 18 def testStartStop(self): | 42 def testStartStop(self): |
| 19 driver = chromedriver.ChromeDriver(_CHROMEDRIVER_LIB, _CHROME_BINARY) | 43 pass |
| 20 driver.Quit() | |
| 21 | 44 |
| 22 def testLoadUrl(self): | 45 def testLoadUrl(self): |
| 23 driver = chromedriver.ChromeDriver(_CHROMEDRIVER_LIB, _CHROME_BINARY) | 46 self._driver.Load(self.GetHttpUrlForFile('/chromedriver/empty.html')) |
| 24 driver.Load('http://www.google.com') | |
| 25 driver.Quit() | |
| 26 | 47 |
| 27 def testEvaluateScript(self): | 48 def testEvaluateScript(self): |
| 28 driver = chromedriver.ChromeDriver(_CHROMEDRIVER_LIB, _CHROME_BINARY) | 49 self.assertEquals(1, self._driver.ExecuteScript('return 1')) |
| 29 self.assertEquals(1, driver.ExecuteScript('return 1')) | 50 self.assertEquals(None, self._driver.ExecuteScript('')) |
| 30 self.assertEquals(None, driver.ExecuteScript('')) | |
| 31 driver.Quit() | |
| 32 | 51 |
| 33 def testEvaluateScriptWithArgs(self): | 52 def testEvaluateScriptWithArgs(self): |
| 34 driver = chromedriver.ChromeDriver(_CHROMEDRIVER_LIB, _CHROME_BINARY) | |
| 35 script = ('document.body.innerHTML = "<div>b</div><div>c</div>";' + | 53 script = ('document.body.innerHTML = "<div>b</div><div>c</div>";' + |
| 36 'return {stuff: document.querySelectorAll("div")};') | 54 'return {stuff: document.querySelectorAll("div")};') |
| 37 stuff = driver.ExecuteScript(script)['stuff'] | 55 stuff = self._driver.ExecuteScript(script)['stuff'] |
| 38 script = 'return arguments[0].innerHTML + arguments[1].innerHTML'; | 56 script = 'return arguments[0].innerHTML + arguments[1].innerHTML'; |
| 39 self.assertEquals('bc', driver.ExecuteScript(script, stuff[0], stuff[1])) | 57 self.assertEquals( |
| 40 driver.Quit() | 58 'bc', self._driver.ExecuteScript(script, stuff[0], stuff[1])) |
| 41 | 59 |
| 42 def testEvaluateInvalidScript(self): | 60 def testEvaluateInvalidScript(self): |
| 43 driver = chromedriver.ChromeDriver(_CHROMEDRIVER_LIB, _CHROME_BINARY) | |
| 44 self.assertRaises(chromedriver.ChromeDriverException, | 61 self.assertRaises(chromedriver.ChromeDriverException, |
| 45 driver.ExecuteScript, '{{{') | 62 self._driver.ExecuteScript, '{{{') |
| 46 driver.Quit() | 63 |
| 64 def testGetTitle(self): | |
| 65 self._driver.Load(self.GetHttpUrlForFile('/chromedriver/page_test.html')) | |
| 66 self.assertEqual('page test', self._driver.GetTitle()) | |
|
chrisgao (Use stgao instead)
2013/01/05 01:19:10
As Load returns before the page is fully loaded, t
kkania
2013/01/08 21:26:29
We can't use load then until it waits for the navi
chrisgao (Use stgao instead)
2013/01/08 22:23:34
Done.
| |
| 47 | 67 |
| 48 | 68 |
| 49 if __name__ == '__main__': | 69 if __name__ == '__main__': |
| 50 if len(sys.argv) != 2 and len(sys.argv) != 3: | 70 if len(sys.argv) != 2 and len(sys.argv) != 3: |
| 51 print ('Usage: %s <path_to_chromedriver_so> [path_to_chrome_binary]' % | 71 print ('Usage: %s <path_to_chromedriver_so> [path_to_chrome_binary]' % |
| 52 __file__) | 72 __file__) |
| 53 sys.exit(1) | 73 sys.exit(1) |
| 54 global _CHROMEDRIVER_LIB | 74 global _CHROMEDRIVER_LIB |
| 55 _CHROMEDRIVER_LIB = os.path.abspath(sys.argv[1]) | 75 _CHROMEDRIVER_LIB = os.path.abspath(sys.argv[1]) |
| 56 global _CHROME_BINARY | 76 global _CHROME_BINARY |
| 57 if len(sys.argv) == 3: | 77 if len(sys.argv) == 3: |
| 58 _CHROME_BINARY = os.path.abspath(sys.argv[2]) | 78 _CHROME_BINARY = os.path.abspath(sys.argv[2]) |
| 59 else: | 79 else: |
| 60 _CHROME_BINARY = None | 80 _CHROME_BINARY = None |
| 61 all_tests_suite = unittest.defaultTestLoader.loadTestsFromModule( | 81 all_tests_suite = unittest.defaultTestLoader.loadTestsFromModule( |
| 62 sys.modules[__name__]) | 82 sys.modules[__name__]) |
| 63 result = unittest.TextTestRunner().run(all_tests_suite) | 83 result = unittest.TextTestRunner().run(all_tests_suite) |
| 64 sys.exit(len(result.failures) + len(result.errors)) | 84 sys.exit(len(result.failures) + len(result.errors)) |
| OLD | NEW |