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/fake_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 21 using ::testing::_; 22 using ::testing::AnyNumber; 23 using ::testing::Return; 24 using ::testing::ReturnRef; 25 using views::Button; 26 27 namespace chromeos { 28 29 class DummyButtonListener : public views::ButtonListener { 30 public: 31 virtual void ButtonPressed(views::Button* sender, 32 const ui::Event& event) OVERRIDE {} 33 }; 34 35 namespace login { 36 37 class 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 46 class 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 FakeDBusThreadManager* fake_dbus_thread_manager = new FakeDBusThreadManager; 57 fake_dbus_thread_manager->SetFakeClients(); 58 fake_session_manager_client_ = new FakeSessionManagerClient; 59 fake_dbus_thread_manager->SetSessionManagerClient( 60 scoped_ptr<SessionManagerClient>(fake_session_manager_client_)); 61 DBusThreadManager::SetInstanceForTesting(fake_dbus_thread_manager); 62 } 63 64 virtual void SetUpOnMainThread() OVERRIDE { 65 WizardInProcessBrowserTest::SetUpOnMainThread(); 66 mock_screen_observer_.reset(new MockScreenObserver()); 67 ASSERT_TRUE(WizardController::default_controller() != NULL); 68 network_screen_ = 69 WizardController::default_controller()->GetNetworkScreen(); 70 ASSERT_TRUE(network_screen_ != NULL); 71 ASSERT_EQ(WizardController::default_controller()->current_screen(), 72 network_screen_); 73 network_screen_->screen_observer_ = mock_screen_observer_.get(); 74 ASSERT_TRUE(network_screen_->actor() != NULL); 75 76 mock_network_state_helper_ = new login::MockNetworkStateHelper; 77 SetDefaultNetworkStateHelperExpectations(); 78 network_screen_->SetNetworkStateHelperForTest(mock_network_state_helper_); 79 } 80 81 virtual void TearDownInProcessBrowserTestFixture() OVERRIDE { 82 InProcessBrowserTest::TearDownInProcessBrowserTestFixture(); 83 } 84 85 void EmulateContinueButtonExit(NetworkScreen* network_screen) { 86 EXPECT_CALL(*mock_screen_observer_, 87 OnExit(ScreenObserver::NETWORK_CONNECTED)) 88 .Times(1); 89 EXPECT_CALL(*mock_network_state_helper_, IsConnected()) 90 .WillOnce(Return(true)); 91 network_screen->OnContinuePressed(); 92 content::RunAllPendingInMessageLoop(); 93 } 94 95 void SetDefaultNetworkStateHelperExpectations() { 96 EXPECT_CALL(*mock_network_state_helper_, GetCurrentNetworkName()) 97 .Times(AnyNumber()) 98 .WillRepeatedly((Return(base::string16()))); 99 EXPECT_CALL(*mock_network_state_helper_, IsConnected()) 100 .Times(AnyNumber()) 101 .WillRepeatedly((Return(false))); 102 EXPECT_CALL(*mock_network_state_helper_, IsConnecting()) 103 .Times(AnyNumber()) 104 .WillRepeatedly((Return(false))); 105 } 106 107 scoped_ptr<MockScreenObserver> mock_screen_observer_; 108 login::MockNetworkStateHelper* mock_network_state_helper_; 109 NetworkScreen* network_screen_; 110 FakeSessionManagerClient* fake_session_manager_client_; 111 112 private: 113 DISALLOW_COPY_AND_ASSIGN(NetworkScreenTest); 114 }; 115 116 IN_PROC_BROWSER_TEST_F(NetworkScreenTest, CanConnect) { 117 EXPECT_CALL(*mock_network_state_helper_, IsConnecting()) 118 .WillOnce((Return(true))); 119 // EXPECT_FALSE(actor_->IsContinueEnabled()); 120 network_screen_->UpdateStatus(); 121 122 EXPECT_CALL(*mock_network_state_helper_, IsConnected()) 123 .Times(2) 124 .WillRepeatedly(Return(true)); 125 // TODO(nkostylev): Add integration with WebUI actor http://crosbug.com/22570 126 // EXPECT_FALSE(actor_->IsContinueEnabled()); 127 // EXPECT_FALSE(actor_->IsConnecting()); 128 network_screen_->UpdateStatus(); 129 130 // EXPECT_TRUE(actor_->IsContinueEnabled()); 131 EmulateContinueButtonExit(network_screen_); 132 } 133 134 IN_PROC_BROWSER_TEST_F(NetworkScreenTest, Timeout) { 135 EXPECT_CALL(*mock_network_state_helper_, IsConnecting()) 136 .WillOnce((Return(true))); 137 // EXPECT_FALSE(actor_->IsContinueEnabled()); 138 network_screen_->UpdateStatus(); 139 140 EXPECT_CALL(*mock_network_state_helper_, IsConnected()) 141 .Times(2) 142 .WillRepeatedly(Return(false)); 143 // TODO(nkostylev): Add integration with WebUI actor http://crosbug.com/22570 144 // EXPECT_FALSE(actor_->IsContinueEnabled()); 145 // EXPECT_FALSE(actor_->IsConnecting()); 146 network_screen_->OnConnectionTimeout(); 147 148 // Close infobubble with error message - it makes the test stable. 149 // EXPECT_FALSE(actor_->IsContinueEnabled()); 150 // EXPECT_FALSE(actor_->IsConnecting()); 151 // actor_->ClearErrors(); 152 } 153 154 } // namespace chromeos 155