chrome/browser/ui/webui/options/chromeos/shared_options_browsertest.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 "base/basictypes.h"
   6 #include "base/compiler_specific.h"
   7 #include "base/prefs/pref_service.h"
   8 #include "chrome/browser/chromeos/login/login_manager_test.h"
   9 #include "chrome/browser/chromeos/login/startup_utils.h"
  10 #include "chrome/browser/chromeos/login/ui/user_adding_screen.h"
  11 #include "chrome/browser/chromeos/login/users/user_manager.h"
  12 #include "chrome/browser/chromeos/settings/cros_settings.h"
  13 #include "chrome/browser/chromeos/settings/stub_cros_settings_provider.h"
  14 #include "chrome/browser/ui/browser.h"
  15 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  16 #include "chrome/common/pref_names.h"
  17 #include "chrome/test/base/ui_test_utils.h"
  18 #include "chromeos/settings/cros_settings_names.h"
  19 #include "content/public/browser/web_contents.h"
  20 #include "content/public/test/browser_test_utils.h"
  21 #include "content/public/test/test_utils.h"
  22
  23 namespace chromeos {
  24
  25 namespace {
  26
  27 const char* kTestUsers[] = { "test-user1@gmail.com", "test-user2@gmail.com" };
  28
  29 }  // namespace
  30
  31 class SharedOptionsTest : public LoginManagerTest {
  32  public:
  33   SharedOptionsTest()
  34     : LoginManagerTest(false),
  35       device_settings_provider_(NULL) {
  36     stub_settings_provider_.Set(kDeviceOwner, base::StringValue(kTestUsers[0]));
  37   }
  38
  39   virtual ~SharedOptionsTest() {
  40   }
  41
  42   virtual void SetUpOnMainThread() OVERRIDE {
  43     LoginManagerTest::SetUpOnMainThread();
  44     CrosSettings* settings = CrosSettings::Get();
  45     device_settings_provider_ = settings->GetProvider(kDeviceOwner);
  46     settings->RemoveSettingsProvider(device_settings_provider_);
  47     settings->AddSettingsProvider(&stub_settings_provider_);
  48   }
  49
  50   virtual void CleanUpOnMainThread() OVERRIDE {
  51     CrosSettings* settings = CrosSettings::Get();
  52     settings->RemoveSettingsProvider(&stub_settings_provider_);
  53     settings->AddSettingsProvider(device_settings_provider_);
  54     LoginManagerTest::CleanUpOnMainThread();
  55   }
  56
  57  protected:
  58   void CheckOptionsUI(const User* user, bool is_primary) {
  59     Profile* profile = UserManager::Get()->GetProfileByUser(user);
  60     profile->GetPrefs()->SetString(prefs::kGoogleServicesUsername,
  61                                    user->email());
  62
  63     ui_test_utils::BrowserAddedObserver observer;
  64     Browser* browser = CreateBrowser(profile);
  65     observer.WaitForSingleNewBrowser();
  66
  67     ui_test_utils::NavigateToURL(browser,
  68                                  GURL("chrome://settings-frame"));
  69     content::WebContents* contents =
  70         browser->tab_strip_model()->GetActiveWebContents();
  71
  72     bool banner_visible;
  73     ASSERT_TRUE(content::ExecuteScriptAndExtractBool(
  74         contents,
  75         "var e = document.getElementById('secondary-user-banner');"
  76         "var visible = e.offsetWidth > 0 && e.offsetHeight > 0;"
  77         "window.domAutomationController.send(visible);",
  78         &banner_visible));
  79     EXPECT_EQ(is_primary, !banner_visible);
  80   }
  81
  82   StubCrosSettingsProvider stub_settings_provider_;
  83   CrosSettingsProvider* device_settings_provider_;
  84
  85  private:
  86   DISALLOW_COPY_AND_ASSIGN(SharedOptionsTest);
  87 };
  88
  89 IN_PROC_BROWSER_TEST_F(SharedOptionsTest, PRE_SharedOptions) {
  90   RegisterUser(kTestUsers[0]);
  91   RegisterUser(kTestUsers[1]);
  92   StartupUtils::MarkOobeCompleted();
  93 }
  94
  95 IN_PROC_BROWSER_TEST_F(SharedOptionsTest, SharedOptions) {
  96   LoginUser(kTestUsers[0]);
  97   UserAddingScreen::Get()->Start();
  98   content::RunAllPendingInMessageLoop();
  99   AddUser(kTestUsers[1]);
 100
 101   UserManager* manager = UserManager::Get();
 102   ASSERT_EQ(2u, manager->GetLoggedInUsers().size());
 103
 104   CheckOptionsUI(manager->FindUser(kTestUsers[0]), true /* is_primary */);
 105   CheckOptionsUI(manager->FindUser(kTestUsers[1]), false /* is_primary */);
 106 }
 107
 108 }  // namespace chromeos