| Index: chrome/browser/chromeos/login/wizard_controller_browsertest.cc
|
| diff --git a/chrome/browser/chromeos/login/wizard_controller_browsertest.cc b/chrome/browser/chromeos/login/wizard_controller_browsertest.cc
|
| index bb216594165112ddcc09118e681e2c7983a6d90c..fc2123da0b0bd6a5b4d57ec5bca2ff3e2504a05c 100644
|
| --- a/chrome/browser/chromeos/login/wizard_controller_browsertest.cc
|
| +++ b/chrome/browser/chromeos/login/wizard_controller_browsertest.cc
|
| @@ -222,20 +222,22 @@ class MockOutShowHide : public T {
|
| std::unique_ptr<H> view_;
|
| };
|
|
|
| -#define MOCK(mock_var, screen_name, mocked_class, view_class) \
|
| - mock_var = new MockOutShowHide<mocked_class, view_class>( \
|
| - WizardController::default_controller(), new view_class); \
|
| - WizardController::default_controller()->screens_[screen_name] = \
|
| - make_linked_ptr(mock_var); \
|
| - EXPECT_CALL(*mock_var, Show()).Times(0); \
|
| +#define MOCK(mock_var, screen_name, mocked_class, view_class) \
|
| + mock_var = new MockOutShowHide<mocked_class, view_class>( \
|
| + WizardController::default_controller(), new view_class); \
|
| + WizardController::default_controller() \
|
| + ->screen_manager() \
|
| + ->screens_[screen_name] = base::WrapUnique(mock_var); \
|
| + EXPECT_CALL(*mock_var, Show()).Times(0); \
|
| EXPECT_CALL(*mock_var, Hide()).Times(0);
|
|
|
| #define MOCK_WITH_DELEGATE(mock_var, screen_name, mocked_class, view_class) \
|
| mock_var = new MockOutShowHide<mocked_class, view_class>( \
|
| WizardController::default_controller(), \
|
| WizardController::default_controller(), new view_class); \
|
| - WizardController::default_controller()->screens_[screen_name] = \
|
| - make_linked_ptr(mock_var); \
|
| + WizardController::default_controller() \
|
| + ->screen_manager() \
|
| + ->screens_[screen_name] = base::WrapUnique(mock_var); \
|
| EXPECT_CALL(*mock_var, Show()).Times(0); \
|
| EXPECT_CALL(*mock_var, Hide()).Times(0);
|
|
|
| @@ -419,11 +421,13 @@ class WizardControllerFlowTest : public WizardControllerTest {
|
| NetworkHandler::Get()->network_state_handler()->SetCheckPortalList("");
|
|
|
| // Set up the mocks for all screens.
|
| - mock_network_screen_.reset(new MockNetworkScreen(
|
| + mock_network_screen_ = new MockNetworkScreen(
|
| WizardController::default_controller(),
|
| - WizardController::default_controller(), GetOobeUI()->GetNetworkView()));
|
| + WizardController::default_controller(), GetOobeUI()->GetNetworkView());
|
| WizardController::default_controller()
|
| - ->screens_[OobeScreen::SCREEN_OOBE_NETWORK] = mock_network_screen_;
|
| + ->screen_manager()
|
| + ->screens_[OobeScreen::SCREEN_OOBE_NETWORK]
|
| + .reset(mock_network_screen_);
|
| EXPECT_CALL(*mock_network_screen_, Show()).Times(0);
|
| EXPECT_CALL(*mock_network_screen_, Hide()).Times(0);
|
|
|
| @@ -442,9 +446,10 @@ class WizardControllerFlowTest : public WizardControllerTest {
|
| OobeScreen::SCREEN_OOBE_ENABLE_DEBUGGING, MockEnableDebuggingScreen,
|
| MockEnableDebuggingScreenView);
|
| device_disabled_screen_view_.reset(new MockDeviceDisabledScreenView);
|
| - wizard_controller->screens_[OobeScreen::SCREEN_DEVICE_DISABLED] =
|
| - make_linked_ptr(new DeviceDisabledScreen(
|
| - wizard_controller, device_disabled_screen_view_.get()));
|
| + wizard_controller->screen_manager()
|
| + ->screens_[OobeScreen::SCREEN_DEVICE_DISABLED] =
|
| + base::MakeUnique<DeviceDisabledScreen>(
|
| + wizard_controller, device_disabled_screen_view_.get());
|
| EXPECT_CALL(*device_disabled_screen_view_, Show()).Times(0);
|
|
|
| // Switch to the initial screen.
|
| @@ -454,7 +459,7 @@ class WizardControllerFlowTest : public WizardControllerTest {
|
| }
|
|
|
| void TearDownOnMainThread() override {
|
| - mock_network_screen_.reset();
|
| + mock_network_screen_ = nullptr;
|
| device_disabled_screen_view_.reset();
|
| WizardControllerTest::TearDownOnMainThread();
|
| }
|
| @@ -512,7 +517,7 @@ class WizardControllerFlowTest : public WizardControllerTest {
|
| }
|
|
|
| void ResetAutoEnrollmentCheckScreen() {
|
| - WizardController::default_controller()->screens_.erase(
|
| + WizardController::default_controller()->screen_manager()->screens_.erase(
|
| OobeScreen::SCREEN_AUTO_ENROLLMENT_CHECK);
|
| }
|
|
|
| @@ -567,7 +572,7 @@ class WizardControllerFlowTest : public WizardControllerTest {
|
| ->GetCurrentTimezoneID()));
|
| }
|
|
|
| - linked_ptr<MockNetworkScreen> mock_network_screen_;
|
| + MockNetworkScreen* mock_network_screen_; // Unowned ptr.
|
| MockOutShowHide<MockUpdateScreen, MockUpdateView>* mock_update_screen_;
|
| MockOutShowHide<MockEulaScreen, MockEulaView>* mock_eula_screen_;
|
| MockOutShowHide<MockEnrollmentScreen, MockEnrollmentScreenView>*
|
|
|