| Index: net/url_request/url_request_http_job_unittest.cc
|
| diff --git a/net/url_request/url_request_http_job_unittest.cc b/net/url_request/url_request_http_job_unittest.cc
|
| index c0665d86df9e195e96e15e7cef3d685dacd955ab..cbb6aa87d3b4f2b6189f0c8014c8c84b5a546cdb 100644
|
| --- a/net/url_request/url_request_http_job_unittest.cc
|
| +++ b/net/url_request/url_request_http_job_unittest.cc
|
| @@ -31,6 +31,7 @@
|
| #include "net/test/cert_test_util.h"
|
| #include "net/test/gtest_util.h"
|
| #include "net/test/test_data_directory.h"
|
| +#include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
|
| #include "net/url_request/url_request.h"
|
| #include "net/url_request/url_request_job_factory_impl.h"
|
| #include "net/url_request/url_request_status.h"
|
| @@ -130,8 +131,9 @@ TEST_F(URLRequestHttpJobSetUpSourceTest, SetUpSourceFails) {
|
| arraysize(writes));
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| - std::unique_ptr<URLRequest> request = context_.CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate_);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_.CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate_, TRAFFIC_ANNOTATION_FOR_TESTS);
|
| std::unique_ptr<TestURLRequestHttpJob> job(
|
| new TestURLRequestHttpJob(request.get()));
|
| job->set_use_null_source_stream(true);
|
| @@ -155,8 +157,9 @@ TEST_F(URLRequestHttpJobSetUpSourceTest, UnknownEncoding) {
|
| arraysize(writes));
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| - std::unique_ptr<URLRequest> request = context_.CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate_);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_.CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate_, TRAFFIC_ANNOTATION_FOR_TESTS);
|
| std::unique_ptr<TestURLRequestHttpJob> job(
|
| new TestURLRequestHttpJob(request.get()));
|
| test_job_interceptor_->set_main_intercept_job(std::move(job));
|
| @@ -182,8 +185,9 @@ TEST_F(URLRequestHttpJobSetUpSourceTest, SdchNotAdvertisedGotSdchResponse) {
|
| // This test expects TestURLRequestContexts to have no SdchManager.
|
| DCHECK(!context_.sdch_manager());
|
|
|
| - std::unique_ptr<URLRequest> request = context_.CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate_);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_.CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate_, TRAFFIC_ANNOTATION_FOR_TESTS);
|
| std::unique_ptr<TestURLRequestHttpJob> job(
|
| new TestURLRequestHttpJob(request.get()));
|
| test_job_interceptor_->set_main_intercept_job(std::move(job));
|
| @@ -206,8 +210,9 @@ class URLRequestHttpJobTest : public ::testing::Test {
|
| context_.set_net_log(&net_log_);
|
| context_.Init();
|
|
|
| - req_ = context_.CreateRequest(GURL("http://www.example.com"),
|
| - DEFAULT_PRIORITY, &delegate_);
|
| + req_ =
|
| + context_.CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate_, TRAFFIC_ANNOTATION_FOR_TESTS);
|
| }
|
|
|
| bool TransactionAcceptsSdchEncoding() {
|
| @@ -279,8 +284,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest,
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| request->Start();
|
| ASSERT_TRUE(request->is_pending());
|
| @@ -308,8 +314,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest,
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| request->Start();
|
| ASSERT_TRUE(request->is_pending());
|
| @@ -339,8 +346,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest, TestContentLengthFailedRequest) {
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| request->Start();
|
| ASSERT_TRUE(request->is_pending());
|
| @@ -371,8 +379,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest,
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| delegate.set_cancel_in_received_data(true);
|
| request->Start();
|
| @@ -408,8 +417,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest,
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| request->Start();
|
| ASSERT_TRUE(request->is_pending());
|
| @@ -443,8 +453,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest,
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| request->Start();
|
| ASSERT_TRUE(request->is_pending());
|
| @@ -471,8 +482,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest,
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| delegate.set_cancel_in_response_started(true);
|
| request->Start();
|
| @@ -501,8 +513,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest,
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| request->Start();
|
| base::RunLoop().Run();
|
| @@ -544,8 +557,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest,
|
| socket_factory_.AddSocketDataProvider(&final_socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.redirect.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.redirect.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| request->Start();
|
| ASSERT_TRUE(request->is_pending());
|
| @@ -576,8 +590,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest,
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| delegate.set_cancel_in_response_started(true);
|
| request->Start();
|
| @@ -601,8 +616,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest,
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| request->Start();
|
| request->Cancel();
|
| @@ -627,8 +643,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest, TestHttpTimeToFirstByte) {
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
| histograms.ExpectTotalCount("Net.HttpTimeToFirstByte", 0);
|
|
|
| request->Start();
|
| @@ -651,8 +668,9 @@ TEST_F(URLRequestHttpJobWithMockSocketsTest,
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| request->Start();
|
| request->Cancel();
|
| @@ -669,8 +687,9 @@ TEST_F(URLRequestHttpJobTest, TestCancelWhileReadingCookies) {
|
| context.Init();
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context.CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context.CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
|
|
| request->Start();
|
| request->Cancel();
|
| @@ -773,8 +792,8 @@ TEST_F(URLRequestHttpJobTest, HSTSInternalRedirectTest) {
|
| SCOPED_TRACE(test.url);
|
| TestDelegate d;
|
| TestNetworkDelegate network_delegate;
|
| - std::unique_ptr<URLRequest> r(
|
| - context_.CreateRequest(GURL(test.url), DEFAULT_PRIORITY, &d));
|
| + std::unique_ptr<URLRequest> r(context_.CreateRequest(
|
| + GURL(test.url), DEFAULT_PRIORITY, &d, TRAFFIC_ANNOTATION_FOR_TESTS));
|
|
|
| net_log_.Clear();
|
| r->Start();
|
| @@ -853,8 +872,9 @@ TEST_F(URLRequestHttpJobWithSdchSupportTest,
|
| sdch_manager.AddObserver(&sdch_observer);
|
| context_.set_sdch_manager(&sdch_manager);
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_.CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_.CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
| request->Start();
|
|
|
| base::RunLoop().Run();
|
| @@ -896,8 +916,8 @@ TEST_F(URLRequestHttpJobWithSdchSupportTest, GetDictionary) {
|
| EXPECT_CALL(sdch_observer,
|
| OnGetDictionary(url, GURL("http://example.com/sdch.dict")));
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request =
|
| - context_.CreateRequest(url, DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request = context_.CreateRequest(
|
| + url, DEFAULT_PRIORITY, &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
| request->Start();
|
| base::RunLoop().RunUntilIdle();
|
|
|
| @@ -905,8 +925,8 @@ TEST_F(URLRequestHttpJobWithSdchSupportTest, GetDictionary) {
|
|
|
| // Second response should be from cache without notification of SdchObserver
|
| TestDelegate delegate2;
|
| - std::unique_ptr<URLRequest> request2 =
|
| - context_.CreateRequest(url, DEFAULT_PRIORITY, &delegate2);
|
| + std::unique_ptr<URLRequest> request2 = context_.CreateRequest(
|
| + url, DEFAULT_PRIORITY, &delegate2, TRAFFIC_ANNOTATION_FOR_TESTS);
|
| request2->Start();
|
| base::RunLoop().RunUntilIdle();
|
|
|
| @@ -942,8 +962,9 @@ TEST_F(URLRequestHttpJobWithBrotliSupportTest, NoBrotliAdvertisementOverHttp) {
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("http://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("http://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
| request->Start();
|
| base::RunLoop().RunUntilIdle();
|
|
|
| @@ -977,8 +998,9 @@ TEST_F(URLRequestHttpJobWithBrotliSupportTest, BrotliAdvertisement) {
|
| socket_factory_.AddSocketDataProvider(&socket_data);
|
|
|
| TestDelegate delegate;
|
| - std::unique_ptr<URLRequest> request = context_->CreateRequest(
|
| - GURL("https://www.example.com"), DEFAULT_PRIORITY, &delegate);
|
| + std::unique_ptr<URLRequest> request =
|
| + context_->CreateRequest(GURL("https://www.example.com"), DEFAULT_PRIORITY,
|
| + &delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
|
| request->Start();
|
| base::RunLoop().RunUntilIdle();
|
|
|
| @@ -1012,7 +1034,8 @@ TEST_F(URLRequestHttpJobTest, AndroidCleartextPermittedTest) {
|
|
|
| TestDelegate delegate;
|
| std::unique_ptr<URLRequest> request =
|
| - context_.CreateRequest(GURL(test.url), DEFAULT_PRIORITY, &delegate);
|
| + context_.CreateRequest(GURL(test.url), DEFAULT_PRIORITY, &delegate,
|
| + TRAFFIC_ANNOTATION_FOR_TESTS);
|
| request->Start();
|
| base::RunLoop().Run();
|
|
|
| @@ -1059,8 +1082,8 @@ class URLRequestHttpJobWebSocketTest
|
| URLRequestHttpJobWebSocketTest()
|
| : req_(context_.CreateRequest(GURL("ws://www.example.com"),
|
| DEFAULT_PRIORITY,
|
| - &delegate_)) {
|
| - }
|
| + &delegate_,
|
| + TRAFFIC_ANNOTATION_FOR_TESTS)) {}
|
|
|
| TestDelegate delegate_;
|
| std::unique_ptr<URLRequest> req_;
|
|
|