| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/message_loop.h" | 5 #include "base/memory/scoped_ptr.h" |
| 6 #include "chrome/browser/chromeos/cros/mock_network_library.h" | 6 #include "chrome/browser/chromeos/cros/mock_network_library.h" |
| 7 #include "chrome/browser/chromeos/login/mock_screen_observer.h" | 7 #include "chrome/browser/chromeos/login/mock_screen_observer.h" |
| 8 #include "chrome/browser/chromeos/login/update_screen.h" | 8 #include "chrome/browser/chromeos/login/update_screen.h" |
| 9 #include "chrome/browser/chromeos/login/wizard_controller.h" | 9 #include "chrome/browser/chromeos/login/wizard_controller.h" |
| 10 #include "chrome/browser/chromeos/login/wizard_in_process_browser_test.h" | 10 #include "chrome/browser/chromeos/login/wizard_in_process_browser_test.h" |
| 11 #include "chromeos/dbus/mock_dbus_thread_manager.h" | 11 #include "chromeos/dbus/mock_dbus_thread_manager.h" |
| 12 #include "chromeos/dbus/mock_session_manager_client.h" | 12 #include "chromeos/dbus/mock_session_manager_client.h" |
| 13 #include "chromeos/dbus/mock_update_engine_client.h" | 13 #include "chromeos/dbus/mock_update_engine_client.h" |
| 14 #include "testing/gmock/include/gmock/gmock.h" |
| 14 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
| 15 | 16 |
| 16 namespace chromeos { | |
| 17 using ::testing::_; | 17 using ::testing::_; |
| 18 using ::testing::AnyNumber; | 18 using ::testing::AnyNumber; |
| 19 using ::testing::AtLeast; | 19 using ::testing::AtLeast; |
| 20 using ::testing::Invoke; |
| 20 using ::testing::Return; | 21 using ::testing::Return; |
| 21 using ::testing::ReturnRef; | 22 |
| 22 using ::testing::Invoke; | 23 namespace chromeos { |
| 23 using chromeos::UpdateEngineClient; | 24 |
| 25 namespace { |
| 24 | 26 |
| 25 static void RequestUpdateCheckSuccess( | 27 static void RequestUpdateCheckSuccess( |
| 26 UpdateEngineClient::UpdateCheckCallback callback) { | 28 UpdateEngineClient::UpdateCheckCallback callback) { |
| 27 callback.Run(UpdateEngineClient::UPDATE_RESULT_SUCCESS); | 29 callback.Run(UpdateEngineClient::UPDATE_RESULT_SUCCESS); |
| 28 } | 30 } |
| 29 | 31 |
| 32 } // namespace |
| 33 |
| 30 class UpdateScreenTest : public WizardInProcessBrowserTest { | 34 class UpdateScreenTest : public WizardInProcessBrowserTest { |
| 31 public: | 35 public: |
| 32 UpdateScreenTest() : WizardInProcessBrowserTest("update"), | 36 UpdateScreenTest() : WizardInProcessBrowserTest("update"), |
| 33 mock_update_engine_client_(NULL), | 37 mock_update_engine_client_(NULL), |
| 34 mock_network_library_(NULL) {} | 38 mock_network_library_(NULL) {} |
| 35 | 39 |
| 36 protected: | 40 protected: |
| 37 virtual void SetUpInProcessBrowserTestFixture() { | 41 virtual void SetUpInProcessBrowserTestFixture() { |
| 38 MockDBusThreadManager* mock_dbus_thread_manager = | 42 MockDBusThreadManager* mock_dbus_thread_manager = |
| 39 new MockDBusThreadManager; | 43 new MockDBusThreadManager; |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 72 .RetiresOnSaturation(); | 76 .RetiresOnSaturation(); |
| 73 EXPECT_CALL(*mock_network_library_, AddNetworkManagerObserver(_)) | 77 EXPECT_CALL(*mock_network_library_, AddNetworkManagerObserver(_)) |
| 74 .Times(1) | 78 .Times(1) |
| 75 .RetiresOnSaturation(); | 79 .RetiresOnSaturation(); |
| 76 EXPECT_CALL(*mock_network_library_, AddUserActionObserver(_)) | 80 EXPECT_CALL(*mock_network_library_, AddUserActionObserver(_)) |
| 77 .Times(AnyNumber()); | 81 .Times(AnyNumber()); |
| 78 EXPECT_CALL(*mock_network_library_, FindWifiDevice()) | 82 EXPECT_CALL(*mock_network_library_, FindWifiDevice()) |
| 79 .Times(AnyNumber()); | 83 .Times(AnyNumber()); |
| 80 EXPECT_CALL(*mock_network_library_, FindEthernetDevice()) | 84 EXPECT_CALL(*mock_network_library_, FindEthernetDevice()) |
| 81 .Times(AnyNumber()); | 85 .Times(AnyNumber()); |
| 86 EXPECT_CALL(*mock_network_library_, LoadOncNetworks(_, _, _, _)) |
| 87 .WillRepeatedly(Return(true)); |
| 82 } | 88 } |
| 83 | 89 |
| 84 virtual void SetUpOnMainThread() { | 90 virtual void SetUpOnMainThread() { |
| 85 WizardInProcessBrowserTest::SetUpOnMainThread(); | 91 WizardInProcessBrowserTest::SetUpOnMainThread(); |
| 86 mock_screen_observer_.reset(new MockScreenObserver()); | 92 mock_screen_observer_.reset(new MockScreenObserver()); |
| 87 ASSERT_TRUE(WizardController::default_controller() != NULL); | 93 ASSERT_TRUE(WizardController::default_controller() != NULL); |
| 88 update_screen_ = WizardController::default_controller()->GetUpdateScreen(); | 94 update_screen_ = WizardController::default_controller()->GetUpdateScreen(); |
| 89 ASSERT_TRUE(update_screen_ != NULL); | 95 ASSERT_TRUE(update_screen_ != NULL); |
| 90 ASSERT_EQ(WizardController::default_controller()->current_screen(), | 96 ASSERT_EQ(WizardController::default_controller()->current_screen(), |
| 91 update_screen_); | 97 update_screen_); |
| (...skipping 125 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 217 EXPECT_CALL(*mock_update_engine_client_, GetLastStatus()) | 223 EXPECT_CALL(*mock_update_engine_client_, GetLastStatus()) |
| 218 .Times(AtLeast(1)) | 224 .Times(AtLeast(1)) |
| 219 .WillRepeatedly(Return(status)); | 225 .WillRepeatedly(Return(status)); |
| 220 EXPECT_CALL(*mock_screen_observer_, | 226 EXPECT_CALL(*mock_screen_observer_, |
| 221 OnExit(ScreenObserver::UPDATE_ERROR_UPDATING)) | 227 OnExit(ScreenObserver::UPDATE_ERROR_UPDATING)) |
| 222 .Times(1); | 228 .Times(1); |
| 223 update_screen_->UpdateStatusChanged(status); | 229 update_screen_->UpdateStatusChanged(status); |
| 224 } | 230 } |
| 225 | 231 |
| 226 } // namespace chromeos | 232 } // namespace chromeos |
| OLD | NEW |