[chromedriver] Implement commands: findChildElement, findChildElements.

chrome/test/chromedriver/run_py_tests.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 """End to end tests for ChromeDriver."""
 
 import ctypes
 import optparse
 import os
 import sys
 import unittest
 
 import chromedriver
 import webserver
+from webelement import WebElement
 
 _THIS_DIR = os.path.abspath(os.path.dirname(__file__))
 sys.path.insert(0, os.path.join(_THIS_DIR, os.pardir, 'pylib'))
 
 from common import chrome_paths
 from common import unittest_util
 
 
 class ChromeDriverTest(unittest.TestCase):
   """End to end tests for ChromeDriver."""
@@ skipping 53 matching lines @@
     self._driver.SwitchToFrame('name')
     self.assertTrue(self._driver.ExecuteScript('return window.top != window'))
     self._driver.SwitchToMainFrame()
     self.assertTrue(self._driver.ExecuteScript('return window.top == window'))
     self._driver.SwitchToFrameByIndex(0)
     self.assertTrue(self._driver.ExecuteScript('return window.top != window'))
 
   def testGetTitle(self):
     script = 'document.title = "title"; return 1;'
     self.assertEquals(1, self._driver.ExecuteScript(script))
-    self.assertEqual('title', self._driver.GetTitle())
+    self.assertEquals('title', self._driver.GetTitle())
 
   def testFindElement(self):
     self._driver.ExecuteScript(
         'document.body.innerHTML = "<div>a</div><div>b</div>";')
-    self.assertEqual(1, len(self._driver.FindElement('tag name', 'div')))
+    self.assertTrue(
+        isinstance(self._driver.FindElement('tag name', 'div'), WebElement))
 
   def testFindElements(self):
     self._driver.ExecuteScript(
         'document.body.innerHTML = "<div>a</div><div>b</div>";')
-    self.assertEqual(2, len(self._driver.FindElements('tag name', 'div')))
+    result = self._driver.FindElements('tag name', 'div')
+    self.assertTrue(isinstance(result, list))
+    self.assertEquals(2, len(result))
+    for item in result:
+      self.assertTrue(isinstance(item, WebElement))
+
+  def testFindChildElement(self):
+    self._driver.ExecuteScript(
+        'document.body.innerHTML = "<div><br><br></div><div><a></a></div>";')
+    element = self._driver.FindElement('tag name', 'div')
+    self.assertTrue(
+        isinstance(element.FindElement('tag name', 'br'), WebElement))
+
+  def testFindChildElements(self):
+    self._driver.ExecuteScript(
+        'document.body.innerHTML = "<div><br><br></div><div><br></div>";')
+    element = self._driver.FindElement('tag name', 'div')
+    result = element.FindElements('tag name', 'br')
+    self.assertTrue(isinstance(result, list))
+    self.assertEquals(2, len(result))
+    for item in result:
+      self.assertTrue(isinstance(item, WebElement))
 
   def testGetCurrentUrl(self):
     self.assertEqual('about:blank', self._driver.GetCurrentUrl())
 
   def testGoBackAndGoForward(self):
     self._driver.Load(self.GetHttpUrlForFile('/chromedriver/empty.html'))
     self._driver.GoBack()
     self._driver.GoForward()
 
   def testRefresh(self):
@@ skipping 26 matching lines @@
   else:
     _CHROME_BINARY = None
 
   all_tests_suite = unittest.defaultTestLoader.loadTestsFromModule(
       sys.modules[__name__])
   tests = unittest_util.FilterTestSuite(all_tests_suite, options.filter)
   ChromeDriverTest.GlobalSetUp();
   result = unittest.TextTestRunner().run(tests)
   ChromeDriverTest.GlobalTearDown();
   sys.exit(len(result.failures) + len(result.errors))