chrome/browser/ui/ash/system_tray_delegate_chromeos_browsertest_chromeos.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/ui/ash/system_tray_delegate_chromeos.h"
   6
   7 #include <string>
   8 #include <vector>
   9
  10 #include "ash/shell.h"
  11 #include "ash/system/date/date_default_view.h"
  12 #include "ash/system/date/date_view.h"
  13 #include "ash/system/date/tray_date.h"
  14 #include "base/prefs/pref_service.h"
  15 #include "chrome/browser/chromeos/login/login_manager_test.h"
  16 #include "chrome/browser/chromeos/login/session/user_session_manager.h"
  17 #include "chrome/browser/chromeos/login/startup_utils.h"
  18 #include "chrome/browser/chromeos/login/ui/user_adding_screen.h"
  19 #include "chrome/browser/chromeos/profiles/profile_helper.h"
  20 #include "chrome/common/pref_names.h"
  21 #include "components/user_manager/user_manager.h"
  22 #include "content/public/test/test_utils.h"
  23
  24 namespace chromeos {
  25
  26 namespace {
  27
  28 const char* kUser1 = "user1@test.com";
  29 const char* kUser2 = "user2@test.com";
  30
  31 base::HourClockType GetHourType() {
  32   const ash::TrayDate* tray_date = ash::Shell::GetInstance()
  33                                        ->GetPrimarySystemTray()
  34                                        ->GetTrayDateForTesting();
  35   const ash::DateDefaultView* date_default_view =
  36       tray_date->GetDefaultViewForTesting();
  37
  38   return date_default_view->GetDateView()->GetHourTypeForTesting();
  39 }
  40
  41 void CreateDefaultView() {
  42   ash::TrayDate* tray_date = ash::Shell::GetInstance()
  43                                  ->GetPrimarySystemTray()
  44                                  ->GetTrayDateForTesting();
  45   tray_date->CreateDefaultViewForTesting(ash::user::LOGGED_IN_NONE);
  46 }
  47
  48 }  // namespace
  49
  50 class SystemTrayDelegateChromeOSTest : public LoginManagerTest {
  51  protected:
  52   SystemTrayDelegateChromeOSTest()
  53       : LoginManagerTest(false /* should_launch_browser */) {}
  54
  55   ~SystemTrayDelegateChromeOSTest() override {}
  56
  57   void SetupUserProfile(std::string user_name, bool use_24_hour_clock) {
  58     const user_manager::User* user =
  59         user_manager::UserManager::Get()->FindUser(user_name);
  60     Profile* profile = ProfileHelper::Get()->GetProfileByUser(user);
  61     profile->GetPrefs()->SetBoolean(prefs::kUse24HourClock, use_24_hour_clock);
  62   }
  63
  64  private:
  65   DISALLOW_COPY_AND_ASSIGN(SystemTrayDelegateChromeOSTest);
  66 };
  67
  68 IN_PROC_BROWSER_TEST_F(SystemTrayDelegateChromeOSTest,
  69                        PRE_TestMultiProfile24HourClock) {
  70   RegisterUser(kUser1);
  71   RegisterUser(kUser2);
  72   StartupUtils::MarkOobeCompleted();
  73 }
  74
  75 // Test that clock type is taken from user profile for current active user.
  76 IN_PROC_BROWSER_TEST_F(SystemTrayDelegateChromeOSTest,
  77                        TestMultiProfile24HourClock) {
  78   LoginUser(kUser1);
  79   SetupUserProfile(kUser1, true /* use_24_hour_clock */);
  80   CreateDefaultView();
  81   EXPECT_EQ(base::k24HourClock, GetHourType());
  82   UserAddingScreen::Get()->Start();
  83   content::RunAllPendingInMessageLoop();
  84   AddUser(kUser2);
  85   SetupUserProfile(kUser2, false /* use_24_hour_clock */);
  86   CreateDefaultView();
  87   EXPECT_EQ(base::k12HourClock, GetHourType());
  88   user_manager::UserManager::Get()->SwitchActiveUser(kUser1);
  89   CreateDefaultView();
  90   EXPECT_EQ(base::k24HourClock, GetHourType());
  91 }
  92
  93 }  // namespace chromeos