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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
|
// 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 "components/data_use_measurement/core/data_use_network_delegate.h"
#include <memory>
#include "base/message_loop/message_loop.h"
#include "base/run_loop.h"
#include "base/test/histogram_tester.h"
#include "components/data_use_measurement/core/data_use_ascriber.h"
#include "components/data_use_measurement/core/url_request_classifier.h"
#include "components/metrics/data_use_tracker.h"
#include "net/socket/socket_test_util.h"
#include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
#include "net/url_request/url_request.h"
#include "net/url_request/url_request_test_util.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"
namespace data_use_measurement {
namespace {
class TestURLRequestClassifier : public base::SupportsUserData::Data,
public URLRequestClassifier {
public:
static const void* const kUserDataKey;
DataUseUserData::DataUseContentType GetContentType(
const net::URLRequest& request,
const net::HttpResponseHeaders& response_headers) const override {
return DataUseUserData::OTHER;
}
void RecordPageTransitionUMA(uint64_t page_transition,
int64_t received_bytes) const override {}
bool IsFavIconRequest(const net::URLRequest& request) const override {
return false;
}
};
class TestDataUseAscriber : public DataUseAscriber {
public:
TestDataUseAscriber() {}
DataUseRecorder* GetOrCreateDataUseRecorder(
net::URLRequest* request) override {
return nullptr;
}
DataUseRecorder* GetDataUseRecorder(const net::URLRequest& request) override {
return nullptr;
}
std::unique_ptr<URLRequestClassifier> CreateURLRequestClassifier()
const override {
return nullptr;
}
};
// static
const void* const TestURLRequestClassifier::kUserDataKey =
&TestURLRequestClassifier::kUserDataKey;
// This function requests a URL, and makes it return a known response. If
// |from_user| is true, it attaches a ResourceRequestInfo to the URLRequest,
// because requests from users have this info. If |from_user| is false, the
// request is presumed to be from a service, and the service name is set in the
// request's user data. (As an example suggestions service tag is attached). if
// |redirect| is true, it adds necessary socket data to have it follow redirect
// before getting the final response.
std::unique_ptr<net::URLRequest> RequestURL(
net::URLRequestContext* context,
net::MockClientSocketFactory* socket_factory,
bool from_user,
bool redirect) {
net::MockRead redirect_mock_reads[] = {
net::MockRead("HTTP/1.1 302 Found\r\n"
"Location: http://bar.com/\r\n\r\n"),
net::MockRead(net::SYNCHRONOUS, net::OK),
};
net::StaticSocketDataProvider redirect_socket_data_provider(
redirect_mock_reads, base::span<net::MockWrite>());
if (redirect)
socket_factory->AddSocketDataProvider(&redirect_socket_data_provider);
net::MockRead response_mock_reads[] = {
net::MockRead("HTTP/1.1 200 OK\r\n\r\n"), net::MockRead("response body"),
net::MockRead(net::SYNCHRONOUS, net::OK),
};
const auto traffic_annotation =
from_user ? net::DefineNetworkTrafficAnnotation("blink_resource_loader",
"blink resource loaded "
"will be treated as "
"user-initiated request")
: TRAFFIC_ANNOTATION_FOR_TESTS;
net::StaticSocketDataProvider response_socket_data_provider(
response_mock_reads, base::span<net::MockWrite>());
socket_factory->AddSocketDataProvider(&response_socket_data_provider);
net::TestDelegate test_delegate;
test_delegate.set_quit_on_complete(true);
std::unique_ptr<net::URLRequest> request(
context->CreateRequest(GURL("http://example.com"), net::DEFAULT_PRIORITY,
&test_delegate, traffic_annotation));
if (!from_user) {
request->SetUserData(
data_use_measurement::DataUseUserData::kUserDataKey,
std::make_unique<data_use_measurement::DataUseUserData>(
data_use_measurement::DataUseUserData::SUGGESTIONS,
data_use_measurement::DataUseUserData::FOREGROUND));
}
request->Start();
base::RunLoop().RunUntilIdle();
return request;
}
class DataUseNetworkDelegateTest : public testing::Test {
public:
DataUseNetworkDelegateTest()
: context_(true),
data_use_network_delegate_(std::make_unique<net::TestNetworkDelegate>(),
&test_data_use_ascriber_,
std::make_unique<TestURLRequestClassifier>(),
metrics::UpdateUsagePrefCallbackType()) {
context_.set_client_socket_factory(&mock_socket_factory_);
context_.set_network_delegate(&data_use_network_delegate_);
context_.Init();
}
net::TestURLRequestContext* context() { return &context_; }
net::MockClientSocketFactory* socket_factory() {
return &mock_socket_factory_;
}
private:
base::MessageLoopForIO message_loop_;
net::MockClientSocketFactory mock_socket_factory_;
net::TestURLRequestContext context_;
TestDataUseAscriber test_data_use_ascriber_;
DataUseNetworkDelegate data_use_network_delegate_;
};
// This function tests data use measurement for requests by services. it makes a
// query which is similar to a query of a service, so it should affect
// DataUse.TrafficSize.System.Dimensions histogram. AppState and ConnectionType
// dimensions are always Foreground and NotCellular respectively.
TEST_F(DataUseNetworkDelegateTest, DataUseMeasurementServiceTest) {
base::HistogramTester histogram_tester;
// A query from a service without redirection.
RequestURL(context(), socket_factory(), false, false);
EXPECT_FALSE(
histogram_tester
.GetTotalCountsForPrefix(
"DataUse.TrafficSize.System.Downstream.Foreground.NotCellular")
.empty());
histogram_tester.ExpectTotalCount(
"DataUse.TrafficSize.System.Upstream.Foreground.NotCellular", 1);
histogram_tester.ExpectTotalCount(
"DataUse.TrafficSize.User.Downstream.Foreground.NotCellular", 0);
histogram_tester.ExpectTotalCount(
"DataUse.TrafficSize.User.Upstream.Foreground.NotCellular", 0);
}
// This function tests data use measurement for requests by user.The query from
// a user should affect DataUse.TrafficSize.User.Dimensions histogram. AppState
// and ConnectionType dimensions are always Foreground and NotCellular
// respectively.
TEST_F(DataUseNetworkDelegateTest, DataUseMeasurementUserTest) {
base::HistogramTester histogram_tester;
// A query from user without redirection.
RequestURL(context(), socket_factory(), true, false);
EXPECT_FALSE(
histogram_tester
.GetTotalCountsForPrefix(
"DataUse.TrafficSize.User.Downstream.Foreground.NotCellular")
.empty());
histogram_tester.ExpectTotalCount(
"DataUse.TrafficSize.User.Upstream.Foreground.NotCellular", 1);
histogram_tester.ExpectTotalCount(
"DataUse.TrafficSize.System.Downstream.Foreground.NotCellular", 0);
histogram_tester.ExpectTotalCount(
"DataUse.TrafficSize.System.Upstream.Foreground.NotCellular", 0);
}
// This function tests data use measurement for requests by services in case the
// request is redirected once. it makes a query which is similar to a query of a
// service, so it should affect DataUse.TrafficSize.System.Dimensions and
// histogram. AppState and ConnectionType dimensions are always Foreground and
// NotCellular respectively.
TEST_F(DataUseNetworkDelegateTest, DataUseMeasurementServiceTestWithRedirect) {
base::HistogramTester histogram_tester;
// A query from user with one redirection.
RequestURL(context(), socket_factory(), false, true);
EXPECT_FALSE(
histogram_tester
.GetTotalCountsForPrefix(
"DataUse.TrafficSize.System.Downstream.Foreground.NotCellular")
.empty());
histogram_tester.ExpectTotalCount(
"DataUse.TrafficSize.System.Upstream.Foreground.NotCellular", 2);
// Two uploads and two downloads message, so totalCount should be 4.
histogram_tester.ExpectTotalCount(
"DataUse.TrafficSize.User.Downstream.Foreground.NotCellular", 0);
histogram_tester.ExpectTotalCount(
"DataUse.TrafficSize.User.Upstream.Foreground.NotCellular", 0);
}
// This function tests data use measurement for requests by user in case the
// request is redirected once.The query from a user should affect
// DataUse.TrafficSize.User.Dimensions histogram. AppState and ConnectionType
// dimensions are always Foreground and NotCellular respectively.
TEST_F(DataUseNetworkDelegateTest, DataUseMeasurementUserTestWithRedirect) {
base::HistogramTester histogram_tester;
// A query from user with one redirection.
RequestURL(context(), socket_factory(), true, true);
EXPECT_FALSE(
histogram_tester
.GetTotalCountsForPrefix(
"DataUse.TrafficSize.User.Downstream.Foreground.NotCellular")
.empty());
histogram_tester.ExpectTotalCount(
"DataUse.TrafficSize.User.Upstream.Foreground.NotCellular", 2);
histogram_tester.ExpectTotalCount(
"DataUse.TrafficSize.System.Downstream.Foreground.NotCellular", 0);
histogram_tester.ExpectTotalCount(
"DataUse.TrafficSize.System.Upstream.Foreground.NotCellular", 0);
}
} // namespace
} // namespace data_use_measurement
|
