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/managed_mode.h" | 5 #include "chrome/browser/managed_mode.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 #include <vector> | 8 #include <vector> |
9 | 9 |
10 #include "base/bind.h" | 10 #include "base/bind.h" |
11 #include "base/bind_helpers.h" | 11 #include "base/bind_helpers.h" |
12 #include "base/message_loop.h" | 12 #include "base/message_loop.h" |
13 #include "chrome/browser/managed_mode.h" | 13 #include "chrome/browser/managed_mode.h" |
14 #include "chrome/browser/ui/browser.h" | 14 #include "chrome/browser/ui/browser.h" |
| 15 #include "chrome/browser/webdata/web_data_service_factory.h" |
15 #include "chrome/test/base/test_browser_window.h" | 16 #include "chrome/test/base/test_browser_window.h" |
16 #include "chrome/test/base/testing_profile.h" | 17 #include "chrome/test/base/testing_profile.h" |
17 #include "content/test/test_browser_thread.h" | 18 #include "content/test/test_browser_thread.h" |
18 #include "testing/gtest/include/gtest/gtest.h" | 19 #include "testing/gtest/include/gtest/gtest.h" |
19 #include "testing/gmock/include/gmock/gmock.h" | 20 #include "testing/gmock/include/gmock/gmock.h" |
20 | 21 |
21 using ::testing::StrictMock; | 22 using ::testing::StrictMock; |
22 | 23 |
23 namespace { | 24 namespace { |
24 | 25 |
(...skipping 82 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
107 FakeManagedMode* managed_mode_; | 108 FakeManagedMode* managed_mode_; |
108 | 109 |
109 DISALLOW_COPY_AND_ASSIGN(MockCallback); | 110 DISALLOW_COPY_AND_ASSIGN(MockCallback); |
110 }; | 111 }; |
111 | 112 |
112 } // namespace | 113 } // namespace |
113 | 114 |
114 class ManagedModeTest : public ::testing::Test { | 115 class ManagedModeTest : public ::testing::Test { |
115 public: | 116 public: |
116 ManagedModeTest() : ui_thread_(content::BrowserThread::UI, &message_loop_) { | 117 ManagedModeTest() : ui_thread_(content::BrowserThread::UI, &message_loop_) { |
| 118 // WebDataService is not used in this test and should be NULL. |
| 119 WebDataServiceFactory::GetInstance()->SetTestingFactory( |
| 120 &managed_mode_profile_, NULL); |
| 121 WebDataServiceFactory::GetInstance()->SetTestingFactory( |
| 122 &other_profile_, NULL); |
117 } | 123 } |
118 | 124 |
119 scoped_refptr<MockCallback> CreateCallback() { | 125 scoped_refptr<MockCallback> CreateCallback() { |
120 return new StrictMock<MockCallback>(&managed_mode_); | 126 return new StrictMock<MockCallback>(&managed_mode_); |
121 } | 127 } |
122 | 128 |
123 base::Callback<void(bool)> CreateExpectedCallback(bool success) { | 129 base::Callback<void(bool)> CreateExpectedCallback(bool success) { |
124 scoped_refptr<MockCallback> callback = CreateCallback(); | 130 scoped_refptr<MockCallback> callback = CreateCallback(); |
125 EXPECT_CALL(*callback, DidEnterManagedMode(success)); | 131 EXPECT_CALL(*callback, DidEnterManagedMode(success)); |
126 return base::Bind(&MockCallback::CheckManagedMode, callback); | 132 return base::Bind(&MockCallback::CheckManagedMode, callback); |
(...skipping 90 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
217 | 223 |
218 TEST_F(ManagedModeTest, Cancelled) { | 224 TEST_F(ManagedModeTest, Cancelled) { |
219 BrowserFixture managed_mode_browser(&managed_mode_, &managed_mode_profile_); | 225 BrowserFixture managed_mode_browser(&managed_mode_, &managed_mode_profile_); |
220 BrowserFixture other_browser(&managed_mode_, &other_profile_); | 226 BrowserFixture other_browser(&managed_mode_, &other_profile_); |
221 | 227 |
222 // If the user cancelled entering managed mode, it should fail immediately. | 228 // If the user cancelled entering managed mode, it should fail immediately. |
223 managed_mode_.set_should_cancel_enter(true); | 229 managed_mode_.set_should_cancel_enter(true); |
224 managed_mode_.EnterManagedModeForTesting(&managed_mode_profile_, | 230 managed_mode_.EnterManagedModeForTesting(&managed_mode_profile_, |
225 CreateExpectedCallback(false)); | 231 CreateExpectedCallback(false)); |
226 } | 232 } |
OLD | NEW |