OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 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 | 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/net/spdyproxy/chrome_data_use_group_provider.h" | 5 #include "chrome/browser/net/spdyproxy/chrome_data_use_group_provider.h" |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 | 8 |
9 #include "base/memory/ref_counted.h" | 9 #include "base/memory/ref_counted.h" |
10 #include "chrome/browser/net/spdyproxy/chrome_data_use_group.h" | 10 #include "chrome/browser/net/spdyproxy/chrome_data_use_group.h" |
11 #include "content/public/browser/resource_request_info.h" | 11 #include "content/public/browser/resource_request_info.h" |
12 #include "content/public/common/previews_state.h" | 12 #include "content/public/common/previews_state.h" |
13 #include "content/public/test/test_browser_thread_bundle.h" | 13 #include "content/public/test/test_browser_thread_bundle.h" |
| 14 #include "net/traffic_annotation/network_traffic_annotation_test_helper.h" |
14 #include "net/url_request/url_request.h" | 15 #include "net/url_request/url_request.h" |
15 #include "net/url_request/url_request_test_util.h" | 16 #include "net/url_request/url_request_test_util.h" |
16 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
17 #include "url/gurl.h" | 18 #include "url/gurl.h" |
18 | 19 |
19 class ChromeDataUseGroupProviderTest : public testing::Test { | 20 class ChromeDataUseGroupProviderTest : public testing::Test { |
20 protected: | 21 protected: |
21 ChromeDataUseGroupProvider* data_use_group_provider() { | 22 ChromeDataUseGroupProvider* data_use_group_provider() { |
22 return &data_use_group_provider_; | 23 return &data_use_group_provider_; |
23 } | 24 } |
24 | 25 |
25 std::unique_ptr<net::URLRequest> CreateRequestForFrame(int render_process_id, | 26 std::unique_ptr<net::URLRequest> CreateRequestForFrame(int render_process_id, |
26 int render_frame_id) { | 27 int render_frame_id) { |
27 std::unique_ptr<net::URLRequest> request = context_.CreateRequest( | 28 std::unique_ptr<net::URLRequest> request = |
28 GURL("http://foo.com/"), net::IDLE, &test_delegate_); | 29 context_.CreateRequest(GURL("http://foo.com/"), net::IDLE, |
| 30 &test_delegate_, TRAFFIC_ANNOTATION_FOR_TESTS); |
29 | 31 |
30 content::ResourceRequestInfo::AllocateForTesting( | 32 content::ResourceRequestInfo::AllocateForTesting( |
31 request.get(), content::RESOURCE_TYPE_MAIN_FRAME, | 33 request.get(), content::RESOURCE_TYPE_MAIN_FRAME, |
32 /*ResourceContext=*/nullptr, render_process_id, /*render_view_id=*/-1, | 34 /*ResourceContext=*/nullptr, render_process_id, /*render_view_id=*/-1, |
33 render_frame_id, /*is_main_frame=*/true, | 35 render_frame_id, /*is_main_frame=*/true, |
34 /*parent_is_main_frame=*/false, | 36 /*parent_is_main_frame=*/false, |
35 /*allow_download=*/true, | 37 /*allow_download=*/true, |
36 /*is_async=*/true, content::PREVIEWS_OFF); | 38 /*is_async=*/true, content::PREVIEWS_OFF); |
37 | 39 |
38 return request; | 40 return request; |
(...skipping 25 matching lines...) Expand all Loading... |
64 std::unique_ptr<net::URLRequest> request = CreateRequestForFrame(1, 1); | 66 std::unique_ptr<net::URLRequest> request = CreateRequestForFrame(1, 1); |
65 scoped_refptr<data_reduction_proxy::DataUseGroup> group1 = | 67 scoped_refptr<data_reduction_proxy::DataUseGroup> group1 = |
66 data_use_group_provider()->GetDataUseGroup(request.get()); | 68 data_use_group_provider()->GetDataUseGroup(request.get()); |
67 | 69 |
68 request = CreateRequestForFrame(2, 2); | 70 request = CreateRequestForFrame(2, 2); |
69 scoped_refptr<data_reduction_proxy::DataUseGroup> group2 = | 71 scoped_refptr<data_reduction_proxy::DataUseGroup> group2 = |
70 data_use_group_provider()->GetDataUseGroup(request.get()); | 72 data_use_group_provider()->GetDataUseGroup(request.get()); |
71 | 73 |
72 EXPECT_NE(group1, group2); | 74 EXPECT_NE(group1, group2); |
73 } | 75 } |
OLD | NEW |