| 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 |