| 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 "chrome/browser/chromeos/cros/cros_mock.h" | 5 #include "chrome/browser/chromeos/cros/cros_mock.h" |
| 6 | 6 |
| 7 #include "base/memory/ref_counted.h" | 7 #include "base/memory/ref_counted.h" |
| 8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
| 9 #include "base/time.h" | 9 #include "base/time.h" |
| 10 #include "chrome/browser/chromeos/cros/mock_cryptohome_library.h" | |
| 11 #include "chrome/browser/chromeos/cros/mock_network_library.h" | 10 #include "chrome/browser/chromeos/cros/mock_network_library.h" |
| 12 #include "chrome/browser/chromeos/login/screens/wizard_screen.h" | 11 #include "chrome/browser/chromeos/login/screens/wizard_screen.h" |
| 13 #include "chrome/browser/chromeos/login/wizard_controller.h" | 12 #include "chrome/browser/chromeos/login/wizard_controller.h" |
| 14 #include "chrome/test/base/in_process_browser_test.h" | 13 #include "chrome/test/base/in_process_browser_test.h" |
| 15 #include "chrome/test/base/ui_test_utils.h" | 14 #include "chrome/test/base/ui_test_utils.h" |
| 16 #include "testing/gmock/include/gmock/gmock.h" | 15 #include "testing/gmock/include/gmock/gmock.h" |
| 17 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 18 | 17 |
| 19 namespace chromeos { | 18 namespace chromeos { |
| 20 | 19 |
| 21 using ::testing::AnyNumber; | 20 using ::testing::AnyNumber; |
| 22 using ::testing::AtMost; | 21 using ::testing::AtMost; |
| 23 using ::testing::InSequence; | 22 using ::testing::InSequence; |
| 24 using ::testing::InvokeWithoutArgs; | 23 using ::testing::InvokeWithoutArgs; |
| 25 using ::testing::Return; | 24 using ::testing::Return; |
| 26 using ::testing::ReturnRef; | 25 using ::testing::ReturnRef; |
| 27 using ::testing::StrictMock; | 26 using ::testing::StrictMock; |
| 28 using ::testing::_; | 27 using ::testing::_; |
| 29 | 28 |
| 30 CrosMock::CrosMock() | 29 CrosMock::CrosMock() : mock_network_library_(NULL) { |
| 31 : mock_cryptohome_library_(NULL), | |
| 32 mock_network_library_(NULL) { | |
| 33 } | 30 } |
| 34 | 31 |
| 35 CrosMock::~CrosMock() { | 32 CrosMock::~CrosMock() { |
| 36 } | 33 } |
| 37 | 34 |
| 38 chromeos::CrosLibrary::TestApi* CrosMock::test_api() { | 35 chromeos::CrosLibrary::TestApi* CrosMock::test_api() { |
| 39 return chromeos::CrosLibrary::Get()->GetTestApi(); | 36 return chromeos::CrosLibrary::Get()->GetTestApi(); |
| 40 } | 37 } |
| 41 | 38 |
| 42 void CrosMock::InitStatusAreaMocks() { | 39 void CrosMock::InitStatusAreaMocks() { |
| 43 InitMockNetworkLibrary(); | 40 InitMockNetworkLibrary(); |
| 44 } | 41 } |
| 45 | 42 |
| 46 void CrosMock::InitMockCryptohomeLibrary() { | |
| 47 if (mock_cryptohome_library_) | |
| 48 return; | |
| 49 mock_cryptohome_library_ = new StrictMock<MockCryptohomeLibrary>(); | |
| 50 test_api()->SetCryptohomeLibrary(mock_cryptohome_library_, true); | |
| 51 } | |
| 52 | |
| 53 void CrosMock::InitMockNetworkLibrary() { | 43 void CrosMock::InitMockNetworkLibrary() { |
| 54 if (mock_network_library_) | 44 if (mock_network_library_) |
| 55 return; | 45 return; |
| 56 mock_network_library_ = new StrictMock<MockNetworkLibrary>(); | 46 mock_network_library_ = new StrictMock<MockNetworkLibrary>(); |
| 57 test_api()->SetNetworkLibrary(mock_network_library_, true); | 47 test_api()->SetNetworkLibrary(mock_network_library_, true); |
| 58 } | 48 } |
| 59 | 49 |
| 60 // Initialization of mocks. | 50 // Initialization of mocks. |
| 61 MockCryptohomeLibrary* CrosMock::mock_cryptohome_library() { | |
| 62 return mock_cryptohome_library_; | |
| 63 } | |
| 64 | |
| 65 MockNetworkLibrary* CrosMock::mock_network_library() { | 51 MockNetworkLibrary* CrosMock::mock_network_library() { |
| 66 return mock_network_library_; | 52 return mock_network_library_; |
| 67 } | 53 } |
| 68 | 54 |
| 69 void CrosMock::SetStatusAreaMocksExpectations() { | 55 void CrosMock::SetStatusAreaMocksExpectations() { |
| 70 SetNetworkLibraryStatusAreaExpectations(); | 56 SetNetworkLibraryStatusAreaExpectations(); |
| 71 } | 57 } |
| 72 | 58 |
| 73 void CrosMock::SetNetworkLibraryStatusAreaExpectations() { | 59 void CrosMock::SetNetworkLibraryStatusAreaExpectations() { |
| 74 // We don't care how often these are called, just set their return values: | 60 // We don't care how often these are called, just set their return values: |
| (...skipping 122 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 197 .WillRepeatedly((Return(false))) | 183 .WillRepeatedly((Return(false))) |
| 198 .RetiresOnSaturation(); | 184 .RetiresOnSaturation(); |
| 199 EXPECT_CALL(*mock_network_library_, ethernet_connecting()) | 185 EXPECT_CALL(*mock_network_library_, ethernet_connecting()) |
| 200 .Times(AnyNumber()) | 186 .Times(AnyNumber()) |
| 201 .WillRepeatedly((Return(false))) | 187 .WillRepeatedly((Return(false))) |
| 202 .RetiresOnSaturation(); | 188 .RetiresOnSaturation(); |
| 203 } | 189 } |
| 204 | 190 |
| 205 void CrosMock::TearDownMocks() { | 191 void CrosMock::TearDownMocks() { |
| 206 // Prevent bogus gMock leak check from firing. | 192 // Prevent bogus gMock leak check from firing. |
| 207 if (mock_cryptohome_library_) | |
| 208 test_api()->SetCryptohomeLibrary(NULL, false); | |
| 209 if (mock_network_library_) | 193 if (mock_network_library_) |
| 210 test_api()->SetNetworkLibrary(NULL, false); | 194 test_api()->SetNetworkLibrary(NULL, false); |
| 211 } | 195 } |
| 212 | 196 |
| 213 } // namespace chromeos | 197 } // namespace chromeos |
| OLD | NEW |