Index: net/nqe/network_quality_estimator_unittest.cc |
diff --git a/net/nqe/network_quality_estimator_unittest.cc b/net/nqe/network_quality_estimator_unittest.cc |
index 183b87caf0e7ad18763c927b54aed1b6cb65107d..fd73b4b626d9c7100169b303962f3dedb1f74be3 100644 |
--- a/net/nqe/network_quality_estimator_unittest.cc |
+++ b/net/nqe/network_quality_estimator_unittest.cc |
@@ -40,6 +40,7 @@ |
#include "net/nqe/observation_buffer.h" |
#include "net/socket/socket_performance_watcher.h" |
#include "net/socket/socket_performance_watcher_factory.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" |
@@ -210,8 +211,9 @@ TEST(NetworkQualityEstimatorTest, TestKbpsRTTUpdates) { |
context.set_network_quality_estimator(&estimator); |
context.Init(); |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request->Start(); |
base::RunLoop().Run(); |
@@ -263,8 +265,9 @@ TEST(NetworkQualityEstimatorTest, TestKbpsRTTUpdates) { |
histogram_tester.ExpectBucketCount( |
"NQE.Kbps.ObservationSource", NETWORK_QUALITY_OBSERVATION_SOURCE_HTTP, 1); |
- std::unique_ptr<URLRequest> request2(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request2( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request2->SetLoadFlags(request2->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request2->Start(); |
base::RunLoop().Run(); |
@@ -324,8 +327,9 @@ TEST(NetworkQualityEstimatorTest, TestKbpsRTTUpdates) { |
EXPECT_FALSE( |
estimator.GetRecentDownlinkThroughputKbps(base::TimeTicks(), &kbps)); |
- std::unique_ptr<URLRequest> request3(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request3( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request3->SetLoadFlags(request2->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request3->Start(); |
base::RunLoop().Run(); |
@@ -380,8 +384,9 @@ TEST(NetworkQualityEstimatorTest, Caching) { |
// the beginning of the second request from the network traffic observed from |
// the first request. |
for (size_t i = 0; i < 2; ++i) { |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request->Start(); |
base::RunLoop().Run(); |
@@ -485,8 +490,9 @@ TEST(NetworkQualityEstimatorTest, CachingDisabled) { |
// the beginning of the second request from the network traffic observed from |
// the first request. |
for (size_t i = 0; i < 2; ++i) { |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request->Start(); |
base::RunLoop().Run(); |
@@ -554,8 +560,9 @@ TEST(NetworkQualityEstimatorTest, StoreObservations) { |
const size_t kMaxObservations = 10; |
for (size_t i = 0; i < kMaxObservations; ++i) { |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->Start(); |
base::RunLoop().Run(); |
EXPECT_TRUE(estimator.GetRecentHttpRTT(base::TimeTicks(), &rtt)); |
@@ -597,8 +604,9 @@ TEST(NetworkQualityEstimatorTest, ComputedPercentiles) { |
context.Init(); |
for (size_t i = 0; i < 10U; ++i) { |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->Start(); |
base::RunLoop().Run(); |
} |
@@ -1637,8 +1645,9 @@ TEST(NetworkQualityEstimatorTest, TestExternalEstimateProviderMergeEstimates) { |
// Start 2 requests to ensure that the RTT estimate computed by the network |
// quality estimator takes into account the RTT observations from the |
// external estimate provider as well as organic observations. |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->Start(); |
base::RunLoop().Run(); |
} |
@@ -1686,8 +1695,9 @@ TEST(NetworkQualityEstimatorTest, TestThroughputNoRequestOverlap) { |
context.set_network_quality_estimator(&estimator); |
context.Init(); |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request->Start(); |
base::RunLoop().Run(); |
@@ -1738,8 +1748,9 @@ TEST(NetworkQualityEstimatorTest, MAYBE_TestEffectiveConnectionTypeObserver) { |
tick_clock_ptr->Advance(base::TimeDelta::FromMinutes(60)); |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request->Start(); |
base::RunLoop().Run(); |
@@ -1769,8 +1780,9 @@ TEST(NetworkQualityEstimatorTest, MAYBE_TestEffectiveConnectionTypeObserver) { |
// Next request should not trigger recomputation of effective connection type |
// since there has been no change in the clock. |
- std::unique_ptr<URLRequest> request2(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request2( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request2->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request2->Start(); |
base::RunLoop().Run(); |
@@ -1841,8 +1853,9 @@ TEST(NetworkQualityEstimatorTest, TestRTTAndThroughputEstimatesObserver) { |
downstream_throughput_kbps); |
tick_clock_ptr->Advance(base::TimeDelta::FromMinutes(60)); |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->Start(); |
base::RunLoop().Run(); |
EXPECT_EQ(http_rtt, observer.http_rtt()); |
@@ -1853,8 +1866,9 @@ TEST(NetworkQualityEstimatorTest, TestRTTAndThroughputEstimatesObserver) { |
// The next request should not trigger recomputation of RTT or throughput |
// since there has been no change in the clock. |
- std::unique_ptr<URLRequest> request2(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request2( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request2->Start(); |
base::RunLoop().Run(); |
EXPECT_LE(1, observer.notifications_received() - notifications_received); |
@@ -1979,8 +1993,9 @@ TEST(NetworkQualityEstimatorTest, |
estimator.set_recent_effective_connection_type(EFFECTIVE_CONNECTION_TYPE_2G); |
tick_clock_ptr->Advance(base::TimeDelta::FromMinutes(60)); |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request->Start(); |
base::RunLoop().Run(); |
@@ -2063,14 +2078,16 @@ TEST(NetworkQualityEstimatorTest, TestRttThroughputObservers) { |
EXPECT_EQ(0U, throughput_observer.observations().size()); |
base::TimeTicks then = base::TimeTicks::Now(); |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request->Start(); |
base::RunLoop().Run(); |
- std::unique_ptr<URLRequest> request2(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request2( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request2->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request2->Start(); |
base::RunLoop().Run(); |
@@ -2181,8 +2198,9 @@ TEST(NetworkQualityEstimatorTest, MAYBE_TestTCPSocketRTT) { |
++before_count_tcp_rtt_observations; |
} |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request->Start(); |
base::RunLoop().Run(); |
@@ -2341,8 +2359,9 @@ TEST(NetworkQualityEstimatorTest, MAYBE_RecordAccuracy) { |
// Start a main-frame request which should cause network quality estimator |
// to record accuracy UMA. |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request->Start(); |
base::RunLoop().Run(); |
@@ -2609,8 +2628,9 @@ TEST(NetworkQualityEstimatorTest, CorrelationHistogram) { |
// Start a main-frame request that should cause network quality estimator to |
// record the network quality at the last main frame request. |
- std::unique_ptr<URLRequest> request_1(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request_1( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request_1->SetLoadFlags(request_1->load_flags() | |
LOAD_MAIN_FRAME_DEPRECATED); |
request_1->Start(); |
@@ -2667,8 +2687,9 @@ TEST(NetworkQualityEstimatorTest, CorrelationHistogram) { |
// Start another main-frame request which is redirected to an HTTPS URL. |
// Redirection should not cause any crashes. |
- std::unique_ptr<URLRequest> request_3(context.CreateRequest( |
- estimator.GetRedirectURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request_3( |
+ context.CreateRequest(estimator.GetRedirectURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request_3->Start(); |
base::RunLoop().Run(); |
EXPECT_FALSE(request_3->original_url().SchemeIsCryptographic()); |
@@ -2783,8 +2804,9 @@ TEST(NetworkQualityEstimatorTest, |
EXPECT_EQ(0U, observer.effective_connection_types().size()); |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request->Start(); |
base::RunLoop().Run(); |
@@ -2903,8 +2925,9 @@ TEST(NetworkQualityEstimatorTest, TypicalNetworkQualities) { |
// Force recomputation of effective connection type by starting a main |
// frame request. |
- std::unique_ptr<URLRequest> request(context.CreateRequest( |
- estimator.GetEchoURL(), DEFAULT_PRIORITY, &test_delegate)); |
+ std::unique_ptr<URLRequest> request( |
+ context.CreateRequest(estimator.GetEchoURL(), DEFAULT_PRIORITY, |
+ &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS)); |
request->SetLoadFlags(request->load_flags() | LOAD_MAIN_FRAME_DEPRECATED); |
request->Start(); |
base::RunLoop().Run(); |