1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/loader/chrome_resource_dispatcher_host_delegate.h"
#include <memory>
#include "base/message_loop/message_loop.h"
#include "chrome/browser/loader/chrome_navigation_data.h"
#include "chrome/browser/loader/chrome_resource_dispatcher_host_delegate.h"
#include "components/data_reduction_proxy/core/browser/data_reduction_proxy_data.h"
#include "content/public/browser/navigation_data.h"
#include "content/public/test/test_browser_thread.h"
#include "net/base/request_priority.h"
#include "net/url_request/url_request.h"
#include "net/url_request/url_request_context.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"
class ChromeResourceDispatcherHostDelegateTest : public testing::Test {
public:
ChromeResourceDispatcherHostDelegateTest()
: ui_thread_(content::BrowserThread::UI, &message_loop_) {}
~ChromeResourceDispatcherHostDelegateTest() override {}
private:
base::MessageLoopForIO message_loop_;
content::TestBrowserThread ui_thread_;
};
TEST_F(ChromeResourceDispatcherHostDelegateTest,
GetNavigationDataWithDataReductionProxyData) {
std::unique_ptr<net::URLRequestContext> context(new net::URLRequestContext());
std::unique_ptr<net::URLRequest> fake_request(context->CreateRequest(
GURL("google.com"), net::RequestPriority::IDLE, nullptr));
// Add DataReductionProxyData to URLRequest
data_reduction_proxy::DataReductionProxyData* data_reduction_proxy_data =
data_reduction_proxy::DataReductionProxyData::GetDataAndCreateIfNecessary(
fake_request.get());
data_reduction_proxy_data->set_used_data_reduction_proxy(true);
std::unique_ptr<ChromeResourceDispatcherHostDelegate> delegate(
new ChromeResourceDispatcherHostDelegate());
ChromeNavigationData* chrome_navigation_data =
static_cast<ChromeNavigationData*>(
delegate->GetNavigationData(fake_request.get()));
data_reduction_proxy::DataReductionProxyData* data_reduction_proxy_data_copy =
chrome_navigation_data->GetDataReductionProxyData();
// The DataReductionProxyData should be a copy of the one on URLRequest
EXPECT_NE(data_reduction_proxy_data_copy, data_reduction_proxy_data);
// Make sure DataReductionProxyData was copied.
EXPECT_TRUE(data_reduction_proxy_data_copy->used_data_reduction_proxy());
EXPECT_EQ(
chrome_navigation_data,
ChromeNavigationData::GetDataAndCreateIfNecessary(fake_request.get()));
}
TEST_F(ChromeResourceDispatcherHostDelegateTest,
GetNavigationDataWithoutDataReductionProxyData) {
std::unique_ptr<net::URLRequestContext> context(new net::URLRequestContext());
std::unique_ptr<net::URLRequest> fake_request(context->CreateRequest(
GURL("google.com"), net::RequestPriority::IDLE, nullptr));
std::unique_ptr<ChromeResourceDispatcherHostDelegate> delegate(
new ChromeResourceDispatcherHostDelegate());
ChromeNavigationData* chrome_navigation_data =
static_cast<ChromeNavigationData*>(
delegate->GetNavigationData(fake_request.get()));
EXPECT_FALSE(chrome_navigation_data->GetDataReductionProxyData());
}
|
