1// Copyright (c) 2012 The Chromium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5#include "base/memory/scoped_ptr.h" 6#include "chrome/browser/chromeos/login/helper.h" 7#include "chrome/browser/chromeos/login/screens/mock_screen_observer.h" 8#include "chrome/browser/chromeos/login/screens/network_screen.h" 9#include "chrome/browser/chromeos/login/screens/wizard_screen.h" 10#include "chrome/browser/chromeos/login/test/wizard_in_process_browser_test.h" 11#include "chrome/browser/chromeos/login/wizard_controller.h" 12#include "chrome/test/base/ui_test_utils.h" 13#include "chromeos/dbus/dbus_thread_manager.h" 14#include "chromeos/dbus/fake_session_manager_client.h" 15#include "content/public/test/test_utils.h" 16#include "testing/gmock/include/gmock/gmock.h" 17#include "testing/gtest/include/gtest/gtest.h" 18#include "third_party/cros_system_api/dbus/service_constants.h" 19#include "ui/views/controls/button/button.h" 20 21using ::testing::_; 22using ::testing::AnyNumber; 23using ::testing::Return; 24using ::testing::ReturnRef; 25using views::Button; 26 27namespace chromeos { 28 29class DummyButtonListener : public views::ButtonListener { 30 public: 31 virtual void ButtonPressed(views::Button* sender, 32 const ui::Event& event) OVERRIDE {} 33}; 34 35namespace login { 36 37class MockNetworkStateHelper : public NetworkStateHelper { 38 public: 39 MOCK_CONST_METHOD0(GetCurrentNetworkName, base::string16(void)); 40 MOCK_CONST_METHOD0(IsConnected, bool(void)); 41 MOCK_CONST_METHOD0(IsConnecting, bool(void)); 42}; 43 44} // namespace login 45 46class NetworkScreenTest : public WizardInProcessBrowserTest { 47 public: 48 NetworkScreenTest(): WizardInProcessBrowserTest("network"), 49 fake_session_manager_client_(NULL) { 50 } 51 52 protected: 53 virtual void SetUpInProcessBrowserTestFixture() OVERRIDE { 54 WizardInProcessBrowserTest::SetUpInProcessBrowserTestFixture(); 55 56 fake_session_manager_client_ = new FakeSessionManagerClient; 57 DBusThreadManager::GetSetterForTesting()->SetSessionManagerClient( 58 scoped_ptr<SessionManagerClient>(fake_session_manager_client_)); 59 } 60 61 virtual void SetUpOnMainThread() OVERRIDE { 62 WizardInProcessBrowserTest::SetUpOnMainThread(); 63 mock_screen_observer_.reset(new MockScreenObserver()); 64 ASSERT_TRUE(WizardController::default_controller() != NULL); 65 network_screen_ = 66 NetworkScreen::Get(WizardController::default_controller()); 67 ASSERT_TRUE(network_screen_ != NULL); 68 ASSERT_EQ(WizardController::default_controller()->current_screen(), 69 network_screen_); 70 network_screen_->screen_observer_ = mock_screen_observer_.get(); 71 ASSERT_TRUE(network_screen_->actor() != NULL); 72 73 mock_network_state_helper_ = new login::MockNetworkStateHelper; 74 SetDefaultNetworkStateHelperExpectations(); 75 network_screen_->SetNetworkStateHelperForTest(mock_network_state_helper_); 76 } 77 78 virtual void TearDownInProcessBrowserTestFixture() OVERRIDE { 79 InProcessBrowserTest::TearDownInProcessBrowserTestFixture(); 80 } 81 82 void EmulateContinueButtonExit(NetworkScreen* network_screen) { 83 EXPECT_CALL(*mock_screen_observer_, 84 OnExit(ScreenObserver::NETWORK_CONNECTED)) 85 .Times(1); 86 EXPECT_CALL(*mock_network_state_helper_, IsConnected()) 87 .WillOnce(Return(true)); 88 network_screen->OnContinuePressed(); 89 content::RunAllPendingInMessageLoop(); 90 } 91 92 void SetDefaultNetworkStateHelperExpectations() { 93 EXPECT_CALL(*mock_network_state_helper_, GetCurrentNetworkName()) 94 .Times(AnyNumber()) 95 .WillRepeatedly((Return(base::string16()))); 96 EXPECT_CALL(*mock_network_state_helper_, IsConnected()) 97 .Times(AnyNumber()) 98 .WillRepeatedly((Return(false))); 99 EXPECT_CALL(*mock_network_state_helper_, IsConnecting()) 100 .Times(AnyNumber()) 101 .WillRepeatedly((Return(false))); 102 } 103 104 scoped_ptr<MockScreenObserver> mock_screen_observer_; 105 login::MockNetworkStateHelper* mock_network_state_helper_; 106 NetworkScreen* network_screen_; 107 FakeSessionManagerClient* fake_session_manager_client_; 108 109 private: 110 DISALLOW_COPY_AND_ASSIGN(NetworkScreenTest); 111}; 112 113IN_PROC_BROWSER_TEST_F(NetworkScreenTest, CanConnect) { 114 EXPECT_CALL(*mock_network_state_helper_, IsConnecting()) 115 .WillOnce((Return(true))); 116 // EXPECT_FALSE(actor_->IsContinueEnabled()); 117 network_screen_->UpdateStatus(); 118 119 EXPECT_CALL(*mock_network_state_helper_, IsConnected()) 120 .Times(2) 121 .WillRepeatedly(Return(true)); 122 // TODO(nkostylev): Add integration with WebUI actor http://crosbug.com/22570 123 // EXPECT_FALSE(actor_->IsContinueEnabled()); 124 // EXPECT_FALSE(actor_->IsConnecting()); 125 network_screen_->UpdateStatus(); 126 127 // EXPECT_TRUE(actor_->IsContinueEnabled()); 128 EmulateContinueButtonExit(network_screen_); 129} 130 131IN_PROC_BROWSER_TEST_F(NetworkScreenTest, Timeout) { 132 EXPECT_CALL(*mock_network_state_helper_, IsConnecting()) 133 .WillOnce((Return(true))); 134 // EXPECT_FALSE(actor_->IsContinueEnabled()); 135 network_screen_->UpdateStatus(); 136 137 EXPECT_CALL(*mock_network_state_helper_, IsConnected()) 138 .Times(2) 139 .WillRepeatedly(Return(false)); 140 // TODO(nkostylev): Add integration with WebUI actor http://crosbug.com/22570 141 // EXPECT_FALSE(actor_->IsContinueEnabled()); 142 // EXPECT_FALSE(actor_->IsConnecting()); 143 network_screen_->OnConnectionTimeout(); 144 145 // Close infobubble with error message - it makes the test stable. 146 // EXPECT_FALSE(actor_->IsContinueEnabled()); 147 // EXPECT_FALSE(actor_->IsConnecting()); 148 // actor_->ClearErrors(); 149} 150 151} // namespace chromeos 152