Componentize LoginManagerTest.

chrome/browser/chromeos/login/managed/managed_user_test_base.cc

 // Copyright 2014 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.
 
 #include "chrome/browser/chromeos/login/managed/managed_user_test_base.h"
 
 #include <string>
 
 #include "base/compiler_specific.h"
 #include "base/run_loop.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/chromeos/login/login_display_host_impl.h"
-#include "chrome/browser/chromeos/login/login_manager_test.h"
 #include "chrome/browser/chromeos/login/managed/supervised_user_authentication.h"
 #include "chrome/browser/chromeos/login/startup_utils.h"
 #include "chrome/browser/chromeos/login/supervised_user_manager.h"
 #include "chrome/browser/chromeos/login/webui_login_view.h"
 #include "chrome/browser/chromeos/net/network_portal_detector_test_impl.h"
 #include "chrome/browser/chromeos/settings/stub_cros_settings_provider.h"
 #include "chrome/browser/managed_mode/managed_user_constants.h"
 #include "chrome/browser/managed_mode/managed_user_registration_utility.h"
 #include "chrome/browser/managed_mode/managed_user_registration_utility_stub.h"
 #include "chrome/browser/managed_mode/managed_user_shared_settings_service.h"
@@ skipping 125 matching lines @@
     const base::Value& value,
     bool acknowledged,
     bool update) {
   syncer::SyncData data =
       ManagedUserSharedSettingsService::CreateSyncDataForSetting(
           mu_id, key, value, acknowledged);
   AddChange(data.GetSpecifics().managed_user_shared_setting(), update);
 }
 
 ManagedUserTestBase::ManagedUserTestBase()
-    : LoginManagerTest(true),
-      mock_async_method_caller_(NULL),
+    : mock_async_method_caller_(NULL),
       mock_homedir_methods_(NULL),
       network_portal_detector_(NULL),
-      registration_utility_stub_(NULL) {
+      registration_utility_stub_(NULL),
+      login_manager_test_helper_(new LoginManagerTestHelper) {
+  set_exit_when_last_browser_closes(false);
 }
 
 ManagedUserTestBase::~ManagedUserTestBase() {
 }
 
+void ManagedUserTestBase::SetUpOnMainThread() {
+  login_manager_test_helper_->SetUp();
+}
+
+void ManagedUserTestBase::SetUpCommandLine(CommandLine* command_line) {
+  login_manager_test_helper_->SetUpCommandLine(command_line);
+}
+
 void ManagedUserTestBase::SetUpInProcessBrowserTestFixture() {
-  LoginManagerTest::SetUpInProcessBrowserTestFixture();
+  login_manager_test_helper_->SetUpLoginUtils(true);
   mock_async_method_caller_ = new cryptohome::MockAsyncMethodCaller;
   mock_async_method_caller_->SetUp(true, cryptohome::MOUNT_ERROR_NONE);
   cryptohome::AsyncMethodCaller::InitializeForTesting(
       mock_async_method_caller_);
 
   mock_homedir_methods_ = new cryptohome::MockHomedirMethods;
   mock_homedir_methods_->SetUp(true, cryptohome::MOUNT_ERROR_NONE);
   cryptohome::HomedirMethods::InitializeForTesting(mock_homedir_methods_);
 
   registration_utility_stub_ = new ManagedUserRegistrationUtilityStub();
   scoped_utility_.reset(new ScopedTestingManagedUserRegistrationUtility(
       registration_utility_stub_));
 
   // Setup network portal detector to return online state for both
   // ethernet and wifi networks. Ethernet is an active network by
   // default.
   network_portal_detector_ = new NetworkPortalDetectorTestImpl();
   NetworkPortalDetector::InitializeForTesting(network_portal_detector_);
   NetworkPortalDetector::CaptivePortalState online_state;
   online_state.status = NetworkPortalDetector::CAPTIVE_PORTAL_STATUS_ONLINE;
   online_state.response_code = 204;
   network_portal_detector_->SetDefaultNetworkPathForTesting(
       kStubEthernetServicePath);
   network_portal_detector_->SetDetectionResultsForTesting(
       kStubEthernetServicePath, online_state);
 }
 
 void ManagedUserTestBase::CleanUpOnMainThread() {
-  LoginManagerTest::CleanUpOnMainThread();
+  login_manager_test_helper_->CleanUp();
 }
 
 void ManagedUserTestBase::TearDown() {
   cryptohome::AsyncMethodCaller::Shutdown();
   cryptohome::HomedirMethods::Shutdown();
   mock_homedir_methods_ = NULL;
   mock_async_method_caller_ = NULL;
-  LoginManagerTest::TearDown();
+  InProcessBrowserTest::TearDown();
 }
 
 void ManagedUserTestBase::TearDownInProcessBrowserTestFixture() {
   NetworkPortalDetector::Shutdown();
 }
 
 void ManagedUserTestBase::JSEval(const std::string& script) {
-  EXPECT_TRUE(content::ExecuteScript(web_contents(), script));
+  EXPECT_TRUE(content::ExecuteScript(login_manager_test_helper_->web_contents(),
+                                     script));
 }
 
 void ManagedUserTestBase::JSExpectAsync(const std::string& function) {
   bool result;
   EXPECT_TRUE(content::ExecuteScriptAndExtractBool(
-      web_contents(),
+      login_manager_test_helper_->web_contents(),
       StringPrintf(
           "(%s)(function() { window.domAutomationController.send(true); });",
           function.c_str()),
       &result));
   EXPECT_TRUE(result);
 }
 
 void ManagedUserTestBase::JSSetTextField(const std::string& element_selector,
                                          const std::string& value) {
   std::string function =
       StringPrintf("document.querySelector('%s').value = '%s'",
                    element_selector.c_str(),
                    value.c_str());
   JSEval(function);
 }
 
 void ManagedUserTestBase::PrepareUsers() {
-  RegisterUser(kTestManager);
-  RegisterUser(kTestOtherUser);
+  login_manager_test_helper_->RegisterUser(kTestManager);
+  login_manager_test_helper_->RegisterUser(kTestOtherUser);
   chromeos::StartupUtils::MarkOobeCompleted();
 }
 
 void ManagedUserTestBase::StartFlowLoginAsManager() {
   // Navigate to supervised user creation screen.
   JSEval("chrome.send('showLocallyManagedUserCreationScreen')");
 
   // Read intro and proceed.
-  JSExpect(StringPrintf("%s == 'intro'", kCurrentPage));
+  login_manager_test_helper_->JSExpect(
+      StringPrintf("%s == 'intro'", kCurrentPage));
 
   JSEval("$('managed-user-creation-start-button').click()");
 
   // Check that both users appear as managers, and test-manager@gmail.com is
   // the first one.
-  JSExpect(StringPrintf("%s == 'manager'", kCurrentPage));
+  login_manager_test_helper_->JSExpect(
+      StringPrintf("%s == 'manager'", kCurrentPage));
 
   std::string manager_pods =
       "document.querySelectorAll('#managed-user-creation-managers-pane "
       ".manager-pod')";
   std::string selected_manager_pods =
       "document.querySelectorAll('#managed-user-creation-managers-pane "
       ".manager-pod.focused')";
 
   int managers_on_device = 2;
 
-  JSExpect(StringPrintf("%s.length == 1", selected_manager_pods.c_str()));
+  login_manager_test_helper_->JSExpect(
+      StringPrintf("%s.length == 1", selected_manager_pods.c_str()));
 
-  JSExpect(
+  login_manager_test_helper_->JSExpect(
       StringPrintf("$('managed-user-creation').managerList_.pods.length == %d",
                    managers_on_device));
-  JSExpect(StringPrintf(
+  login_manager_test_helper_->JSExpect(StringPrintf(
       "%s.length == %d", manager_pods.c_str(), managers_on_device));
-  JSExpect(StringPrintf("%s[%d].user.emailAddress == '%s'",
-                        manager_pods.c_str(),
-                        0,
-                        kTestManager));
+  login_manager_test_helper_->JSExpect(
+      StringPrintf("%s[%d].user.emailAddress == '%s'",
+                   manager_pods.c_str(),
+                   0,
+                   kTestManager));
 
   // Select the first user as manager, and enter password.
-  JSExpect("$('managed-user-creation-next-button').disabled");
+  login_manager_test_helper_->JSExpect(
+      "$('managed-user-creation-next-button').disabled");
   JSSetTextField("#managed-user-creation .manager-pod.focused input",
                  kTestManagerPassword);
 
   JSEval("$('managed-user-creation').updateNextButtonForManager_()");
 
   // Next button is now enabled.
-  JSExpect("!$('managed-user-creation-next-button').disabled");
-  SetExpectedCredentials(kTestManager, kTestManagerPassword);
+  login_manager_test_helper_->JSExpect(
+      "!$('managed-user-creation-next-button').disabled");
+  login_manager_test_helper_->SetExpectedCredentials(kTestManager,
+                                                     kTestManagerPassword);
   content::WindowedNotificationObserver login_observer(
       chrome::NOTIFICATION_LOGIN_USER_PROFILE_PREPARED,
       content::NotificationService::AllSources());
 
   // Log in as manager.
   JSEval("$('managed-user-creation-next-button').click()");
   login_observer.Wait();
 
   // OAuth token is valid.
   UserManager::Get()->SaveUserOAuthStatus(kTestManager,
                                           User::OAUTH2_TOKEN_STATUS_VALID);
   base::RunLoop().RunUntilIdle();
 
   // Check the page have changed.
-  JSExpect(StringPrintf("%s == 'username'", kCurrentPage));
+  login_manager_test_helper_->JSExpect(
+      StringPrintf("%s == 'username'", kCurrentPage));
 }
 
 void ManagedUserTestBase::FillNewUserData(const std::string& display_name) {
-  JSExpect("$('managed-user-creation-next-button').disabled");
+  login_manager_test_helper_->JSExpect(
+      "$('managed-user-creation-next-button').disabled");
   JSSetTextField("#managed-user-creation-name", display_name);
   JSEval("$('managed-user-creation').checkUserName_()");
 
   base::RunLoop().RunUntilIdle();
 
   JSSetTextField("#managed-user-creation-password",
                  kTestSupervisedUserPassword);
   JSSetTextField("#managed-user-creation-password-confirm",
                  kTestSupervisedUserPassword);
 
   JSEval("$('managed-user-creation').updateNextButtonForUser_()");
-  JSExpect("!$('managed-user-creation-next-button').disabled");
+  login_manager_test_helper_->JSExpect(
+      "!$('managed-user-creation-next-button').disabled");
 }
 
 void ManagedUserTestBase::StartUserCreation(
     const std::string& button_id,
     const std::string& expected_display_name) {
   EXPECT_CALL(*mock_homedir_methods_, MountEx(_, _, _, _)).Times(1);
   EXPECT_CALL(*mock_homedir_methods_, AddKeyEx(_, _, _, _, _)).Times(1);
 
   JSEval(std::string("$('").append(button_id).append("').click()"));
 
   ::testing::Mock::VerifyAndClearExpectations(mock_homedir_methods_);
 
   EXPECT_TRUE(registration_utility_stub_->register_was_called());
   EXPECT_EQ(registration_utility_stub_->display_name(),
             base::UTF8ToUTF16(expected_display_name));
 
   registration_utility_stub_->RunSuccessCallback("token");
 
   // Token writing moves control to BlockingPool and back.
   base::RunLoop().RunUntilIdle();
   content::BrowserThread::GetBlockingPool()->FlushForTesting();
   base::RunLoop().RunUntilIdle();
 
-  JSExpect(StringPrintf("%s == 'created'", kCurrentPage));
+  login_manager_test_helper_->JSExpect(
+      StringPrintf("%s == 'created'", kCurrentPage));
   JSEval("$('managed-user-creation-gotit-button').click()");
 }
 
 void ManagedUserTestBase::SigninAsSupervisedUser(
     bool check_homedir_calls,
     int user_index,
     const std::string& expected_display_name) {
   if (check_homedir_calls)
     EXPECT_CALL(*mock_homedir_methods_, MountEx(_, _, _, _)).Times(1);
 
   // Log in as supervised user, make sure that everything works.
   ASSERT_EQ(3UL, UserManager::Get()->GetUsers().size());
 
   // Created supervised user have to be first in a list.
   const User* user = UserManager::Get()->GetUsers().at(user_index);
   ASSERT_EQ(base::UTF8ToUTF16(expected_display_name), user->display_name());
-  LoginUser(user->email());
+  login_manager_test_helper_->LogInUser(user->email());
   if (check_homedir_calls)
     ::testing::Mock::VerifyAndClearExpectations(mock_homedir_methods_);
   Profile* profile = UserManager::Get()->GetProfileByUser(user);
   shared_settings_adapter_.reset(
       new ManagedUsersSharedSettingsSyncTestAdapter(profile));
 }
 
 void ManagedUserTestBase::SigninAsManager(int user_index) {
   // Log in as supervised user, make sure that everything works.
   ASSERT_EQ(3UL, UserManager::Get()->GetUsers().size());
 
   // Created supervised user have to be first in a list.
   const User* user = UserManager::Get()->GetUsers().at(user_index);
-  LoginUser(user->email());
+  login_manager_test_helper_->LogInUser(user->email());
   Profile* profile = UserManager::Get()->GetProfileByUser(user);
   shared_settings_adapter_.reset(
       new ManagedUsersSharedSettingsSyncTestAdapter(profile));
   managed_users_adapter_.reset(new ManagedUsersSyncTestAdapter(profile));
 }
 
 void ManagedUserTestBase::RemoveSupervisedUser(
     unsigned long original_user_count,
     int user_index,
     const std::string& expected_display_name) {
   // Remove supervised user.
   ASSERT_EQ(original_user_count, UserManager::Get()->GetUsers().size());
 
   // Created supervised user have to be first in a list.
   const User* user = UserManager::Get()->GetUsers().at(user_index);
   ASSERT_EQ(base::UTF8ToUTF16(expected_display_name), user->display_name());
 
   // Open pod menu.
-  JSExpect(
+  login_manager_test_helper_->JSExpect(
       StringPrintf("!$('pod-row').pods[%d].isActionBoxMenuActive", user_index));
   JSEval(StringPrintf(
       "$('pod-row').pods[%d].querySelector('.action-box-button').click()",
       user_index));
-  JSExpect(
+  login_manager_test_helper_->JSExpect(
       StringPrintf("$('pod-row').pods[%d].isActionBoxMenuActive", user_index));
 
   // Select "Remove user" element.
-  JSExpect(StringPrintf(
+  login_manager_test_helper_->JSExpect(StringPrintf(
       "$('pod-row').pods[%d].actionBoxRemoveUserWarningElement.hidden",
       user_index));
   JSEval(StringPrintf(
       "$('pod-row').pods[%d].querySelector('.action-box-menu-remove').click()",
       user_index));
-  JSExpect(StringPrintf(
+  login_manager_test_helper_->JSExpect(StringPrintf(
       "!$('pod-row').pods[%d].actionBoxRemoveUserWarningElement.hidden",
       user_index));
 
   EXPECT_CALL(*mock_async_method_caller_, AsyncRemove(_, _)).Times(1);
 
   // Confirm deletion.
   JSEval(StringPrintf(
       "$('pod-row').pods[%d].querySelector('.remove-warning-button').click()",
       user_index));
 
   // Make sure there is no supervised user in list.
   ASSERT_EQ(original_user_count - 1, UserManager::Get()->GetUsers().size());
 }
 
 }  // namespace testing
 
 }  // namespace chromeos