OLD | NEW |
(Empty) | |
| 1 // Copyright 2017 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/command_line.h" |
| 6 #include "base/message_loop/message_loop.h" |
| 7 #include "base/strings/utf_string_conversions.h" |
| 8 #include "base/test/scoped_mock_time_message_loop_task_runner.h" |
| 9 #include "chrome/browser/chromeos/input_method/input_method_configuration.h" |
| 10 #include "chrome/browser/chromeos/input_method/mock_input_method_manager_impl.h" |
| 11 #include "chrome/browser/chromeos/login/mock_network_state_helper.h" |
| 12 #include "chrome/browser/chromeos/login/screens/mock_base_screen_delegate.h" |
| 13 #include "chrome/browser/chromeos/login/screens/mock_model_view_channel.h" |
| 14 #include "chrome/browser/chromeos/login/screens/mock_network_screen.h" |
| 15 #include "chrome/browser/chromeos/settings/cros_settings.h" |
| 16 #include "chrome/browser/chromeos/settings/device_settings_service.h" |
| 17 #include "chrome/test/base/testing_browser_process.h" |
| 18 #include "chromeos/chromeos_switches.h" |
| 19 #include "chromeos/dbus/dbus_thread_manager.h" |
| 20 #include "chromeos/system/fake_statistics_provider.h" |
| 21 #include "content/public/test/test_browser_thread_bundle.h" |
| 22 #include "testing/gtest/include/gtest/gtest.h" |
| 23 #include "ui/base/ime/chromeos/mock_component_extension_ime_manager.h" |
| 24 |
| 25 using testing::_; |
| 26 using testing::AnyNumber; |
| 27 using testing::Return; |
| 28 |
| 29 namespace chromeos { |
| 30 |
| 31 class NetworkScreenUnitTest : public testing::Test { |
| 32 public: |
| 33 NetworkScreenUnitTest() {} |
| 34 |
| 35 base::ScopedMockTimeMessageLoopTaskRunner* GetTestMessageLoopTaskRunner() { |
| 36 return &runner_; |
| 37 } |
| 38 |
| 39 void FastForwardTime(base::TimeDelta time) { |
| 40 runner_.task_runner()->FastForwardBy(time); |
| 41 } |
| 42 |
| 43 // testing::Test: |
| 44 void SetUp() override { |
| 45 // Initialize the thread manager. |
| 46 DBusThreadManager::Initialize(); |
| 47 |
| 48 // Configure the browser to use Hands-Off Enrollment. |
| 49 base::CommandLine::ForCurrentProcess()->AppendSwitchASCII( |
| 50 switches::kEnterpriseEnableZeroTouchEnrollment, "hands-off"); |
| 51 |
| 52 // Replace the regular InputMethodManager with a mock. |
| 53 input_method::MockInputMethodManagerImpl* mock_input_manager = |
| 54 new input_method::MockInputMethodManagerImpl(); |
| 55 mock_input_manager->SetComponentExtensionIMEManager( |
| 56 std::unique_ptr<MockComponentExtensionIMEManager>( |
| 57 new MockComponentExtensionIMEManager())); |
| 58 input_method::InitializeForTesting(mock_input_manager); |
| 59 |
| 60 // Create the NetworkScreen we will use for testing. |
| 61 network_screen_.reset( |
| 62 new NetworkScreen(&mock_base_screen_delegate_, nullptr, &mock_view_)); |
| 63 network_screen_->set_model_view_channel(&mock_channel_); |
| 64 mock_network_state_helper_ = new login::MockNetworkStateHelper(); |
| 65 network_screen_->SetNetworkStateHelperForTest(mock_network_state_helper_); |
| 66 } |
| 67 |
| 68 void TearDown() override { |
| 69 TestingBrowserProcess::GetGlobal()->SetShuttingDown(true); |
| 70 network_screen_.reset(); |
| 71 input_method::Shutdown(); |
| 72 DBusThreadManager::Shutdown(); |
| 73 } |
| 74 |
| 75 protected: |
| 76 // A pointer to the NetworkScreen. |
| 77 std::unique_ptr<NetworkScreen> network_screen_; |
| 78 |
| 79 // Accessory objects needed by NetworkScreen. |
| 80 MockBaseScreenDelegate mock_base_screen_delegate_; |
| 81 login::MockNetworkStateHelper* mock_network_state_helper_ = nullptr; |
| 82 |
| 83 private: |
| 84 // Test versions of core browser infrastructure. |
| 85 content::TestBrowserThreadBundle threads_; |
| 86 base::ScopedMockTimeMessageLoopTaskRunner runner_; |
| 87 |
| 88 // More accessory objects needed by NetworkScreen. |
| 89 MockNetworkView mock_view_; |
| 90 MockModelViewChannel mock_channel_; |
| 91 |
| 92 // Scoped test versions of required global objects. |
| 93 ScopedTestDeviceSettingsService device_settings_; |
| 94 ScopedTestCrosSettings cros_settings_; |
| 95 system::ScopedFakeStatisticsProvider provider_; |
| 96 |
| 97 DISALLOW_COPY_AND_ASSIGN(NetworkScreenUnitTest); |
| 98 }; |
| 99 |
| 100 TEST_F(NetworkScreenUnitTest, ContinuesAutomatically) { |
| 101 // Verify that we are using the right TaskRunner. |
| 102 EXPECT_EQ(GetTestMessageLoopTaskRunner()->task_runner(), |
| 103 base::MessageLoop::current()->task_runner().get()); |
| 104 |
| 105 // Set expectation that NetworkScreen will finish. |
| 106 EXPECT_CALL(mock_base_screen_delegate_, |
| 107 OnExit(_, ScreenExitCode::NETWORK_CONNECTED, _)) |
| 108 .Times(1); |
| 109 |
| 110 // Simulate a network connection. |
| 111 EXPECT_CALL(*mock_network_state_helper_, IsConnected()) |
| 112 .Times(AnyNumber()) |
| 113 .WillRepeatedly((Return(true))); |
| 114 network_screen_->UpdateStatus(); |
| 115 |
| 116 // Fast forward time by 3 minutes. |
| 117 FastForwardTime(base::TimeDelta::FromMinutes(3)); |
| 118 |
| 119 // Check that we continued once |
| 120 EXPECT_EQ(1, network_screen_->continue_attempts_); |
| 121 } |
| 122 |
| 123 TEST_F(NetworkScreenUnitTest, ContinuesOnlyOnce) { |
| 124 // Verify that we are using the right TaskRunner. |
| 125 EXPECT_EQ(GetTestMessageLoopTaskRunner()->task_runner(), |
| 126 base::MessageLoop::current()->task_runner().get()); |
| 127 |
| 128 // Set expectation that NetworkScreen will finish. |
| 129 EXPECT_CALL(mock_base_screen_delegate_, |
| 130 OnExit(_, ScreenExitCode::NETWORK_CONNECTED, _)) |
| 131 .Times(1); |
| 132 |
| 133 // Connect to network "net0". |
| 134 EXPECT_CALL(*mock_network_state_helper_, GetCurrentNetworkName()) |
| 135 .Times(AnyNumber()) |
| 136 .WillRepeatedly(Return(base::ASCIIToUTF16("net0"))); |
| 137 EXPECT_CALL(*mock_network_state_helper_, IsConnected()) |
| 138 .Times(AnyNumber()) |
| 139 .WillRepeatedly(Return(true)); |
| 140 |
| 141 // Stop waiting for net0. |
| 142 network_screen_->StopWaitingForConnection(base::ASCIIToUTF16("net0")); |
| 143 |
| 144 // Fast forward time by 3 minutes. |
| 145 FastForwardTime(base::TimeDelta::FromMinutes(3)); |
| 146 |
| 147 // Check that we have continued exactly once. |
| 148 ASSERT_EQ(1, network_screen_->continue_attempts_); |
| 149 |
| 150 // Stop waiting for another network, net1. |
| 151 network_screen_->StopWaitingForConnection(base::ASCIIToUTF16("net1")); |
| 152 |
| 153 // Fast forward time by 3 minutes. |
| 154 FastForwardTime(base::TimeDelta::FromMinutes(3)); |
| 155 |
| 156 // Check that we have still continued only once. |
| 157 EXPECT_EQ(1, network_screen_->continue_attempts_); |
| 158 } |
| 159 |
| 160 } // namespace chromeos |
OLD | NEW |