OLD | NEW |
(Empty) | |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "content/public/test/browser_test_utils.h" |
| 6 #include "content/public/test/content_browser_test.h" |
| 7 #include "content/public/test/content_browser_test_utils.h" |
| 8 #include "content/shell/browser/shell.h" |
| 9 #include "net/dns/host_resolver_proc.h" |
| 10 #include "net/dns/mock_host_resolver.h" |
| 11 #include "testing/gmock/include/gmock/gmock.h" |
| 12 #include "testing/gtest/include/gtest/gtest.h" |
| 13 |
| 14 namespace content { |
| 15 |
| 16 using ::testing::_; |
| 17 |
| 18 class MockHostResolverProc : public net::HostResolverProc { |
| 19 public: |
| 20 MockHostResolverProc() : net::HostResolverProc(nullptr) {} |
| 21 |
| 22 MOCK_METHOD5(Resolve, |
| 23 int(const std::string& host, |
| 24 net::AddressFamily address_family, |
| 25 net::HostResolverFlags flags, |
| 26 net::AddressList* addresses, |
| 27 int* os_error)); |
| 28 |
| 29 private: |
| 30 ~MockHostResolverProc() override {} |
| 31 }; |
| 32 |
| 33 class ResourceHintsBrowserTest : public ContentBrowserTest { |
| 34 public: |
| 35 ResourceHintsBrowserTest() |
| 36 : mock_host_resolver_proc_(new MockHostResolverProc), |
| 37 scoped_host_resolver_proc_(nullptr) {} |
| 38 |
| 39 protected: |
| 40 scoped_refptr<MockHostResolverProc> mock_host_resolver_proc_; |
| 41 std::unique_ptr<net::ScopedDefaultHostResolverProc> |
| 42 scoped_host_resolver_proc_; |
| 43 }; |
| 44 |
| 45 IN_PROC_BROWSER_TEST_F(ResourceHintsBrowserTest, DnsPrefetch) { |
| 46 scoped_host_resolver_proc_.reset( |
| 47 new net::ScopedDefaultHostResolverProc(mock_host_resolver_proc_.get())); |
| 48 ASSERT_TRUE(embedded_test_server()->Start()); |
| 49 |
| 50 EXPECT_CALL(*mock_host_resolver_proc_, Resolve("chromium.org", _, _, _, _)) |
| 51 .Times(1); |
| 52 EXPECT_CALL(*mock_host_resolver_proc_, |
| 53 Resolve(testing::Ne("chromium.org"), _, _, _, _)) |
| 54 .Times(0); |
| 55 EXPECT_TRUE(NavigateToURL(shell(), embedded_test_server()->GetURL( |
| 56 "/resource_hints/dns_prefetch.html"))); |
| 57 EXPECT_TRUE( |
| 58 NavigateToURL(shell(), embedded_test_server()->GetURL("/title1.html"))); |
| 59 scoped_host_resolver_proc_.reset(); |
| 60 } |
| 61 |
| 62 } // namespace content |
OLD | NEW |