Add a hook for capturing a user profile photo and saving it to file and local state.

chrome/test/functional/chromeos_prefs.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.
 
 import pyauto_functional  # Must be imported before pyauto
 import pyauto
 
 
 class ChromeosPrefsTest(pyauto.PyUITest):
   """TestCase for ChromeOS Preferences."""
 
+  # Defined in src/chrome/browser/chromeos/login/user_manager.cc
+  k_logged_in_users = 'LoggedInUsers'
+  k_user_images = 'UserImages'
+  # Defined in src/chrome/browser/chromeos/login/default_user_images.cc
+  default_image_prefix = 'default:'
+
   def testDefaultUserImage(self):
     """Verify changing default user image prefs work."""
 
-    # Defined in src/chrome/browser/chromeos/login/user_manager.cc
-    k_logged_in_users = 'LoggedInUsers'
-    k_user_images = 'UserImages'
-    # Defined in src/chrome/browser/chromeos/login/default_user_images.cc
-    image1 = u'default:4'
-    image2 = u'default:5'
+    k_logged_in_users = ChromeosPrefsTest.k_logged_in_users
+    k_user_images = ChromeosPrefsTest.k_user_images

[dennis_jeffrey, 2011/07/29 20:10:06]

nit: maybe we can avoid defining these two variables and just use
"ChromeosPrefsTest.k_logged_in_users" and "ChromeosPrefsTest.k_user_images"
wherever necessary.  Same comment for lnes 41-42 below.

[frankf, 2011/07/29 21:50:40]

Done.

+    image1 = ChromeosPrefsTest.default_image_prefix + '4'
+    image2 = ChromeosPrefsTest.default_image_prefix + '5'
 
     for image in image1, image2:
       logged_in_user = \
           self.GetLocalStatePrefsInfo().Prefs(k_logged_in_users)[0]
       user_images = {}
       user_images[logged_in_user] = image
       self.SetLocalStatePrefs(k_user_images, user_images)
       self.RestartBrowser(clear_profile=False)
       current_user_images = self.GetLocalStatePrefsInfo().Prefs(k_user_images)
       current_image = current_user_images.get(logged_in_user)
       self.assertEqual(image, current_image)
 
+  def testCaptureUserPhoto(self):
+    """Verify capturing/saving user photo works."""
+
+    k_logged_in_users = ChromeosPrefsTest.k_logged_in_users
+    k_user_images = ChromeosPrefsTest.k_user_images
+
+    logged_in_user = \
+        self.GetLocalStatePrefsInfo().Prefs(k_logged_in_users)[0]
+    self.CaptureProfilePhoto()
+    self.RestartBrowser(clear_profile=False)
+    current_user_images = self.GetLocalStatePrefsInfo().Prefs(k_user_images)
+    current_image_path = current_user_images.get(logged_in_user)
+    # Defined in src/chrome/browser/chromeos/login/user_manager.cc
+    expected_photo_name = logged_in_user + '.png'
+    self.assertTrue(expected_photo_name in current_image_path)

[dennis_jeffrey, 2011/07/29 20:10:06]

I recommend adding a "msg='...'" parameter to this assertion that prints a
descriptive error message in the event this assertion fails.  Same at line 36
above.

[frankf, 2011/07/29 21:50:40]

Yes, that's very useful.

On 2011/07/29 20:10:06, dennis_jeffrey wrote:

+
 
 if __name__ == '__main__':
   pyauto_functional.Main()
 

[dennis_jeffrey, 2011/07/29 20:10:06]

nit: remove this blank line

[frankf, 2011/07/29 21:50:40]

Done.