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 |