Network traffic annotation added to URLLoaderImpl.

content/browser/loader/url_loader_factory_impl_unittest.cc

 // 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 "content/browser/loader/url_loader_factory_impl.h"
 
 #include <stdint.h>
 
 #include <memory>
 #include <string>
@@ skipping 33 matching lines @@
 #include "mojo/public/cpp/system/data_pipe.h"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
 #include "net/http/http_response_headers.h"
 #include "net/http/http_response_info.h"
 #include "net/http/http_status_code.h"
 #include "net/http/http_util.h"
 #include "net/test/url_request/url_request_failed_job.h"
 #include "net/test/url_request/url_request_mock_http_job.h"
 #include "net/test/url_request/url_request_slow_download_job.h"
+#include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
 #include "net/url_request/url_request_filter.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 #include "url/origin.h"
 
 namespace content {
 
 namespace {
 
 constexpr int kChildId = 99;
@@ skipping 94 matching lines @@
   TestURLLoaderClient client;
   // Assume the file contents is small enough to be stored in the data pipe.
   request.url = net::URLRequestMockHTTPJob::GetMockUrl("hello.html");
   request.method = "GET";
   // |resource_type| can't be a frame type. It is because when PlzNavigate is
   // enabled, the url scheme of frame type requests from the renderer process
   // must be blob scheme.
   request.resource_type = RESOURCE_TYPE_XHR;
   // Need to set |request_initiator| for non main frame type request.
   request.request_initiator = url::Origin();
-  factory_->CreateLoaderAndStart(mojo::MakeRequest(&loader), kRoutingId,
-                                 kRequestId, mojom::kURLLoadOptionNone, request,
-                                 client.CreateInterfacePtr());
+  factory_->CreateLoaderAndStart(
+      mojo::MakeRequest(&loader), kRoutingId, kRequestId,
+      mojom::kURLLoadOptionNone, request, client.CreateInterfacePtr(),
+      net::MutableNetworkTrafficAnnotationTag(TRAFFIC_ANNOTATION_FOR_TESTS));
 
   ASSERT_FALSE(client.has_received_response());
   ASSERT_FALSE(client.response_body().is_valid());
   ASSERT_FALSE(client.has_received_completion());
 
   client.RunUntilResponseReceived();
 
   net::URLRequest* url_request =
       rdh_.GetURLRequest(GlobalRequestID(kChildId, kRequestId));
   ASSERT_TRUE(url_request);
@@ skipping 53 matching lines @@
   net::URLRequestFailedJob::AddUrlHandler();
   request.url = net::URLRequestFailedJob::GetMockHttpUrlWithFailurePhase(
       net::URLRequestFailedJob::START, net::ERR_TIMED_OUT);
   request.method = "GET";
   // |resource_type| can't be a frame type. It is because when PlzNavigate is
   // enabled, the url scheme of frame type requests from the renderer process
   // must be blob scheme.
   request.resource_type = RESOURCE_TYPE_XHR;
   // Need to set |request_initiator| for non main frame type request.
   request.request_initiator = url::Origin();
-  factory_->CreateLoaderAndStart(mojo::MakeRequest(&loader), 2, 1,
-                                 mojom::kURLLoadOptionNone, request,
-                                 client.CreateInterfacePtr());
+  factory_->CreateLoaderAndStart(
+      mojo::MakeRequest(&loader), 2, 1, mojom::kURLLoadOptionNone, request,
+      client.CreateInterfacePtr(),
+      net::MutableNetworkTrafficAnnotationTag(TRAFFIC_ANNOTATION_FOR_TESTS));
 
   client.RunUntilComplete();
   ASSERT_FALSE(client.has_received_response());
   ASSERT_FALSE(client.response_body().is_valid());
 
   EXPECT_EQ(net::ERR_TIMED_OUT, client.completion_status().error_code);
   EXPECT_EQ(0, client.completion_status().encoded_data_length);
   EXPECT_EQ(0, client.completion_status().encoded_body_length);
 }
 
 // In this case, the loading fails after receiving a response.
 TEST_P(URLLoaderFactoryImplTest, GetFailedResponse2) {
   NavigationResourceThrottle::set_ui_checks_always_succeed_for_testing(true);
   mojom::URLLoaderAssociatedPtr loader;
   ResourceRequest request;
   TestURLLoaderClient client;
   net::URLRequestFailedJob::AddUrlHandler();
   request.url = net::URLRequestFailedJob::GetMockHttpUrlWithFailurePhase(
       net::URLRequestFailedJob::READ_ASYNC, net::ERR_TIMED_OUT);
   request.method = "GET";
   // |resource_type| can't be a frame type. It is because when PlzNavigate is
   // enabled, the url scheme of frame type requests from the renderer process
   // must be blob scheme.
   request.resource_type = RESOURCE_TYPE_XHR;
   // Need to set |request_initiator| for non main frame type request.
   request.request_initiator = url::Origin();
-  factory_->CreateLoaderAndStart(mojo::MakeRequest(&loader), 2, 1,
-                                 mojom::kURLLoadOptionNone, request,
-                                 client.CreateInterfacePtr());
+  factory_->CreateLoaderAndStart(
+      mojo::MakeRequest(&loader), 2, 1, mojom::kURLLoadOptionNone, request,
+      client.CreateInterfacePtr(),
+      net::MutableNetworkTrafficAnnotationTag(TRAFFIC_ANNOTATION_FOR_TESTS));
 
   client.RunUntilComplete();
   ASSERT_FALSE(client.has_received_response());
   ASSERT_FALSE(client.response_body().is_valid());
 
   EXPECT_EQ(net::ERR_TIMED_OUT, client.completion_status().error_code);
   EXPECT_GT(client.completion_status().encoded_data_length, 0);
   EXPECT_EQ(0, client.completion_status().encoded_body_length);
 }
 
 // This test tests a case where resource loading is cancelled before started.
 TEST_P(URLLoaderFactoryImplTest, InvalidURL) {
   mojom::URLLoaderAssociatedPtr loader;
   ResourceRequest request;
   TestURLLoaderClient client;
   request.url = GURL();
   request.method = "GET";
   // |resource_type| can't be a frame type. It is because when PlzNavigate is
   // enabled, the url scheme of frame type requests from the renderer process
   // must be blob scheme.
   request.resource_type = RESOURCE_TYPE_XHR;
   // Need to set |request_initiator| for non main frame type request.
   request.request_initiator = url::Origin();
   ASSERT_FALSE(request.url.is_valid());
-  factory_->CreateLoaderAndStart(mojo::MakeRequest(&loader), 2, 1,
-                                 mojom::kURLLoadOptionNone, request,
-                                 client.CreateInterfacePtr());
+  factory_->CreateLoaderAndStart(
+      mojo::MakeRequest(&loader), 2, 1, mojom::kURLLoadOptionNone, request,
+      client.CreateInterfacePtr(),
+      net::MutableNetworkTrafficAnnotationTag(TRAFFIC_ANNOTATION_FOR_TESTS));
 
   client.RunUntilComplete();
   ASSERT_FALSE(client.has_received_response());
   ASSERT_FALSE(client.response_body().is_valid());
 
   EXPECT_EQ(net::ERR_ABORTED, client.completion_status().error_code);
 }
 
 // This test tests a case where resource loading is cancelled before started.
 TEST_P(URLLoaderFactoryImplTest, ShouldNotRequestURL) {
   mojom::URLLoaderAssociatedPtr loader;
   RejectingResourceDispatcherHostDelegate rdh_delegate;
   rdh_.SetDelegate(&rdh_delegate);
   ResourceRequest request;
   TestURLLoaderClient client;
   request.url = GURL("http://localhost/");
   request.method = "GET";
   // |resource_type| can't be a frame type. It is because when PlzNavigate is
   // enabled, the url scheme of frame type requests from the renderer process
   // must be blob scheme.
   request.resource_type = RESOURCE_TYPE_XHR;
   // Need to set |request_initiator| for non main frame type request.
   request.request_initiator = url::Origin();
-  factory_->CreateLoaderAndStart(mojo::MakeRequest(&loader), 2, 1,
-                                 mojom::kURLLoadOptionNone, request,
-                                 client.CreateInterfacePtr());
+  factory_->CreateLoaderAndStart(
+      mojo::MakeRequest(&loader), 2, 1, mojom::kURLLoadOptionNone, request,
+      client.CreateInterfacePtr(),
+      net::MutableNetworkTrafficAnnotationTag(TRAFFIC_ANNOTATION_FOR_TESTS));
 
   client.RunUntilComplete();
   rdh_.SetDelegate(nullptr);
 
   ASSERT_FALSE(client.has_received_response());
   ASSERT_FALSE(client.response_body().is_valid());
 
   EXPECT_EQ(net::ERR_ABORTED, client.completion_status().error_code);
 }
 
 TEST_P(URLLoaderFactoryImplTest, DownloadToFile) {
   constexpr int32_t kRoutingId = 1;
   constexpr int32_t kRequestId = 2;
 
   mojom::URLLoaderAssociatedPtr loader;
   base::FilePath root;
   PathService::Get(DIR_TEST_DATA, &root);
   net::URLRequestMockHTTPJob::AddUrlHandlers(root,
                                              BrowserThread::GetBlockingPool());
 
   ResourceRequest request;
   TestURLLoaderClient client;
   request.url = net::URLRequestMockHTTPJob::GetMockUrl("hello.html");
   request.method = "GET";
   request.resource_type = RESOURCE_TYPE_XHR;
   request.download_to_file = true;
   request.request_initiator = url::Origin();
-  factory_->CreateLoaderAndStart(mojo::MakeRequest(&loader), kRoutingId,
-                                 kRequestId, 0, request,
-                                 client.CreateInterfacePtr());
+  factory_->CreateLoaderAndStart(
+      mojo::MakeRequest(&loader), kRoutingId, kRequestId, 0, request,
+      client.CreateInterfacePtr(),
+      net::MutableNetworkTrafficAnnotationTag(TRAFFIC_ANNOTATION_FOR_TESTS));
   ASSERT_FALSE(client.has_received_response());
   ASSERT_FALSE(client.has_data_downloaded());
   ASSERT_FALSE(client.has_received_completion());
 
   client.RunUntilResponseReceived();
 
   net::URLRequest* url_request =
       rdh_.GetURLRequest(GlobalRequestID(kChildId, kRequestId));
   ASSERT_TRUE(url_request);
   ResourceRequestInfoImpl* request_info =
@@ skipping 44 matching lines @@
   PathService::Get(DIR_TEST_DATA, &root);
   net::URLRequestSlowDownloadJob::AddUrlHandler();
 
   ResourceRequest request;
   TestURLLoaderClient client;
   request.url = GURL(net::URLRequestSlowDownloadJob::kKnownSizeUrl);
   request.method = "GET";
   request.resource_type = RESOURCE_TYPE_XHR;
   request.download_to_file = true;
   request.request_initiator = url::Origin();
-  factory_->CreateLoaderAndStart(mojo::MakeRequest(&loader), kRoutingId,
-                                 kRequestId, 0, request,
-                                 client.CreateInterfacePtr());
+  factory_->CreateLoaderAndStart(
+      mojo::MakeRequest(&loader), kRoutingId, kRequestId, 0, request,
+      client.CreateInterfacePtr(),
+      net::MutableNetworkTrafficAnnotationTag(TRAFFIC_ANNOTATION_FOR_TESTS));
   ASSERT_FALSE(client.has_received_response());
   ASSERT_FALSE(client.has_data_downloaded());
   ASSERT_FALSE(client.has_received_completion());
 
   client.RunUntilResponseReceived();
 
   net::URLRequest* url_request =
       rdh_.GetURLRequest(GlobalRequestID(kChildId, kRequestId));
   ASSERT_TRUE(url_request);
   ResourceRequestInfoImpl* request_info =
@@ skipping 40 matching lines @@
   // Assume the file contents is small enough to be stored in the data pipe.
   request.url = net::URLRequestMockHTTPJob::GetMockUrl("gzip-content.svgz");
   request.method = "GET";
   // |resource_type| can't be a frame type. It is because when PlzNavigate is
   // enabled, the url scheme of frame type requests from the renderer process
   // must be blob scheme.
   request.resource_type = RESOURCE_TYPE_XHR;
   // Need to set |request_initiator| for non main frame type request.
   request.request_initiator = url::Origin();
   request.report_raw_headers = true;
-  factory_->CreateLoaderAndStart(mojo::MakeRequest(&loader), kRoutingId,
-                                 kRequestId, mojom::kURLLoadOptionNone, request,
-                                 client.CreateInterfacePtr());
+  factory_->CreateLoaderAndStart(
+      mojo::MakeRequest(&loader), kRoutingId, kRequestId,
+      mojom::kURLLoadOptionNone, request, client.CreateInterfacePtr(),
+      net::MutableNetworkTrafficAnnotationTag(TRAFFIC_ANNOTATION_FOR_TESTS));
 
   client.RunUntilComplete();
 
   std::string contents;
   while (true) {
     char buffer[16];
     uint32_t read_size = sizeof(buffer);
     MojoResult r = mojo::ReadDataRaw(client.response_body(), buffer, &read_size,
                                      MOJO_READ_DATA_FLAG_NONE);
     if (r == MOJO_RESULT_FAILED_PRECONDITION)
@@ skipping 37 matching lines @@
   // Assume the file contents is small enough to be stored in the data pipe.
   request.url = net::URLRequestFailedJob::GetMockHttpUrl(net::ERR_IO_PENDING);
   request.method = "GET";
   request.is_main_frame = true;
   // |resource_type| can't be a frame type. It is because when PlzNavigate is
   // enabled, the url scheme of frame type requests from the renderer process
   // must be blob scheme.
   request.resource_type = RESOURCE_TYPE_XHR;
   // Need to set |request_initiator| for non main frame type request.
   request.request_initiator = url::Origin();
-  factory_->CreateLoaderAndStart(mojo::MakeRequest(&loader), kRoutingId,
-                                 kRequestId, mojom::kURLLoadOptionNone, request,
-                                 client.CreateInterfacePtr());
+  factory_->CreateLoaderAndStart(
+      mojo::MakeRequest(&loader), kRoutingId, kRequestId,
+      mojom::kURLLoadOptionNone, request, client.CreateInterfacePtr(),
+      net::MutableNetworkTrafficAnnotationTag(TRAFFIC_ANNOTATION_FOR_TESTS));
 
   base::RunLoop().RunUntilIdle();
   ASSERT_TRUE(rdh_.GetURLRequest(GlobalRequestID(kChildId, kRequestId)));
   ASSERT_FALSE(client.has_received_response());
   ASSERT_FALSE(client.response_body().is_valid());
   ASSERT_FALSE(client.has_received_completion());
 
   loader = nullptr;
   base::RunLoop().RunUntilIdle();
   ASSERT_FALSE(rdh_.GetURLRequest(GlobalRequestID(kChildId, kRequestId)));
 }
 
 INSTANTIATE_TEST_CASE_P(URLLoaderFactoryImplTest,
                         URLLoaderFactoryImplTest,
                         ::testing::Values(128, 32 * 1024));
 
 }  // namespace
 
 }  // namespace content