Migrate (Network)ErrorScreen to ScreenContext

chrome/browser/chromeos/login/screens/update_screen_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/command_line.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/prefs/pref_service.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chromeos/login/screens/mock_base_screen_delegate.h"
 #include "chrome/browser/chromeos/login/screens/mock_error_screen.h"
+#include "chrome/browser/chromeos/login/screens/network_error.h"
 #include "chrome/browser/chromeos/login/screens/update_screen.h"
 #include "chrome/browser/chromeos/login/startup_utils.h"
 #include "chrome/browser/chromeos/login/test/wizard_in_process_browser_test.h"
 #include "chrome/browser/chromeos/login/wizard_controller.h"
 #include "chrome/browser/chromeos/net/network_portal_detector_test_impl.h"
 #include "chrome/common/pref_names.h"
 #include "chromeos/chromeos_switches.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "chromeos/dbus/fake_update_engine_client.h"
 #include "chromeos/network/portal_detector/network_portal_detector.h"
@@ skipping 41 matching lines @@
     online_state.response_code = 204;
     SetDefaultNetwork(kStubEthernetGuid);
     SetDetectionResults(kStubEthernetGuid, online_state);
     SetDetectionResults(kStubWifiGuid, online_state);
   }
 
   void SetUpOnMainThread() override {
     WizardInProcessBrowserTest::SetUpOnMainThread();
 
     mock_base_screen_delegate_.reset(new MockBaseScreenDelegate());
-    mock_error_screen_actor_.reset(new MockErrorScreenActor());
+    mock_network_error_view_.reset(new MockNetworkErrorView());
     mock_error_screen_.reset(new MockErrorScreen(
-        mock_base_screen_delegate_.get(), mock_error_screen_actor_.get()));
+        mock_base_screen_delegate_.get(), mock_network_error_view_.get()));
     EXPECT_CALL(*mock_base_screen_delegate_, ShowCurrentScreen())
         .Times(AnyNumber());
     EXPECT_CALL(*mock_base_screen_delegate_, GetErrorScreen())
         .Times(AnyNumber())
         .WillRepeatedly(Return(mock_error_screen_.get()));
 
     ASSERT_TRUE(WizardController::default_controller() != NULL);
     update_screen_ = UpdateScreen::Get(WizardController::default_controller());
     ASSERT_TRUE(update_screen_ != NULL);
     ASSERT_EQ(WizardController::default_controller()->current_screen(),
               update_screen_);
     update_screen_->base_screen_delegate_ = mock_base_screen_delegate_.get();
   }
 
   void TearDownOnMainThread() override {
     mock_error_screen_.reset();
-    mock_error_screen_actor_.reset();
+    mock_network_error_view_.reset();
     WizardInProcessBrowserTest::TearDownOnMainThread();
   }
 
   void TearDownInProcessBrowserTestFixture() override {
     NetworkPortalDetector::Shutdown();
     WizardInProcessBrowserTest::TearDownInProcessBrowserTestFixture();
   }
 
   void SetDefaultNetwork(const std::string& guid) {
     DCHECK(network_portal_detector_);
     network_portal_detector_->SetDefaultNetworkForTesting(guid);
   }
 
   void SetDetectionResults(
       const std::string& guid,
       const NetworkPortalDetector::CaptivePortalState& state) {
     DCHECK(network_portal_detector_);
     network_portal_detector_->SetDetectionResultsForTesting(guid, state);
   }
 
   void NotifyPortalDetectionCompleted() {
     DCHECK(network_portal_detector_);
     network_portal_detector_->NotifyObserversForTesting();
   }
 
   FakeUpdateEngineClient* fake_update_engine_client_;
   scoped_ptr<MockBaseScreenDelegate> mock_base_screen_delegate_;
-  scoped_ptr<MockErrorScreenActor> mock_error_screen_actor_;
+  scoped_ptr<MockNetworkErrorView> mock_network_error_view_;
   scoped_ptr<MockErrorScreen> mock_error_screen_;
   UpdateScreen* update_screen_;
   NetworkPortalDetectorTestImpl* network_portal_detector_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(UpdateScreenTest);
 };
 
 IN_PROC_BROWSER_TEST_F(UpdateScreenTest, TestBasic) {
   ASSERT_TRUE(update_screen_->view_ != NULL);
@@ skipping 104 matching lines @@
   update_screen_->CancelUpdate();
 
   // Change ethernet state to portal.
   NetworkPortalDetector::CaptivePortalState portal_state;
   portal_state.status = NetworkPortalDetector::CAPTIVE_PORTAL_STATUS_PORTAL;
   portal_state.response_code = 200;
   SetDetectionResults(kStubEthernetGuid, portal_state);
 
   // Update screen will show error message about portal state because
   // ethernet is behind captive portal.
-  EXPECT_CALL(*mock_error_screen_actor_,
-              SetUIState(ErrorScreen::UI_STATE_UPDATE))
-      .Times(1);
-  EXPECT_CALL(*mock_error_screen_actor_,
-              SetErrorState(ErrorScreen::ERROR_STATE_PORTAL, std::string()))
-      .Times(1);
-  EXPECT_CALL(*mock_error_screen_actor_, FixCaptivePortal())
-      .Times(1);
+  EXPECT_CALL(*mock_error_screen_,
+              MockSetUIState(NetworkError::UI_STATE_UPDATE)).Times(1);
+  EXPECT_CALL(*mock_error_screen_,
+              MockSetErrorState(NetworkError::ERROR_STATE_PORTAL,
+                                std::string())).Times(1);
+  EXPECT_CALL(*mock_error_screen_, MockFixCaptivePortal()).Times(1);
   EXPECT_CALL(*mock_base_screen_delegate_, ShowErrorScreen()).Times(1);
 
   update_screen_->StartNetworkCheck();
 
   NetworkPortalDetector::CaptivePortalState online_state;
   online_state.status = NetworkPortalDetector::CAPTIVE_PORTAL_STATUS_ONLINE;
   online_state.response_code = 204;
   SetDetectionResults(kStubEthernetGuid, online_state);
 
   // Second notification from portal detector will be about online state,
@@ skipping 16 matching lines @@
   update_screen_->CancelUpdate();
 
   // Change ethernet state to portal.
   NetworkPortalDetector::CaptivePortalState portal_state;
   portal_state.status = NetworkPortalDetector::CAPTIVE_PORTAL_STATUS_PORTAL;
   portal_state.response_code = 200;
   SetDetectionResults(kStubEthernetGuid, portal_state);
 
   // Update screen will show error message about portal state because
   // ethernet is behind captive portal.
-  EXPECT_CALL(*mock_error_screen_actor_,
-              SetUIState(ErrorScreen::UI_STATE_UPDATE))
-      .Times(1);
-  EXPECT_CALL(*mock_error_screen_actor_,
-              SetErrorState(ErrorScreen::ERROR_STATE_PORTAL, std::string()))
-      .Times(1);
-  EXPECT_CALL(*mock_error_screen_actor_, FixCaptivePortal())
-      .Times(1);
+  EXPECT_CALL(*mock_error_screen_,
+              MockSetUIState(NetworkError::UI_STATE_UPDATE)).Times(1);
+  EXPECT_CALL(*mock_error_screen_,
+              MockSetErrorState(NetworkError::ERROR_STATE_PORTAL,
+                                std::string())).Times(1);
+  EXPECT_CALL(*mock_error_screen_, MockFixCaptivePortal()).Times(1);
   EXPECT_CALL(*mock_base_screen_delegate_, ShowErrorScreen()).Times(1);
 
   update_screen_->StartNetworkCheck();
 
   // Change active network to the wifi behind proxy.
   NetworkPortalDetector::CaptivePortalState proxy_state;
   proxy_state.status =
       NetworkPortalDetector::CAPTIVE_PORTAL_STATUS_PROXY_AUTH_REQUIRED;
   proxy_state.response_code = -1;
   SetDefaultNetwork(kStubWifiGuid);
   SetDetectionResults(kStubWifiGuid, proxy_state);
 
   // Update screen will show message about proxy error because wifie
   // network requires proxy authentication.
-  EXPECT_CALL(*mock_error_screen_actor_,
-              SetErrorState(ErrorScreen::ERROR_STATE_PROXY, std::string()))
+  EXPECT_CALL(*mock_error_screen_,
+              MockSetErrorState(NetworkError::ERROR_STATE_PROXY, std::string()))
       .Times(1);
 
   NotifyPortalDetectionCompleted();
 }
 
 IN_PROC_BROWSER_TEST_F(UpdateScreenTest, TestVoidNetwork) {
   SetDefaultNetwork(std::string());
 
   // Cancels pending update request.
   EXPECT_CALL(*mock_base_screen_delegate_,
               OnExit(_, BaseScreenDelegate::UPDATE_NOUPDATE, _)).Times(1);
   update_screen_->CancelUpdate();
 
   // First portal detection attempt returns NULL network and undefined
   // results, so detection is restarted.
-  EXPECT_CALL(*mock_error_screen_actor_,
-              SetUIState(_))
-      .Times(Exactly(0));
-  EXPECT_CALL(*mock_error_screen_actor_,
-              SetErrorState(_, _))
-      .Times(Exactly(0));
+  EXPECT_CALL(*mock_error_screen_, MockSetUIState(_)).Times(Exactly(0));
+  EXPECT_CALL(*mock_error_screen_, MockSetErrorState(_, _)).Times(Exactly(0));
   EXPECT_CALL(*mock_base_screen_delegate_, ShowErrorScreen()).Times(Exactly(0));
   update_screen_->StartNetworkCheck();
 
   // Second portal detection also returns NULL network and undefined
   // results.  In this case, offline message should be displayed.
-  EXPECT_CALL(*mock_error_screen_actor_,
-              SetUIState(ErrorScreen::UI_STATE_UPDATE))
-      .Times(1);
-  EXPECT_CALL(*mock_error_screen_actor_,
-              SetErrorState(ErrorScreen::ERROR_STATE_OFFLINE, std::string()))
-      .Times(1);
+  EXPECT_CALL(*mock_error_screen_,
+              MockSetUIState(NetworkError::UI_STATE_UPDATE)).Times(1);
+  EXPECT_CALL(*mock_error_screen_,
+              MockSetErrorState(NetworkError::ERROR_STATE_OFFLINE,
+                                std::string())).Times(1);
   EXPECT_CALL(*mock_base_screen_delegate_, ShowErrorScreen()).Times(1);
   base::MessageLoop::current()->RunUntilIdle();
   NotifyPortalDetectionCompleted();
 }
 
 IN_PROC_BROWSER_TEST_F(UpdateScreenTest, TestAPReselection) {
   EXPECT_CALL(*mock_base_screen_delegate_,
               OnExit(_, BaseScreenDelegate::UPDATE_NOUPDATE, _)).Times(1);
   update_screen_->CancelUpdate();
 
   // Change ethernet state to portal.
   NetworkPortalDetector::CaptivePortalState portal_state;
   portal_state.status = NetworkPortalDetector::CAPTIVE_PORTAL_STATUS_PORTAL;
   portal_state.response_code = 200;
   SetDetectionResults(kStubEthernetGuid, portal_state);
 
   // Update screen will show error message about portal state because
   // ethernet is behind captive portal.
-  EXPECT_CALL(*mock_error_screen_actor_,
-              SetUIState(ErrorScreen::UI_STATE_UPDATE))
-      .Times(1);
-  EXPECT_CALL(*mock_error_screen_actor_,
-              SetErrorState(ErrorScreen::ERROR_STATE_PORTAL, std::string()))
-      .Times(1);
-  EXPECT_CALL(*mock_error_screen_actor_, FixCaptivePortal())
-      .Times(1);
+  EXPECT_CALL(*mock_error_screen_,
+              MockSetUIState(NetworkError::UI_STATE_UPDATE)).Times(1);
+  EXPECT_CALL(*mock_error_screen_,
+              MockSetErrorState(NetworkError::ERROR_STATE_PORTAL,
+                                std::string())).Times(1);
+  EXPECT_CALL(*mock_error_screen_, MockFixCaptivePortal()).Times(1);
   EXPECT_CALL(*mock_base_screen_delegate_, ShowErrorScreen()).Times(1);
 
   update_screen_->StartNetworkCheck();
 
   // User re-selects the same network manually. In this case, hide
   // offline message and skip network check. Since ethernet is still
   // behind portal, update engine fails to update.
   EXPECT_CALL(*mock_base_screen_delegate_, HideErrorScreen(update_screen_))
       .Times(1);
   fake_update_engine_client_->set_update_check_result(
       chromeos::UpdateEngineClient::UPDATE_RESULT_FAILED);
   EXPECT_CALL(*mock_base_screen_delegate_,
               OnExit(_, BaseScreenDelegate::UPDATE_ERROR_CHECKING_FOR_UPDATE,
                      _)).Times(1);
 
   update_screen_->OnConnectToNetworkRequested();
   base::MessageLoop::current()->RunUntilIdle();
 }
 
 }  // namespace chromeos