Home | History | Annotate | Download | only in screens 
      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/wizard_controller.h"
     11 #include "chrome/browser/chromeos/login/wizard_in_process_browser_test.h"
     12 #include "chrome/test/base/ui_test_utils.h"
     13 #include "chromeos/dbus/fake_session_manager_client.h"
     14 #include "chromeos/dbus/mock_dbus_thread_manager_without_gmock.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, 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     MockDBusThreadManagerWithoutGMock* mock_dbus_thread_manager =
     57         new MockDBusThreadManagerWithoutGMock;
     58     DBusThreadManager::InitializeForTesting(mock_dbus_thread_manager);
     59     fake_session_manager_client_ =
     60         mock_dbus_thread_manager->fake_session_manager_client();
     61   }
     62 
     63   virtual void SetUpOnMainThread() OVERRIDE {
     64     WizardInProcessBrowserTest::SetUpOnMainThread();
     65     mock_screen_observer_.reset(new MockScreenObserver());
     66     ASSERT_TRUE(WizardController::default_controller() != NULL);
     67     network_screen_ =
     68         WizardController::default_controller()->GetNetworkScreen();
     69     ASSERT_TRUE(network_screen_ != NULL);
     70     ASSERT_EQ(WizardController::default_controller()->current_screen(),
     71               network_screen_);
     72     network_screen_->screen_observer_ = mock_screen_observer_.get();
     73     ASSERT_TRUE(network_screen_->actor() != NULL);
     74 
     75     mock_network_state_helper_ = new login::MockNetworkStateHelper;
     76     SetDefaultNetworkStateHelperExpectations();
     77     network_screen_->SetNetworkStateHelperForTest(mock_network_state_helper_);
     78   }
     79 
     80   virtual void TearDownInProcessBrowserTestFixture() OVERRIDE {
     81     CrosInProcessBrowserTest::TearDownInProcessBrowserTestFixture();
     82     DBusThreadManager::Shutdown();
     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(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_->NetworkManagerChanged();
    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_->NetworkManagerChanged();
    129 
    130   // EXPECT_TRUE(actor_->IsContinueEnabled());
    131   EmulateContinueButtonExit(network_screen_);
    132   EXPECT_EQ(
    133       1, fake_session_manager_client_->emit_login_prompt_ready_call_count());
    134 }
    135 
    136 IN_PROC_BROWSER_TEST_F(NetworkScreenTest, Timeout) {
    137   EXPECT_CALL(*mock_network_state_helper_, IsConnecting())
    138       .WillOnce((Return(true)));
    139   // EXPECT_FALSE(actor_->IsContinueEnabled());
    140   network_screen_->NetworkManagerChanged();
    141 
    142   EXPECT_CALL(*mock_network_state_helper_, IsConnected())
    143       .Times(2)
    144       .WillRepeatedly(Return(false));
    145   // TODO(nkostylev): Add integration with WebUI actor http://crosbug.com/22570
    146   // EXPECT_FALSE(actor_->IsContinueEnabled());
    147   // EXPECT_FALSE(actor_->IsConnecting());
    148   network_screen_->OnConnectionTimeout();
    149 
    150   // Close infobubble with error message - it makes the test stable.
    151   // EXPECT_FALSE(actor_->IsContinueEnabled());
    152   // EXPECT_FALSE(actor_->IsConnecting());
    153   // actor_->ClearErrors();
    154   EXPECT_EQ(
    155       1, fake_session_manager_client_->emit_login_prompt_ready_call_count());
    156 }
    157 
    158 }  // namespace chromeos
    159