Fix autoupdate_EndToEndTest on veyron_rialto

chrome/browser/chromeos/login/screens/network_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 "chrome/browser/chromeos/login/screens/network_screen.h"
 
 #include <memory>
 
 #include "base/command_line.h"
 #include "base/macros.h"
-#include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/chromeos/login/enrollment/enrollment_screen.h"
 #include "chrome/browser/chromeos/login/helper.h"
+#include "chrome/browser/chromeos/login/mock_network_state_helper.h"
 #include "chrome/browser/chromeos/login/screens/base_screen.h"
 #include "chrome/browser/chromeos/login/screens/mock_base_screen_delegate.h"
 #include "chrome/browser/chromeos/login/test/oobe_screen_waiter.h"
 #include "chrome/browser/chromeos/login/test/wizard_in_process_browser_test.h"
 #include "chrome/browser/chromeos/login/ui/login_display_host.h"
 #include "chrome/browser/chromeos/login/wizard_controller.h"
 #include "chromeos/chromeos_switches.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "chromeos/dbus/fake_session_manager_client.h"
 #include "chromeos/dbus/shill_manager_client.h"
@@ skipping 10 matching lines @@
 using ::testing::ReturnRef;
 using views::Button;
 
 namespace chromeos {
 
 class DummyButtonListener : public views::ButtonListener {
  public:
   void ButtonPressed(views::Button* sender, const ui::Event& event) override {}
 };
 
-namespace login {
-
-class MockNetworkStateHelper : public NetworkStateHelper {
- public:
-  MOCK_CONST_METHOD0(GetCurrentNetworkName, base::string16(void));
-  MOCK_CONST_METHOD0(IsConnected, bool(void));
-  MOCK_CONST_METHOD0(IsConnecting, bool(void));
-};
-
-}  // namespace login
-
 class NetworkScreenTest : public WizardInProcessBrowserTest {
  public:
   NetworkScreenTest()
       : WizardInProcessBrowserTest(OobeScreen::SCREEN_OOBE_NETWORK),
         fake_session_manager_client_(nullptr) {}
 
  protected:
   void SetUpInProcessBrowserTestFixture() override {
     WizardInProcessBrowserTest::SetUpInProcessBrowserTestFixture();
 
@@ skipping 85 matching lines @@
   // EXPECT_FALSE(view_->IsContinueEnabled());
   // EXPECT_FALSE(view_->IsConnecting());
   network_screen_->OnConnectionTimeout();
 
   // Close infobubble with error message - it makes the test stable.
   // EXPECT_FALSE(view_->IsContinueEnabled());
   // EXPECT_FALSE(view_->IsConnecting());
   // view_->ClearErrors();
 }
 
-class HandsOffNetworkScreenTest : public NetworkScreenTest {
- public:
-  HandsOffNetworkScreenTest() {}
-
- protected:
-  void SetUpOnMainThread() override {
-    NetworkScreenTest::SetUpOnMainThread();
-
-    // Set up fake networks.
-    DBusThreadManager::Get()
-        ->GetShillManagerClient()
-        ->GetTestInterface()
-        ->SetupDefaultEnvironment();
-  }
-
- private:
-  // Overridden from InProcessBrowserTest:
-  void SetUpCommandLine(base::CommandLine* command_line) override {
-    command_line->AppendSwitchASCII(
-        switches::kEnterpriseEnableZeroTouchEnrollment, "hands-off");
-  }
-
-  DISALLOW_COPY_AND_ASSIGN(HandsOffNetworkScreenTest);
-};
-
-#if defined(OS_CHROMEOS)
-// Flaky on ChromeOS. See https://crbug.com/674782.
-#define MAYBE_RequiresNoInput DISABLED_RequiresNoInput
-#else
-#define MAYBE_RequiresNoInput RequiresNoInput
-#endif
-IN_PROC_BROWSER_TEST_F(HandsOffNetworkScreenTest, MAYBE_RequiresNoInput) {
-  WizardController* wizard_controller = WizardController::default_controller();
-
-  // Allow the WizardController to advance throught the enrollment flow.
-  network_screen_->base_screen_delegate_ = wizard_controller;
-
-  // Simulate a network connection.
-  EXPECT_CALL(*mock_network_state_helper_, IsConnected())
-      .Times(AnyNumber())
-      .WillRepeatedly((Return(true)));
-  network_screen_->UpdateStatus();
-
-  // Check if NetworkScreen::OnContinueButtonPressed() is called
-  // Note that checking network_screen_->continue_pressed_ is not
-  // sufficient since there are cases where OnContinueButtonPressed()
-  // is called but where this variable is not set to true.
-  ASSERT_TRUE((!network_screen_->is_network_subscribed_) &&
-              network_screen_->network_state_helper_->IsConnected());
-
-  // Check that we reach the enrollment screen.
-  OobeScreenWaiter(OobeScreen::SCREEN_OOBE_ENROLLMENT).Wait();
-
-  // Check that attestation-based enrollment finishes
-  // with either success or error.
-  bool done = false;
-  ASSERT_TRUE(content::ExecuteScriptAndExtractBool(
-      LoginDisplayHost::default_host()->GetOobeUI()->web_ui()->GetWebContents(),
-      "var count = 0;"
-      "function isVisible(el) {"
-      "  return window.getComputedStyle(document.getElementById(el)).display"
-      "    !== 'none';"
-      "}"
-      "function SendReplyIfEnrollmentDone() {"
-      "  if (isVisible('oauth-enroll-step-abe-success') ||"
-      "      isVisible('oauth-enroll-step-error')) {"
-      "    domAutomationController.send(true);"
-      "  } else if (count > 10) {"
-      "    domAutomationController.send(false);"
-      "  } else {"
-      "    count++;"
-      "    setTimeout(SendReplyIfEnrollmentDone, 1000);"
-      "  }"
-      "}"
-      "SendReplyIfEnrollmentDone();",
-      &done));
-
-  // Reset the enrollment helper so there is no side effect with other tests.
-  static_cast<EnrollmentScreen*>(wizard_controller->current_screen())
-      ->enrollment_helper_.reset();
-}
-
-IN_PROC_BROWSER_TEST_F(HandsOffNetworkScreenTest, ContinueClickedOnlyOnce) {
-  WizardController* wizard_controller = WizardController::default_controller();
-
-  // Allow the WizardController to advance through the enrollment flow.
-  network_screen_->base_screen_delegate_ = wizard_controller;
-
-  // Check that OnContinueButtonPressed has not been called yet.
-  ASSERT_EQ(0, network_screen_->continue_attempts_);
-
-  // Connect to network "net0".
-  EXPECT_CALL(*mock_network_state_helper_, GetCurrentNetworkName())
-      .Times(AnyNumber())
-      .WillRepeatedly(Return(base::ASCIIToUTF16("net0")));
-  EXPECT_CALL(*mock_network_state_helper_, IsConnected())
-      .Times(AnyNumber())
-      .WillRepeatedly(Return(true));
-
-  // Stop waiting for net0.
-  network_screen_->StopWaitingForConnection(base::ASCIIToUTF16("net0"));
-
-  // Check that OnContinueButtonPressed has been called exactly once.
-  ASSERT_EQ(1, network_screen_->continue_attempts_);
-
-  // Stop waiting for another network, net1.
-  network_screen_->StopWaitingForConnection(base::ASCIIToUTF16("net1"));
-
-  // Check that OnContinueButtonPressed stil has been called exactly once
-  ASSERT_EQ(1, network_screen_->continue_attempts_);
-
-  // Wait for the enrollment screen.
-  OobeScreenWaiter(OobeScreen::SCREEN_OOBE_ENROLLMENT)
-      .WaitNoAssertCurrentScreen();
-
-  // Reset the enrollment helper so there is no side effect with other tests.
-  static_cast<EnrollmentScreen*>(wizard_controller->current_screen())
-      ->enrollment_helper_.reset();
-}
-
 }  // namespace chromeos