Chromium Code Reviews| Index: chrome/browser/chromeos/login/wallpaper_manager_policy_browsertest.cc |
| diff --git a/chrome/browser/chromeos/login/wallpaper_manager_policy_browsertest.cc b/chrome/browser/chromeos/login/wallpaper_manager_policy_browsertest.cc |
| index d4751fe6fd79b828a5313e28a4ac2119781094bc..215ab6127d804cb51bdd0cda48af499795e5778a 100644 |
| --- a/chrome/browser/chromeos/login/wallpaper_manager_policy_browsertest.cc |
| +++ b/chrome/browser/chromeos/login/wallpaper_manager_policy_browsertest.cc |
| @@ -18,7 +18,7 @@ |
| #include "base/path_service.h" |
| #include "base/run_loop.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/startup_utils.h" |
| #include "chrome/browser/chromeos/login/user.h" |
| #include "chrome/browser/chromeos/login/user_manager.h" |
| @@ -29,6 +29,7 @@ |
| #include "chrome/browser/profiles/profile.h" |
| #include "chrome/common/chrome_paths.h" |
| #include "chrome/common/chrome_switches.h" |
| +#include "chrome/test/base/in_process_browser_test.h" |
| #include "chromeos/chromeos_paths.h" |
| #include "chromeos/chromeos_switches.h" |
| #include "chromeos/dbus/cryptohome_client.h" |
| @@ -130,15 +131,16 @@ SkColor GetAverageBackgroundColor() { |
| } // namespace |
| class WallpaperManagerPolicyTest |
| - : public LoginManagerTest, |
| + : public InProcessBrowserTest, |
| public ash::DesktopBackgroundControllerObserver, |
| public testing::WithParamInterface<bool> { |
| protected: |
| WallpaperManagerPolicyTest() |
| - : LoginManagerTest(true), |
| - wallpaper_change_count_(0), |
| + : wallpaper_change_count_(0), |
| + login_manager_test_helper_(new LoginManagerTestHelper(true)), |
| fake_dbus_thread_manager_(new FakeDBusThreadManager), |
| fake_session_manager_client_(new FakeSessionManagerClient) { |
| + set_exit_when_last_browser_closes(false); |
| fake_dbus_thread_manager_->SetFakeClients(); |
| fake_dbus_thread_manager_->SetSessionManagerClient( |
| scoped_ptr<SessionManagerClient>(fake_session_manager_client_)); |
| @@ -171,15 +173,15 @@ class WallpaperManagerPolicyTest |
| // LoginManagerTest: |
| virtual void SetUpInProcessBrowserTestFixture() OVERRIDE { |
| DBusThreadManager::SetInstanceForTesting(fake_dbus_thread_manager_); |
| - LoginManagerTest::SetUpInProcessBrowserTestFixture(); |
| + login_manager_test_helper_->SetUpLoginUtils(); |
| ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &test_data_dir_)); |
| } |
| virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { |
| - // Set the same switches as LoginManagerTest, except that kMultiProfiles is |
| - // only set when GetParam() is true and except that kLoginProfile is set |
| - // when GetParam() is false. The latter seems to be required for the sane |
| - // start-up of user profiles. |
| + // Set the same switches as LoginManagerTestHelper, except that |
| + // kMultiProfiles is only set when GetParam() is true and kLoginProfile is |
| + // set when GetParam() is false. The latter seems to be required for the |
| + // sanestart-up of user profiles. |
|
michaelpg
2014/05/07 00:55:05
"sane start-up"
michaelpg
2014/05/07 02:48:24
Done.
|
| command_line->AppendSwitch(switches::kLoginManager); |
| command_line->AppendSwitch(switches::kForceLoginManagerInTests); |
| if (GetParam()) |
| @@ -189,7 +191,7 @@ class WallpaperManagerPolicyTest |
| } |
| virtual void SetUpOnMainThread() OVERRIDE { |
| - LoginManagerTest::SetUpOnMainThread(); |
| + login_manager_test_helper_->SetUp(); |
| ash::Shell::GetInstance()-> |
| desktop_background_controller()->AddObserver(this); |
| @@ -203,7 +205,11 @@ class WallpaperManagerPolicyTest |
| virtual void TearDownOnMainThread() OVERRIDE { |
| ash::Shell::GetInstance()-> |
| desktop_background_controller()->RemoveObserver(this); |
| - LoginManagerTest::TearDownOnMainThread(); |
| + InProcessBrowserTest::TearDownOnMainThread(); |
| + } |
| + |
| + virtual void CleanUpOnMainThread() OVERRIDE { |
| + login_manager_test_helper_->CleanUp(); |
| } |
| // ash::DesktopBackgroundControllerObserver: |
| @@ -269,6 +275,7 @@ class WallpaperManagerPolicyTest |
| base::FilePath test_data_dir_; |
| scoped_ptr<base::RunLoop> run_loop_; |
| int wallpaper_change_count_; |
| + scoped_ptr<LoginManagerTestHelper> login_manager_test_helper_; |
| scoped_ptr<policy::UserPolicyBuilder> user_policy_builders_[2]; |
| FakeDBusThreadManager* fake_dbus_thread_manager_; |
| FakeSessionManagerClient* fake_session_manager_client_; |
| @@ -278,8 +285,8 @@ class WallpaperManagerPolicyTest |
| }; |
| IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, PRE_SetResetClear) { |
| - RegisterUser(kTestUsers[0]); |
| - RegisterUser(kTestUsers[1]); |
| + login_manager_test_helper_->RegisterUser(kTestUsers[0]); |
| + login_manager_test_helper_->RegisterUser(kTestUsers[1]); |
| StartupUtils::MarkOobeCompleted(); |
| } |
| @@ -289,7 +296,7 @@ IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, PRE_SetResetClear) { |
| // reverts to default. |
| IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, SetResetClear) { |
| WallpaperInfo info; |
| - LoginUser(kTestUsers[0]); |
| + login_manager_test_helper_->LogInUser(kTestUsers[0]); |
| base::RunLoop().RunUntilIdle(); |
| // First user: Wait until default wallpaper has been loaded (happens |
| @@ -330,14 +337,14 @@ IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, SetResetClear) { |
| IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, |
| PRE_PRE_PRE_WallpaperOnLoginScreen) { |
| - RegisterUser(kTestUsers[0]); |
| - RegisterUser(kTestUsers[1]); |
| + login_manager_test_helper_->RegisterUser(kTestUsers[0]); |
| + login_manager_test_helper_->RegisterUser(kTestUsers[1]); |
| StartupUtils::MarkOobeCompleted(); |
| } |
| IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, |
| PRE_PRE_WallpaperOnLoginScreen) { |
| - LoginUser(kTestUsers[0]); |
| + login_manager_test_helper_->LogInUser(kTestUsers[0]); |
| // Wait until default wallpaper has been loaded. |
| RunUntilWallpaperChangeCount(1); |
| @@ -351,7 +358,7 @@ IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, |
| } |
| IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, PRE_WallpaperOnLoginScreen) { |
| - LoginUser(kTestUsers[1]); |
| + login_manager_test_helper_->LogInUser(kTestUsers[1]); |
| // Wait until default wallpaper has been loaded. |
| RunUntilWallpaperChangeCount(1); |
| @@ -380,12 +387,12 @@ IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, WallpaperOnLoginScreen) { |
| } |
| IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, PRE_PRE_PersistOverLogout) { |
| - RegisterUser(kTestUsers[0]); |
| + login_manager_test_helper_->RegisterUser(kTestUsers[0]); |
| StartupUtils::MarkOobeCompleted(); |
| } |
| IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, PRE_PersistOverLogout) { |
| - LoginUser(kTestUsers[0]); |
| + login_manager_test_helper_->LogInUser(kTestUsers[0]); |
| // Wait until default wallpaper has been loaded. |
| RunUntilWallpaperChangeCount(1); |
| @@ -399,7 +406,7 @@ IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, PRE_PersistOverLogout) { |
| } |
| IN_PROC_BROWSER_TEST_P(WallpaperManagerPolicyTest, PersistOverLogout) { |
| - LoginUser(kTestUsers[0]); |
| + login_manager_test_helper_->LogInUser(kTestUsers[0]); |
| // Wait until wallpaper has been loaded. |
| RunUntilWallpaperChangeCount(1); |