| 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 89eb85aea2cfb352b49c4d1db2d6bbebc5f210d7..fa5264d89f4120ddb81e5bf75cc0a48771996de5 100644
|
| --- a/chrome/browser/chromeos/login/reset_browsertest.cc
|
| +++ b/chrome/browser/chromeos/login/reset_browsertest.cc
|
| @@ -76,6 +76,11 @@ class ResetTest : public LoginManagerTest {
|
| OobeScreenWaiter(OobeDisplay::SCREEN_OOBE_RESET).Wait();
|
| }
|
|
|
| + void InvokeRollbackOption() {
|
| + ASSERT_TRUE(JSExecuted(
|
| + "cr.ui.Oobe.handleAccelerator('show_rollback_on_reset_screen');"));
|
| + }
|
| +
|
| void CloseResetScreen() {
|
| ASSERT_TRUE(JSExecuted("$('reset-cancel-button').click();"));
|
| }
|
| @@ -84,6 +89,13 @@ class ResetTest : public LoginManagerTest {
|
| ASSERT_TRUE(JSExecuted("$('reset-button').click();"));
|
| }
|
|
|
| + void ClickRestartButton() {
|
| + ASSERT_TRUE(JSExecuted("$('reset-restart-button').click();"));
|
| + }
|
| + void ClickToConfirmButton() {
|
| + ASSERT_TRUE(JSExecuted("$('reset-toconfirm-button').click();"));
|
| + }
|
| +
|
| FakeUpdateEngineClient* update_engine_client_;
|
| FakeSessionManagerClient* session_manager_client_;
|
| FakePowerManagerClient* power_manager_client_;
|
| @@ -124,34 +136,10 @@ IN_PROC_BROWSER_TEST_F(ResetTest, RestartBeforePowerwash) {
|
| InvokeResetScreen();
|
| EXPECT_EQ(0, power_manager_client_->num_request_restart_calls());
|
| EXPECT_EQ(0, session_manager_client_->start_device_wipe_call_count());
|
| - ClickResetButton();
|
| - ASSERT_EQ(1, power_manager_client_->num_request_restart_calls());
|
| - ASSERT_EQ(0, session_manager_client_->start_device_wipe_call_count());
|
| -
|
| - EXPECT_FALSE(prefs->GetBoolean(prefs::kRollbackRequested));
|
| - EXPECT_TRUE(prefs->GetBoolean(prefs::kFactoryResetRequested));
|
| -}
|
| -
|
| -IN_PROC_BROWSER_TEST_F(ResetTest, PRE_RestartBeforeRollback) {
|
| - RegisterSomeUser();
|
| - update_engine_client_->set_can_rollback_check_result(true);
|
| -}
|
| -
|
| -IN_PROC_BROWSER_TEST_F(ResetTest, RestartBeforeRollback) {
|
| - PrefService* prefs = g_browser_process->local_state();
|
| -
|
| - InvokeResetScreen();
|
| - 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(),
|
| - "$('reset-rollback-checkbox').checked = true;"));
|
| - ClickResetButton();
|
| + ClickRestartButton();
|
| ASSERT_EQ(1, power_manager_client_->num_request_restart_calls());
|
| ASSERT_EQ(0, session_manager_client_->start_device_wipe_call_count());
|
| - EXPECT_EQ(0, update_engine_client_->rollback_call_count());
|
|
|
| - EXPECT_TRUE(prefs->GetBoolean(prefs::kRollbackRequested));
|
| EXPECT_TRUE(prefs->GetBoolean(prefs::kFactoryResetRequested));
|
| }
|
|
|
| @@ -166,6 +154,8 @@ IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest, ShortcutInvokedCases) {
|
| EXPECT_EQ(0, power_manager_client_->num_request_restart_calls());
|
| EXPECT_EQ(0, session_manager_client_->start_device_wipe_call_count());
|
| EXPECT_EQ(0, update_engine_client_->rollback_call_count());
|
| + InvokeRollbackOption();
|
| + ClickToConfirmButton();
|
| ClickResetButton();
|
| EXPECT_EQ(0, power_manager_client_->num_request_restart_calls());
|
| EXPECT_EQ(1, session_manager_client_->start_device_wipe_call_count());
|
| @@ -176,6 +166,7 @@ IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest, ShortcutInvokedCases) {
|
| update_engine_client_->set_can_rollback_check_result(true);
|
| // rollback available and unchecked
|
| InvokeResetScreen();
|
| + ClickToConfirmButton();
|
| ClickResetButton();
|
| EXPECT_EQ(0, power_manager_client_->num_request_restart_calls());
|
| EXPECT_EQ(2, session_manager_client_->start_device_wipe_call_count());
|
| @@ -185,9 +176,8 @@ IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest, ShortcutInvokedCases) {
|
|
|
| // rollback available and checked
|
| InvokeResetScreen();
|
| - ASSERT_TRUE(content::ExecuteScript(
|
| - web_contents(),
|
| - "$('reset-rollback-checkbox').checked = true;"));
|
| + InvokeRollbackOption();
|
| + ClickToConfirmButton();
|
| ClickResetButton();
|
| EXPECT_EQ(0, power_manager_client_->num_request_restart_calls());
|
| EXPECT_EQ(2, session_manager_client_->start_device_wipe_call_count());
|
| @@ -196,7 +186,6 @@ IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest, ShortcutInvokedCases) {
|
|
|
| IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest, PRE_PowerwashRequested) {
|
| PrefService* prefs = g_browser_process->local_state();
|
| - prefs->SetBoolean(prefs::kRollbackRequested, false);
|
| prefs->SetBoolean(prefs::kFactoryResetRequested, true);
|
| RegisterSomeUser();
|
| }
|
| @@ -206,6 +195,10 @@ IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest, PowerwashRequested) {
|
| EXPECT_EQ(0, power_manager_client_->num_request_restart_calls());
|
| EXPECT_EQ(0, session_manager_client_->start_device_wipe_call_count());
|
| EXPECT_EQ(0, update_engine_client_->rollback_call_count());
|
| + ClickToConfirmButton();
|
| + EXPECT_EQ(0, power_manager_client_->num_request_restart_calls());
|
| + EXPECT_EQ(0, session_manager_client_->start_device_wipe_call_count());
|
| + EXPECT_EQ(0, update_engine_client_->rollback_call_count());
|
| ClickResetButton();
|
| EXPECT_EQ(0, power_manager_client_->num_request_restart_calls());
|
| EXPECT_EQ(1, session_manager_client_->start_device_wipe_call_count());
|
| @@ -214,22 +207,24 @@ IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest, PowerwashRequested) {
|
|
|
| IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest, PRE_ErrorOnRollbackRequested) {
|
| PrefService* prefs = g_browser_process->local_state();
|
| - prefs->SetBoolean(prefs::kRollbackRequested, true);
|
| prefs->SetBoolean(prefs::kFactoryResetRequested, true);
|
| RegisterSomeUser();
|
| }
|
|
|
| IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest, ErrorOnRollbackRequested) {
|
| + update_engine_client_->set_can_rollback_check_result(true);
|
| OobeScreenWaiter(OobeDisplay::SCREEN_OOBE_RESET).Wait();
|
| EXPECT_EQ(0, power_manager_client_->num_request_restart_calls());
|
| EXPECT_EQ(0, session_manager_client_->start_device_wipe_call_count());
|
| EXPECT_EQ(0, update_engine_client_->rollback_call_count());
|
| - JSExpect("!$('reset').classList.contains('revert-promise')");
|
| + JSExpect("!$('reset').classList.contains('revert-promise-view')");
|
| + InvokeRollbackOption();
|
| + ClickToConfirmButton();
|
| ClickResetButton();
|
| EXPECT_EQ(0, power_manager_client_->num_request_restart_calls());
|
| EXPECT_EQ(0, session_manager_client_->start_device_wipe_call_count());
|
| EXPECT_EQ(1, update_engine_client_->rollback_call_count());
|
| - JSExpect("$('reset').classList.contains('revert-promise')");
|
| + JSExpect("$('reset').classList.contains('revert-promise-view')");
|
| UpdateEngineClient::Status error_update_status;
|
| error_update_status.status = UpdateEngineClient::UPDATE_STATUS_ERROR;
|
| update_engine_client_->NotifyObserversThatStatusChanged(
|
| @@ -240,13 +235,15 @@ IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest, ErrorOnRollbackRequested) {
|
| IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest,
|
| PRE_SuccessOnRollbackRequested) {
|
| PrefService* prefs = g_browser_process->local_state();
|
| - prefs->SetBoolean(prefs::kRollbackRequested, true);
|
| prefs->SetBoolean(prefs::kFactoryResetRequested, true);
|
| RegisterSomeUser();
|
| }
|
|
|
| IN_PROC_BROWSER_TEST_F(ResetFirstAfterBootTest, SuccessOnRollbackRequested) {
|
| + update_engine_client_->set_can_rollback_check_result(true);
|
| OobeScreenWaiter(OobeDisplay::SCREEN_OOBE_RESET).Wait();
|
| + InvokeRollbackOption();
|
| + ClickToConfirmButton();
|
| ClickResetButton();
|
| EXPECT_EQ(0, power_manager_client_->num_request_restart_calls());
|
| EXPECT_EQ(0, session_manager_client_->start_device_wipe_call_count());
|
|
|