OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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" | 10 #include "chrome/browser/chromeos/cros/mock_cryptohome_library.h" |
11 #include "chrome/browser/chromeos/cros/mock_library_loader.h" | 11 #include "chrome/browser/chromeos/cros/mock_library_loader.h" |
12 #include "chrome/browser/chromeos/cros/mock_network_library.h" | 12 #include "chrome/browser/chromeos/cros/mock_network_library.h" |
13 #include "chrome/browser/chromeos/cros/mock_screen_lock_library.h" | |
14 #include "chrome/browser/chromeos/login/wizard_controller.h" | 13 #include "chrome/browser/chromeos/login/wizard_controller.h" |
15 #include "chrome/browser/chromeos/login/wizard_screen.h" | 14 #include "chrome/browser/chromeos/login/wizard_screen.h" |
16 #include "chrome/test/base/in_process_browser_test.h" | 15 #include "chrome/test/base/in_process_browser_test.h" |
17 #include "chrome/test/base/ui_test_utils.h" | 16 #include "chrome/test/base/ui_test_utils.h" |
18 #include "testing/gmock/include/gmock/gmock.h" | 17 #include "testing/gmock/include/gmock/gmock.h" |
19 #include "testing/gtest/include/gtest/gtest.h" | 18 #include "testing/gtest/include/gtest/gtest.h" |
20 | 19 |
21 namespace chromeos { | 20 namespace chromeos { |
22 | 21 |
23 using ::testing::AnyNumber; | 22 using ::testing::AnyNumber; |
24 using ::testing::AtMost; | 23 using ::testing::AtMost; |
25 using ::testing::InSequence; | 24 using ::testing::InSequence; |
26 using ::testing::InvokeWithoutArgs; | 25 using ::testing::InvokeWithoutArgs; |
27 using ::testing::Return; | 26 using ::testing::Return; |
28 using ::testing::ReturnRef; | 27 using ::testing::ReturnRef; |
29 using ::testing::StrictMock; | 28 using ::testing::StrictMock; |
30 using ::testing::_; | 29 using ::testing::_; |
31 | 30 |
32 CrosMock::CrosMock() | 31 CrosMock::CrosMock() |
33 : loader_(NULL), | 32 : loader_(NULL), |
34 mock_cryptohome_library_(NULL), | 33 mock_cryptohome_library_(NULL), |
35 mock_network_library_(NULL), | 34 mock_network_library_(NULL) { |
36 mock_screen_lock_library_(NULL) { | |
37 } | 35 } |
38 | 36 |
39 CrosMock::~CrosMock() { | 37 CrosMock::~CrosMock() { |
40 } | 38 } |
41 | 39 |
42 chromeos::CrosLibrary::TestApi* CrosMock::test_api() { | 40 chromeos::CrosLibrary::TestApi* CrosMock::test_api() { |
43 return chromeos::CrosLibrary::Get()->GetTestApi(); | 41 return chromeos::CrosLibrary::Get()->GetTestApi(); |
44 } | 42 } |
45 | 43 |
46 void CrosMock::InitStatusAreaMocks() { | 44 void CrosMock::InitStatusAreaMocks() { |
(...skipping 19 matching lines...) Expand all Loading... |
66 } | 64 } |
67 | 65 |
68 void CrosMock::InitMockNetworkLibrary() { | 66 void CrosMock::InitMockNetworkLibrary() { |
69 InitMockLibraryLoader(); | 67 InitMockLibraryLoader(); |
70 if (mock_network_library_) | 68 if (mock_network_library_) |
71 return; | 69 return; |
72 mock_network_library_ = new StrictMock<MockNetworkLibrary>(); | 70 mock_network_library_ = new StrictMock<MockNetworkLibrary>(); |
73 test_api()->SetNetworkLibrary(mock_network_library_, true); | 71 test_api()->SetNetworkLibrary(mock_network_library_, true); |
74 } | 72 } |
75 | 73 |
76 void CrosMock::InitMockScreenLockLibrary() { | |
77 InitMockLibraryLoader(); | |
78 if (mock_screen_lock_library_) | |
79 return; | |
80 mock_screen_lock_library_ = new StrictMock<MockScreenLockLibrary>(); | |
81 test_api()->SetScreenLockLibrary(mock_screen_lock_library_, true); | |
82 } | |
83 | |
84 // Initialization of mocks. | 74 // Initialization of mocks. |
85 MockCryptohomeLibrary* CrosMock::mock_cryptohome_library() { | 75 MockCryptohomeLibrary* CrosMock::mock_cryptohome_library() { |
86 return mock_cryptohome_library_; | 76 return mock_cryptohome_library_; |
87 } | 77 } |
88 | 78 |
89 MockNetworkLibrary* CrosMock::mock_network_library() { | 79 MockNetworkLibrary* CrosMock::mock_network_library() { |
90 return mock_network_library_; | 80 return mock_network_library_; |
91 } | 81 } |
92 | 82 |
93 MockScreenLockLibrary* CrosMock::mock_screen_lock_library() { | |
94 return mock_screen_lock_library_; | |
95 } | |
96 | |
97 void CrosMock::SetStatusAreaMocksExpectations() { | 83 void CrosMock::SetStatusAreaMocksExpectations() { |
98 SetNetworkLibraryStatusAreaExpectations(); | 84 SetNetworkLibraryStatusAreaExpectations(); |
99 } | 85 } |
100 | 86 |
101 void CrosMock::SetNetworkLibraryStatusAreaExpectations() { | 87 void CrosMock::SetNetworkLibraryStatusAreaExpectations() { |
102 // We don't care how often these are called, just set their return values: | 88 // We don't care how often these are called, just set their return values: |
103 EXPECT_CALL(*mock_network_library_, AddNetworkManagerObserver(_)) | 89 EXPECT_CALL(*mock_network_library_, AddNetworkManagerObserver(_)) |
104 .Times(AnyNumber()); | 90 .Times(AnyNumber()); |
105 EXPECT_CALL(*mock_network_library_, AddNetworkDeviceObserver(_, _)) | 91 EXPECT_CALL(*mock_network_library_, AddNetworkDeviceObserver(_, _)) |
106 .Times(AnyNumber()); | 92 .Times(AnyNumber()); |
(...skipping 97 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
204 } | 190 } |
205 | 191 |
206 void CrosMock::TearDownMocks() { | 192 void CrosMock::TearDownMocks() { |
207 // Prevent bogus gMock leak check from firing. | 193 // Prevent bogus gMock leak check from firing. |
208 if (loader_) | 194 if (loader_) |
209 test_api()->SetLibraryLoader(NULL, false); | 195 test_api()->SetLibraryLoader(NULL, false); |
210 if (mock_cryptohome_library_) | 196 if (mock_cryptohome_library_) |
211 test_api()->SetCryptohomeLibrary(NULL, false); | 197 test_api()->SetCryptohomeLibrary(NULL, false); |
212 if (mock_network_library_) | 198 if (mock_network_library_) |
213 test_api()->SetNetworkLibrary(NULL, false); | 199 test_api()->SetNetworkLibrary(NULL, false); |
214 if (mock_screen_lock_library_) | |
215 test_api()->SetScreenLockLibrary(NULL, false); | |
216 } | 200 } |
217 | 201 |
218 } // namespace chromeos | 202 } // namespace chromeos |
OLD | NEW |