| OLD | NEW |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 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/extensions/extension_with_management_policy_apitest.h" | 5 #include "chrome/browser/extensions/extension_with_management_policy_apitest.h" |
| 6 #include "components/policy/core/browser/browser_policy_connector.h" | 6 #include "components/policy/core/browser/browser_policy_connector.h" |
| 7 #include "net/dns/mock_host_resolver.h" |
| 7 #include "net/test/embedded_test_server/http_request.h" | 8 #include "net/test/embedded_test_server/http_request.h" |
| 8 | 9 |
| 9 ExtensionApiTestWithManagementPolicy::ExtensionApiTestWithManagementPolicy() | 10 ExtensionApiTestWithManagementPolicy::ExtensionApiTestWithManagementPolicy() |
| 10 : ExtensionApiTest() {} | 11 : ExtensionApiTest() {} |
| 11 | 12 |
| 12 ExtensionApiTestWithManagementPolicy::~ExtensionApiTestWithManagementPolicy() {} | 13 ExtensionApiTestWithManagementPolicy::~ExtensionApiTestWithManagementPolicy() {} |
| 13 | 14 |
| 14 void ExtensionApiTestWithManagementPolicy::SetUpInProcessBrowserTestFixture() { | 15 void ExtensionApiTestWithManagementPolicy::SetUpInProcessBrowserTestFixture() { |
| 15 ExtensionApiTest::SetUpInProcessBrowserTestFixture(); | 16 ExtensionApiTest::SetUpInProcessBrowserTestFixture(); |
| 16 EXPECT_CALL(policy_provider_, IsInitializationComplete(testing::_)) | 17 EXPECT_CALL(policy_provider_, IsInitializationComplete(testing::_)) |
| 17 .WillRepeatedly(testing::Return(true)); | 18 .WillRepeatedly(testing::Return(true)); |
| 18 policy_provider_.SetAutoRefresh(); | 19 policy_provider_.SetAutoRefresh(); |
| 19 policy::BrowserPolicyConnector::SetPolicyProviderForTesting( | 20 policy::BrowserPolicyConnector::SetPolicyProviderForTesting( |
| 20 &policy_provider_); | 21 &policy_provider_); |
| 21 } | 22 } |
| 23 |
| 24 void ExtensionApiTestWithManagementPolicy::SetUpOnMainThread() { |
| 25 ExtensionApiTest::SetUpOnMainThread(); |
| 26 host_resolver()->AddRule("*", "127.0.0.1"); |
| 27 } |
| OLD | NEW |