network_screen_browsertest.cc revision 8bcbed890bc3ce4d7a057a8f32cab53fa534672e 
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_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, 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    FakeDBusThreadManager* fake_dbus_thread_manager =
57        new FakeDBusThreadManager;
58    DBusThreadManager::InitializeForTesting(fake_dbus_thread_manager);
59    fake_session_manager_client_ =
60        fake_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    InProcessBrowserTest::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
116IN_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  EXPECT_EQ(
133      1, fake_session_manager_client_->emit_login_prompt_ready_call_count());
134}
135
136IN_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_->UpdateStatus();
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