Added a timezone policy and pyauto tests for it.

chrome/test/functional/chromeos_device_policy.py

 # 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.
 
 import pyauto_functional  # Must come before pyauto (and thus, policy_base).
 import policy_base
 
 
 class ChromeosDevicePolicy(policy_base.PolicyTestBase):
   """Tests various ChromeOS device policies."""
@@ skipping 169 matching lines @@
 
     self.SetDevicePolicy({'show_user_names': True,
                           'user_whitelist': [self._usernames[1]]})
     self._WaitForPodVisibility(username=self._usernames[0], visible=False)
     self._WaitForPodVisibility(username=self._usernames[1], visible=True)
 
     self.SetDevicePolicy({'show_user_names': True})
     self._WaitForPodVisibility(username=self._usernames[0], visible=True)
     self._WaitForPodVisibility(username=self._usernames[1], visible=True)
 
+  _timezones = ['America/Barbados', 'Europe/Helsinki']
+
+  def testTimezoneSettingWithoutPolicy(self):
+    """Without timezone policy, timezone changes by user are persistent."""
+    self.SetDevicePolicy(refresh=False)
+
+    for timezone in self._timezones:
+      self.Login(user_index=1, expect_success=True)
+      self.SetTimezone(timezone)
+      self.assertEqual(timezone, self.GetTimeInfo()['timezone'])
+
+      self.Logout()
+      self.assertEqual(timezone, self.GetTimeInfo()['timezone'])
+
+  def testTimezoneSettingWithPolicy(self):
+    """With timezone policy, timezone changes by user are reset on logout."""
+    self.SetDevicePolicy({'timezone':self._timezones[0]}, refresh=True)
+
+    # Timezones are set on startup, i.e. everytime when loading the login
+    # screen. Something like a browser restart may work, too.
+    self.Login(user_index=1, expect_success=True)
+    self.Logout()
+
+    self.assertEqual(self._timezones[0], self.GetTimeInfo()['timezone'])
+
+    self.Login(user_index=1, expect_success=True)
+    self.SetTimezone(self._timezones[1])
+    self.assertEqual(self._timezones[1], self.GetTimeInfo()['timezone'])
+
+    self.Logout()
+    self.assertEqual(self._timezones[0], self.GetTimeInfo()['timezone'])
 
 if __name__ == '__main__':
   pyauto_functional.Main()