Index: chrome/browser/chromeos/login/reset_browsertest.cc |
diff --git a/chrome/browser/chromeos/login/reset_browsertest.cc b/chrome/browser/chromeos/login/reset_browsertest.cc |
index 045671cf61ae709a07dc007d037e6239d1026e3e..33d2ada4059675381ee1983c0e1d94023820cd10 100644 |
--- a/chrome/browser/chromeos/login/reset_browsertest.cc |
+++ b/chrome/browser/chromeos/login/reset_browsertest.cc |
@@ -7,13 +7,14 @@ |
#include "base/prefs/pref_service.h" |
#include "chrome/browser/browser_process.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/oobe_display.h" |
#include "chrome/browser/chromeos/login/startup_utils.h" |
#include "chrome/browser/chromeos/login/test/oobe_screen_waiter.h" |
#include "chrome/browser/chromeos/login/webui_login_view.h" |
#include "chrome/common/chrome_switches.h" |
#include "chrome/common/pref_names.h" |
+#include "chrome/test/base/in_process_browser_test.h" |
#include "chromeos/chromeos_switches.h" |
#include "chromeos/dbus/dbus_thread_manager.h" |
#include "chromeos/dbus/fake_dbus_thread_manager.h" |
@@ -30,15 +31,26 @@ const char kTestUser1[] = "test-user1@gmail.com"; |
} // namespace |
-class ResetTest : public LoginManagerTest { |
+class ResetTest : public InProcessBrowserTest { |
public: |
- ResetTest() : LoginManagerTest(false), |
- update_engine_client_(NULL), |
- session_manager_client_(NULL), |
- power_manager_client_(NULL) { |
+ ResetTest() |
+ : login_manager_test_helper_(new LoginManagerTestHelper), |
+ update_engine_client_(NULL), |
+ session_manager_client_(NULL), |
+ power_manager_client_(NULL) { |
+ set_exit_when_last_browser_closes(false); |
} |
+ |
virtual ~ResetTest() {} |
+ virtual void SetUpOnMainThread() OVERRIDE { |
+ login_manager_test_helper_->SetUp(); |
+ } |
+ |
+ virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { |
+ login_manager_test_helper_->SetUpCommandLine(command_line); |
+ } |
+ |
// LoginManagerTest overrides: |
virtual void SetUpInProcessBrowserTestFixture() OVERRIDE { |
FakeDBusThreadManager* dbus_manager = new FakeDBusThreadManager; |
@@ -54,16 +66,21 @@ class ResetTest : public LoginManagerTest { |
scoped_ptr<UpdateEngineClient>(update_engine_client_)); |
DBusThreadManager::SetInstanceForTesting(dbus_manager); |
- LoginManagerTest::SetUpInProcessBrowserTestFixture(); |
+ login_manager_test_helper_->SetUpLoginUtils(false); |
+ } |
+ |
+ virtual void CleanUpOnMainThread() OVERRIDE { |
+ login_manager_test_helper_->CleanUp(); |
} |
void RegisterSomeUser() { |
- RegisterUser(kTestUser1); |
+ login_manager_test_helper_->RegisterUser(kTestUser1); |
StartupUtils::MarkOobeCompleted(); |
} |
bool JSExecuted(const std::string& script) { |
- return content::ExecuteScript(web_contents(), script); |
+ return content::ExecuteScript(login_manager_test_helper_->web_contents(), |
+ script); |
} |
void InvokeResetScreen() { |
@@ -79,6 +96,7 @@ class ResetTest : public LoginManagerTest { |
ASSERT_TRUE(JSExecuted("$('reset-button').click();")); |
} |
+ scoped_ptr<LoginManagerTestHelper> login_manager_test_helper_; |
FakeUpdateEngineClient* update_engine_client_; |
FakeSessionManagerClient* session_manager_client_; |
FakePowerManagerClient* power_manager_client_; |
@@ -89,7 +107,7 @@ class ResetFirstAfterBootTest : public ResetTest { |
virtual ~ResetFirstAfterBootTest() {} |
virtual void SetUpCommandLine(base::CommandLine* command_line) OVERRIDE { |
- LoginManagerTest::SetUpCommandLine(command_line); |
+ login_manager_test_helper_->SetUpCommandLine(command_line); |
command_line->AppendSwitch(switches::kFirstExecAfterBoot); |
} |
}; |
@@ -99,12 +117,15 @@ IN_PROC_BROWSER_TEST_F(ResetTest, PRE_ShowAndCancel) { |
} |
IN_PROC_BROWSER_TEST_F(ResetTest, ShowAndCancel) { |
- JSExpect("!!document.querySelector('#reset.hidden')"); |
+ login_manager_test_helper_->JSExpect( |
+ "!!document.querySelector('#reset.hidden')"); |
EXPECT_EQ(0, update_engine_client_->can_rollback_call_count()); |
InvokeResetScreen(); |
- JSExpect("!document.querySelector('#reset.hidden')"); |
+ login_manager_test_helper_->JSExpect( |
+ "!document.querySelector('#reset.hidden')"); |
CloseResetScreen(); |
- JSExpect("!!document.querySelector('#reset.hidden')"); |
+ login_manager_test_helper_->JSExpect( |
+ "!!document.querySelector('#reset.hidden')"); |
EXPECT_EQ(1, update_engine_client_->can_rollback_call_count()); |
} |
@@ -138,7 +159,7 @@ IN_PROC_BROWSER_TEST_F(ResetTest, RestartBeforeRollback) { |
EXPECT_EQ(0, power_manager_client_->num_request_restart_calls()); |
EXPECT_EQ(0, session_manager_client_->start_device_wipe_call_count()); |
ASSERT_TRUE(content::ExecuteScript( |
- web_contents(), |
+ login_manager_test_helper_->web_contents(), |
"$('reset-rollback-checkbox').checked = true;")); |
ClickResetButton(); |
ASSERT_EQ(1, power_manager_client_->num_request_restart_calls()); |
@@ -180,7 +201,7 @@ IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest, ShortcutInvokedCases) { |
// rollback available and checked |
InvokeResetScreen(); |
ASSERT_TRUE(content::ExecuteScript( |
- web_contents(), |
+ login_manager_test_helper_->web_contents(), |
"$('reset-rollback-checkbox').checked = true;")); |
ClickResetButton(); |
EXPECT_EQ(0, power_manager_client_->num_request_restart_calls()); |