Test Autofill's ability to merge duplicate profiles and...

chrome/test/functional/autofill.py

 #!/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.
 
 import os
 
+import pickle

[Nirnimesh, 2011/02/07 23:26:20]

group with os above.
pickle is a system import

[dyu1, 2011/02/09 19:44:56]

Done.

 import pyauto_functional  # Must be imported before pyauto
 import pyauto
 
 
 class AutoFillTest(pyauto.PyUITest):
   """Tests that autofill works correctly"""
 
   def Debug(self):
     """Test method for experimentation.
 
@@ skipping 68 matching lines @@
     self.assertEqual([without_invalid],
                      self.GetAutoFillProfile()['profiles'])
 
     # Then try credit cards with invalid input.  Should strip off all non-digits
     credit_card = {'CREDIT_CARD_NUMBER': 'Not_0123-5Checked'}
     expected_credit_card = {'CREDIT_CARD_NUMBER': '01235'}
     self.FillAutoFillProfile(credit_cards=[credit_card])
     self.assertEqual([expected_credit_card],
                      self.GetAutoFillProfile()['credit_cards'])
 
-  def testAutofillCrowdSourcing(self):
-    """Test able to send POST request of web form to crowd source server.
-    Require a loop of 1000 submits as the source server only collects 1% of
+  def testAutofillCrowdsourcing(self):
+    """Test able to send POST request of web form to Autofill server.
+    Require a loop of 1000 submits as the Autofill server only collects 1% of
     the data posted."""
     # HTML file needs to be run from a specific http:// url to be able to verify
     # the results a few days later by visiting the same url.
     url = 'http://www.corp.google.com/~dyu/autofill/crowdsourcing-test.html'
     # Adding crowdsourcing Autofill profile.
     file_path = os.path.join(self.DataDir(), 'autofill',
                              'crowdsource_autofill.txt')
     profiles = self.EvalDataFrom(file_path)
     self.FillAutoFillProfile(profiles=profiles)
     for i in range(1000):
       fname = self.GetAutoFillProfile()['profiles'][0]['NAME_FIRST']
       lname = self.GetAutoFillProfile()['profiles'][0]['NAME_LAST']
       email = self.GetAutoFillProfile()['profiles'][0]['EMAIL_ADDRESS']
       # Submit form to collect crowdsourcing data for Autofill.
       self.NavigateToURL(url, 0, 0)
       fname_field = 'document.getElementById("fn").value = "%s"; ' \
                     'window.domAutomationController.send("done")' % fname
       lname_field = 'document.getElementById("ln").value = "%s"; ' \
                     'window.domAutomationController.send("done")' % lname
       email_field = 'document.getElementById("em").value = "%s"; ' \
                     'window.domAutomationController.send("done")' % email
       self.ExecuteJavascript(fname_field, 0, 0);
       self.ExecuteJavascript(lname_field, 0, 0);
       self.ExecuteJavascript(email_field, 0, 0);
       self.ExecuteJavascript('document.getElementById("frmsubmit").submit();'
                              'window.domAutomationController.send("done")',
                              0, 0)
 
+  def testMergeDuplicateProfilesInAutofill(self):
+    """Test Autofill ability to merge duplicate profiles and throw away junk
+    profiles. The goal is to submit a couple hundred profiles and see the merge
+    outcome."""
+    # HTML file needs to be run from a http:// url.
+    url = self.GetHttpURLForDataPath(
+        os.path.join('autofill', 'dup-profiles-test.html'))
+    # Add and set Autofill profiles.
+    file_path = os.path.join(self.DataDir(), 'autofill',
+                             'dataset_duplicate-profiles.txt')

[Nirnimesh, 2011/02/07 23:26:20]

Why use this precooked file? Why not call the csv parser script to generate this
file on the fly?

[dyu1, 2011/02/09 19:44:56]

Done.

+    profiles = self.EvalDataFrom(file_path)
+    self.FillAutoFillProfile(profiles=profiles)
+    for i in range(len(self.GetAutoFillProfile()['profiles'])):
+      self.NavigateToURL(url)
+      for key, value in self.GetAutoFillProfile()['profiles'][i].iteritems():
+        script = 'document.getElementById("%s").value = "%s"; ' \
+                 'window.domAutomationController.send("done")' % (key, value)
+        self.ExecuteJavascript(script, 0, 0)
+      self.ExecuteJavascript('document.getElementById("merge_dup").submit();'
+                             'window.domAutomationController.send("done")',
+                             0, 0)
+    # Write profile dictionary to a file.
+    profile_dict = self.GetAutoFillProfile()['profiles']
+    output = open('merged-profile.txt', 'wb')
+    pickle.dump(profile_dict, output)
+    output.close()
+
 
 if __name__ == '__main__':
   pyauto_functional.Main()