Add a pref to track progress of OOBE and resume if needed.

chrome/browser/chromeos/login/wizard_controller_browsertest.cc

 // Copyright (c) 2012 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 "base/basictypes.h"
 #include "base/command_line.h"
 #include "base/compiler_specific.h"
 #include "base/path_service.h"
 #include "base/prefs/pref_registry_simple.h"
 #include "base/prefs/pref_service.h"
@@ skipping 852 matching lines @@
   EnrollmentScreen* screen =
       WizardController::default_controller()->GetEnrollmentScreen();
   EXPECT_EQ(screen, WizardController::default_controller()->current_screen());
   OnExit(ScreenObserver::ENTERPRISE_ENROLLMENT_BACK);
 
   EXPECT_EQ(WizardController::default_controller()->GetNetworkScreen(),
             WizardController::default_controller()->current_screen());
   EXPECT_FALSE(StartupUtils::IsOobeCompleted());
 }
 
+class WizardControllerOobeResumeTest : public WizardControllerTest {
+ protected:
+  WizardControllerOobeResumeTest() {}
+  // Overriden from InProcessBrowserTest:
+  virtual void SetUpOnMainThread() OVERRIDE {
+    WizardControllerTest::SetUpOnMainThread();
+
+    // Make sure that OOBE is run as an "official" build.
+    WizardController::default_controller()->is_official_build_ = true;
+
+    // Clear portal list (as it is by default in OOBE).
+    NetworkHandler::Get()->network_state_handler()->SetCheckPortalList("");
+
+    // Set up the mocks for all screens.
+    MOCK(mock_network_screen_, network_screen_,
+         MockNetworkScreen, MockNetworkScreenActor);
+    MOCK(mock_enrollment_screen_, enrollment_screen_,
+         MockEnrollmentScreen, MockEnrollmentScreenActor);
+  }
+
+  void OnExit(ScreenObserver::ExitCodes exit_code) {
+    WizardController::default_controller()->OnExit(exit_code);
+  }
+
+  std::string GetFirstScreenName() {
+    return WizardController::default_controller()->first_screen_name();
+  }
+
+  MockOutShowHide<MockNetworkScreen, MockNetworkScreenActor>*
+      mock_network_screen_;
+  MockOutShowHide<MockEnrollmentScreen,
+      MockEnrollmentScreenActor>* mock_enrollment_screen_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(WizardControllerOobeResumeTest);
+};
+
+IN_PROC_BROWSER_TEST_F(WizardControllerOobeResumeTest,
+                       PRE_ControlFlowResumeInterruptedOobe) {
+  // Switch to the initial screen.
+  EXPECT_CALL(*mock_network_screen_, Show()).Times(1);
+  WizardController::default_controller()->AdvanceToScreen(
+      WizardController::kNetworkScreenName);
+  EXPECT_EQ(WizardController::default_controller()->GetNetworkScreen(),
+            WizardController::default_controller()->current_screen());
+  EXPECT_CALL(*mock_enrollment_screen_->actor(),
+              SetParameters(mock_enrollment_screen_,
+                            EnrollmentScreenActor::ENROLLMENT_MODE_MANUAL,
+                            ""))
+      .Times(1);
+  EXPECT_CALL(*mock_enrollment_screen_, Show()).Times(1);
+  EXPECT_CALL(*mock_network_screen_, Hide()).Times(1);
+
+  WizardController::default_controller()->AdvanceToScreen(
+      WizardController::kEnrollmentScreenName);
+  EXPECT_EQ(WizardController::default_controller()->GetEnrollmentScreen(),
+            WizardController::default_controller()->current_screen());
+}
+
+IN_PROC_BROWSER_TEST_F(WizardControllerOobeResumeTest,
+                       ControlFlowResumeInterruptedOobe) {
+  EXPECT_EQ(WizardController::kEnrollmentScreenName, GetFirstScreenName());
+}
+
 // TODO(dzhioev): Add test emaulating device with wrong HWID.
 
 // TODO(nkostylev): Add test for WebUI accelerators http://crosbug.com/22571
 
 // TODO(merkulova): Add tests for bluetooth HID detection screen variations when
 // UI and logic is ready. http://crbug.com/127016
 
 COMPILE_ASSERT(ScreenObserver::EXIT_CODES_COUNT == 21,
                add_tests_for_new_control_flow_you_just_introduced);
 
 }  // namespace chromeos