Added a test for Autofill to check the postal code and state label change based on corresponding ...

chrome/test/webdriver/chromedriver_tests.py

 #!/usr/bin/python
 
 # Copyright (c) 2011 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.
 
 """Tests for ChromeDriver.
 
 If your test is testing a specific part of the WebDriver API, consider adding
 it to the appropriate place in the WebDriver tree instead.
@@ skipping 333 matching lines @@
 
     url_parts = urlparse.urlparse(self.session_url)[2].split('/')
     self.assertEquals(5, len(url_parts))
     self.assertEquals('', url_parts[0])
     self.assertEquals('wd', url_parts[1])
     self.assertEquals('hub', url_parts[2])
     self.assertEquals('session', url_parts[3])
     self.assertEquals(data['sessionId'], url_parts[4])
 
 
+class AutofillTest(unittest.TestCase):
+  """Autofill tests that interacts with web UI."""
+  AUTOFILL_EDIT_ADDRESS = 'chrome://settings/autoFillEditAddress'
+
+  def setUp(self):
+    self._launcher = ChromeDriverLauncher()
+
+  def tearDown(self):
+    self._launcher.Kill()
+
+  def NewDriver(self):

[Huyen, 2011/03/30 19:17:02]

Fold the code here into where you're calling NewDriver().

[dyu1, 2011/03/30 21:51:45]

What do you mean by fold the code here? If I didn't have this function then I
would have to type out "self._launcher.GetURL()" instead of just use the
"driver" object.

[Huyen, 2011/03/30 22:08:03]

Inline WebDriver(self._launcher.GetUrl(), {}) where you need a driver. For
example, line 374:
driver = WebDriver(...)

[dyu1, 2011/03/30 22:33:10]

If I fold it in then instead of driver.get for all future tests I write in this
class, I will need to type in the longer version, which will be tedious. As you
can see below there are a lot of driver objects.

[Huyen, 2011/03/30 23:23:12]

When you add more tests later you can add that method or you can change the
setUp() step to instantiate a new driver. But for now you don't need this.

+    return WebDriver(self._launcher.GetURL(), {})
+
+  def testPostalCodeAndStateLabelsBasedOnCountry(self):
+    """Verify postal code and state labels based on selected country."""
+    file_path = os.path.join('state_zip_labels.txt')
+    import simplejson
+    print dir(simplejson)

[Huyen, 2011/03/30 19:17:02]

Do you need this print? Remove if it is for debugging.

[dyu1, 2011/03/30 21:51:45]

Done.

+    test_data = simplejson.loads(open(file_path).read())
+
+    driver = self.NewDriver()
+    driver.get(self.AUTOFILL_EDIT_ADDRESS)
+    state_label = driver.find_element_by_id('state-label').text
+    self.assertEqual('State', state_label)
+    for country_code in test_data:
+      query = '//option[@value="%s"]' % country_code
+      driver.find_element_by_id(

[Huyen, 2011/03/30 19:17:02]

This is fine but if multiple postal codes are not present only the first missing
code will be reported.

[dyu1, 2011/03/30 21:51:45]

What do you mean by multiple postal codes? In this case the data in the
dictionary file is set so if the test does break then I'll know there is a bug
or something changed.

[Huyen, 2011/03/30 22:08:03]

You're looping through the list of postal codes to verify that they all exist in
the select menu right? So if multiple codes are not in the select menu your test
would not detect the subsequent missing codes, only the first one.

+          'country').find_element_by_xpath(query).select()
+
+      # Compare postal labels.
+      actual_postal_label = driver.find_element_by_id(
+          'postal-code-label').text
+      expected_postal_label = test_data[country_code]['postalCodeLabel']
+      self.assertEqual(
+          actual_postal_label, expected_postal_label,
+          'Postal code label on the UI does not correspond to the selected \

[Huyen, 2011/03/30 19:17:02]

This is a UI test so you don't need to say "on the UI". Just say "Postal code
label does not match country: %s".

[dyu1, 2011/03/30 21:51:45]

Done.

+           Country "%s"' % country_code)
+      # Compare state labels.
+      actual_state_label = driver.find_element_by_id('state-label').text
+      expected_state_label = test_data[country_code]['stateLabel']
+      self.assertEqual(
+          actual_state_label, expected_state_label,
+          'State label on the UI does not correspond to the selected \

[Huyen, 2011/03/30 19:17:02]

Same as above.

[dyu1, 2011/03/30 21:51:45]

Done.

+           Country "%s"' % country_code)
+

[Huyen, 2011/03/30 19:17:02]

You might want to call driver.quit() here.

[dyu1, 2011/03/30 21:51:45]

The driver quits on it's own when I run the test. Ken mentioned that this was
not needed. 

+
 # TODO(jleyba): Port this to WebDriver's own python test suite.
 class ElementEqualityTest(unittest.TestCase):
   """Tests that the server properly checks element equality."""
 
   def setUp(self):
     self._launcher = ChromeDriverLauncher(root_path=os.path.dirname(__file__))
     self._driver = WebDriver(self._launcher.GetURL(), {})
 
   def tearDown(self):
     self._driver.quit()
     self._launcher.Kill()
 
   def testElementEquality(self):
     self._driver.get(self._launcher.GetURL() + '/test_page.html')
     body1 = self._driver.find_element_by_tag_name('body')
     body2 = self._driver.execute_script('return document.body')
 
     # TODO(jleyba): WebDriver's python bindings should expose a proper API
     # for this.
     result = body1.execute(Command.ELEMENT_EQUALS, {
       'other': body2.id
     })
     self.assertTrue(result['value'])
 
 
 if __name__ == '__main__':
   unittest.main(module='chromedriver_tests',
                 testRunner=GTestTextTestRunner(verbosity=1))