Componentize LoginManagerTest.

chrome/browser/chromeos/login/user_adding_screen_browsertest.cc

 // Copyright 2013 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 "ash/session/session_state_delegate.h"
 #include "ash/shell.h"
 #include "base/prefs/pref_service.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/login_manager_test_helper.h"
 #include "chrome/browser/chromeos/login/multi_profile_user_controller.h"
 #include "chrome/browser/chromeos/login/startup_utils.h"
 #include "chrome/browser/chromeos/login/user_adding_screen.h"
 #include "chrome/browser/chromeos/login/user_manager.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "content/public/test/test_utils.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using namespace testing;
 
 namespace {
 
 const char* kTestUsers[] = {"test-user1@gmail.com",
                             "test-user2@gmail.com",
                             "test-user3@gmail.com"};
 
 }  // anonymous namespace
 
 namespace chromeos {
 
-class UserAddingScreenTest : public LoginManagerTest,
+class UserAddingScreenTest : public InProcessBrowserTest,
                              public UserAddingScreen::Observer {
  public:
-  UserAddingScreenTest() : LoginManagerTest(false),
-                           user_adding_started_(0),
-                           user_adding_finished_(0) {
+  UserAddingScreenTest()
+      : login_manager_test_helper_(new LoginManagerTestHelper),
+        user_adding_started_(0),
+        user_adding_finished_(0) {
+    set_exit_when_last_browser_closes(false);
+  }
+
+  virtual void SetUpOnMainThread() OVERRIDE {
+    login_manager_test_helper_->SetUp();
   }
 
   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
-    LoginManagerTest::SetUpCommandLine(command_line);
+    login_manager_test_helper_->SetUpCommandLine(command_line);
     command_line->AppendSwitch(::switches::kMultiProfiles);
   }
 
   virtual void SetUpInProcessBrowserTestFixture() OVERRIDE {
-    LoginManagerTest::SetUpInProcessBrowserTestFixture();
+    login_manager_test_helper_->SetUpLoginUtils(false);
     UserAddingScreen::Get()->AddObserver(this);
   }
 
+  virtual void CleanUpOnMainThread() OVERRIDE {
+    login_manager_test_helper_->CleanUp();
+  }
+
   virtual void OnUserAddingFinished() OVERRIDE { ++user_adding_finished_; }
 
   virtual void OnUserAddingStarted() OVERRIDE { ++user_adding_started_; }
 
   void SetUserCanLock(User* user, bool can_lock) {
     user->set_can_lock(can_lock);
   }
 
   int user_adding_started() { return user_adding_started_; }
 
   int user_adding_finished() { return user_adding_finished_; }
 
+ protected:
+  scoped_ptr<LoginManagerTestHelper> login_manager_test_helper_;
+
  private:
   int user_adding_started_;
   int user_adding_finished_;
 
   DISALLOW_COPY_AND_ASSIGN(UserAddingScreenTest);
 };
 
 IN_PROC_BROWSER_TEST_F(UserAddingScreenTest, PRE_CancelAdding) {
-  RegisterUser(kTestUsers[0]);
-  RegisterUser(kTestUsers[1]);
-  RegisterUser(kTestUsers[2]);
+  login_manager_test_helper_->RegisterUser(kTestUsers[0]);
+  login_manager_test_helper_->RegisterUser(kTestUsers[1]);
+  login_manager_test_helper_->RegisterUser(kTestUsers[2]);
   StartupUtils::MarkOobeCompleted();
 }
 
 IN_PROC_BROWSER_TEST_F(UserAddingScreenTest, CancelAdding) {
-  EXPECT_CALL(login_utils(), DoBrowserLaunch(_, _)).Times(1);
+  EXPECT_CALL(login_manager_test_helper_->login_utils(),
+              DoBrowserLaunch(_, _)).Times(1);
   EXPECT_EQ(3u, UserManager::Get()->GetUsers().size());
   EXPECT_EQ(0u, UserManager::Get()->GetLoggedInUsers().size());
 
   EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_LOGIN_PRIMARY,
             ash::Shell::GetInstance()->session_state_delegate()->
                 GetSessionState());
 
-  LoginUser(kTestUsers[0]);
+  login_manager_test_helper_->LogInUser(kTestUsers[0]);
   EXPECT_EQ(1u, UserManager::Get()->GetLoggedInUsers().size());
   EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_ACTIVE,
             ash::Shell::GetInstance()->session_state_delegate()->
                 GetSessionState());
 
   UserAddingScreen::Get()->Start();
   content::RunAllPendingInMessageLoop();
   EXPECT_EQ(1, user_adding_started());
   EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_LOGIN_SECONDARY,
             ash::Shell::GetInstance()->session_state_delegate()->
                 GetSessionState());
 
   UserAddingScreen::Get()->Cancel();
   content::RunAllPendingInMessageLoop();
   EXPECT_EQ(1, user_adding_finished());
   EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_ACTIVE,
             ash::Shell::GetInstance()->session_state_delegate()->
                 GetSessionState());
 
   EXPECT_TRUE(LoginDisplayHostImpl::default_host() == NULL);
   EXPECT_EQ(1u, UserManager::Get()->GetLoggedInUsers().size());
   EXPECT_EQ(kTestUsers[0], UserManager::Get()->GetActiveUser()->email());
 }
 
 IN_PROC_BROWSER_TEST_F(UserAddingScreenTest, PRE_AddingSeveralUsers) {
-  RegisterUser(kTestUsers[0]);
-  RegisterUser(kTestUsers[1]);
-  RegisterUser(kTestUsers[2]);
+  login_manager_test_helper_->RegisterUser(kTestUsers[0]);
+  login_manager_test_helper_->RegisterUser(kTestUsers[1]);
+  login_manager_test_helper_->RegisterUser(kTestUsers[2]);
   StartupUtils::MarkOobeCompleted();
 }
 
 IN_PROC_BROWSER_TEST_F(UserAddingScreenTest, AddingSeveralUsers) {
   EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_LOGIN_PRIMARY,
             ash::Shell::GetInstance()->session_state_delegate()->
                 GetSessionState());
-  EXPECT_CALL(login_utils(), DoBrowserLaunch(_, _)).Times(3);
-  LoginUser(kTestUsers[0]);
+  EXPECT_CALL(login_manager_test_helper_->login_utils(),
+              DoBrowserLaunch(_, _)).Times(3);
+  login_manager_test_helper_->LogInUser(kTestUsers[0]);
   EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_ACTIVE,
             ash::Shell::GetInstance()->session_state_delegate()->
                 GetSessionState());
 
   UserManager* user_manager = UserManager::Get();
 
   for (int i = 1; i < 3; ++i) {
     UserAddingScreen::Get()->Start();
     content::RunAllPendingInMessageLoop();
     EXPECT_EQ(i, user_adding_started());
     EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_LOGIN_SECONDARY,
               ash::Shell::GetInstance()->session_state_delegate()->
                   GetSessionState());
-    AddUser(kTestUsers[i]);
+    login_manager_test_helper_->AddUser(kTestUsers[i]);
     EXPECT_EQ(i, user_adding_finished());
     EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_ACTIVE,
               ash::Shell::GetInstance()->session_state_delegate()->
                   GetSessionState());
     EXPECT_TRUE(LoginDisplayHostImpl::default_host() == NULL);
     ASSERT_EQ(unsigned(i + 1), user_manager->GetLoggedInUsers().size());
   }
 
   EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_ACTIVE,
             ash::Shell::GetInstance()->session_state_delegate()->
@@ skipping 60 matching lines @@
   // not be listed on screen lock.
   prefs3->SetString(prefs::kMultiProfileUserBehavior,
                     MultiProfileUserController::kBehaviorNotAllowed);
   unlock_users = user_manager->GetUnlockUsers();
   ASSERT_EQ(2UL, unlock_users.size());
   for (int i = 0; i < 2; ++i)
     EXPECT_EQ(kTestUsers[i], unlock_users[i]->email());
 }
 
 }  // namespace chromeos