Network traffic annotation added to chrome/browser/ unittests.

chrome/browser/extensions/extension_protocols_unittest.cc

 // Copyright 2014 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 <stddef.h>
 
 #include <memory>
 #include <string>
 
 #include "base/command_line.h"
@@ skipping 16 matching lines @@
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "content/public/test/test_utils.h"
 #include "extensions/browser/content_verifier.h"
 #include "extensions/browser/extension_protocols.h"
 #include "extensions/browser/info_map.h"
 #include "extensions/common/constants.h"
 #include "extensions/common/extension.h"
 #include "extensions/common/extension_builder.h"
 #include "extensions/common/file_util.h"
 #include "net/base/request_priority.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"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using content::ResourceType;
 
 namespace extensions {
 namespace {
@@ skipping 192 matching lines @@
   int DoRequest(const Extension& extension, const std::string& relative_path) {
     if (!extension_info_map_->extensions().Contains(extension.id())) {
       extension_info_map_->AddExtension(&extension,
                                         base::Time::Now(),
                                         false,   // incognito_enabled
                                         false);  // notifications_disabled
     }
     std::unique_ptr<net::URLRequest> request(
         resource_context_.GetRequestContext()->CreateRequest(
             extension.GetResourceURL(relative_path), net::DEFAULT_PRIORITY,
-            &test_delegate_));
+            &test_delegate_, TRAFFIC_ANNOTATION_FOR_TESTS));
     StartRequest(request.get(), content::RESOURCE_TYPE_MAIN_FRAME);
     return test_delegate_.request_status();
   }
 
  protected:
   content::TestBrowserThreadBundle thread_bundle_;
   scoped_refptr<InfoMap> extension_info_map_;
   net::URLRequestJobFactoryImpl job_factory_;
   const net::URLRequestJobFactory* old_factory_;
   net::TestDelegate test_delegate_;
@@ skipping 34 matching lines @@
         extension.get(), base::Time::Now(), cases[i].incognito_enabled, false);
 
     // First test a main frame request.
     {
       // It doesn't matter that the resource doesn't exist. If the resource
       // is blocked, we should see BLOCKED_BY_CLIENT. Otherwise, the request
       // should just fail because the file doesn't exist.
       std::unique_ptr<net::URLRequest> request(
           resource_context_.GetRequestContext()->CreateRequest(
               extension->GetResourceURL("404.html"), net::DEFAULT_PRIORITY,
-              &test_delegate_));
+              &test_delegate_, TRAFFIC_ANNOTATION_FOR_TESTS));
       StartRequest(request.get(), content::RESOURCE_TYPE_MAIN_FRAME);
 
       if (cases[i].should_allow_main_frame_load) {
         EXPECT_EQ(net::ERR_FILE_NOT_FOUND, test_delegate_.request_status())
             << cases[i].name;
       } else {
         EXPECT_EQ(net::ERR_BLOCKED_BY_CLIENT, test_delegate_.request_status())
             << cases[i].name;
       }
     }
 
     // Now do a subframe request.
     {
       // With PlzNavigate, the subframe navigation requests are blocked in
       // ExtensionNavigationThrottle which isn't added in this unit test. This
       // is tested in an integration test in
       // ExtensionResourceRequestPolicyTest.IframeNavigateToInaccessible.
       if (!content::IsBrowserSideNavigationEnabled()) {
         std::unique_ptr<net::URLRequest> request(
             resource_context_.GetRequestContext()->CreateRequest(
                 extension->GetResourceURL("404.html"), net::DEFAULT_PRIORITY,
-                &test_delegate_));
+                &test_delegate_, TRAFFIC_ANNOTATION_FOR_TESTS));
         StartRequest(request.get(), content::RESOURCE_TYPE_SUB_FRAME);
 
         if (cases[i].should_allow_sub_frame_load) {
           EXPECT_EQ(net::ERR_FILE_NOT_FOUND, test_delegate_.request_status())
               << cases[i].name;
         } else {
           EXPECT_EQ(net::ERR_BLOCKED_BY_CLIENT, test_delegate_.request_status())
               << cases[i].name;
         }
       }
@@ skipping 21 matching lines @@
   extension_info_map_->AddExtension(extension.get(),
                                     base::Time::Now(),
                                     false,
                                     false);
 
   // First test it with the extension enabled.
   {
     std::unique_ptr<net::URLRequest> request(
         resource_context_.GetRequestContext()->CreateRequest(
             extension->GetResourceURL("webstore_icon_16.png"),
-            net::DEFAULT_PRIORITY, &test_delegate_));
+            net::DEFAULT_PRIORITY, &test_delegate_,
+            TRAFFIC_ANNOTATION_FOR_TESTS));
     StartRequest(request.get(), content::RESOURCE_TYPE_MEDIA);
     EXPECT_EQ(net::OK, test_delegate_.request_status());
     CheckForContentLengthHeader(request.get());
   }
 
   // And then test it with the extension disabled.
   extension_info_map_->RemoveExtension(extension->id(),
                                        UnloadedExtensionInfo::REASON_DISABLE);
   {
     std::unique_ptr<net::URLRequest> request(
         resource_context_.GetRequestContext()->CreateRequest(
             extension->GetResourceURL("webstore_icon_16.png"),
-            net::DEFAULT_PRIORITY, &test_delegate_));
+            net::DEFAULT_PRIORITY, &test_delegate_,
+            TRAFFIC_ANNOTATION_FOR_TESTS));
     StartRequest(request.get(), content::RESOURCE_TYPE_MEDIA);
     EXPECT_EQ(net::OK, test_delegate_.request_status());
     CheckForContentLengthHeader(request.get());
   }
 }
 
 // Tests that a URL request for resource from an extension returns a few
 // expected response headers.
 TEST_F(ExtensionProtocolsTest, ResourceRequestResponseHeaders) {
   // Register a non-incognito extension protocol handler.
   SetProtocolHandler(false);
 
   scoped_refptr<Extension> extension = CreateTestResponseHeaderExtension();
   extension_info_map_->AddExtension(extension.get(),
                                     base::Time::Now(),
                                     false,
                                     false);
 
   {
     std::unique_ptr<net::URLRequest> request(
         resource_context_.GetRequestContext()->CreateRequest(
             extension->GetResourceURL("test.dat"), net::DEFAULT_PRIORITY,
-            &test_delegate_));
+            &test_delegate_, TRAFFIC_ANNOTATION_FOR_TESTS));
     StartRequest(request.get(), content::RESOURCE_TYPE_MEDIA);
     EXPECT_EQ(net::OK, test_delegate_.request_status());
 
     // Check that cache-related headers are set.
     std::string etag;
     request->GetResponseHeaderByName("ETag", &etag);
     EXPECT_TRUE(base::StartsWith(etag, "\"", base::CompareCase::SENSITIVE));
     EXPECT_TRUE(base::EndsWith(etag, "\"", base::CompareCase::SENSITIVE));
 
     std::string revalidation_header;
@@ skipping 20 matching lines @@
                                     false,
                                     false);
 
   // All MAIN_FRAME requests should succeed. SUB_FRAME requests that are not
   // explicitly listed in web_accesible_resources or same-origin to the parent
   // should not succeed.
   {
     std::unique_ptr<net::URLRequest> request(
         resource_context_.GetRequestContext()->CreateRequest(
             extension->GetResourceURL("test.dat"), net::DEFAULT_PRIORITY,
-            &test_delegate_));
+            &test_delegate_, TRAFFIC_ANNOTATION_FOR_TESTS));
     StartRequest(request.get(), content::RESOURCE_TYPE_MAIN_FRAME);
     EXPECT_EQ(net::OK, test_delegate_.request_status());
   }
   {
     // With PlzNavigate, the subframe navigation requests are blocked in
     // ExtensionNavigationThrottle which isn't added in this unit test. This is
     // tested in an integration test in
     // ExtensionResourceRequestPolicyTest.IframeNavigateToInaccessible.
     if (!content::IsBrowserSideNavigationEnabled()) {
       std::unique_ptr<net::URLRequest> request(
           resource_context_.GetRequestContext()->CreateRequest(
               extension->GetResourceURL("test.dat"), net::DEFAULT_PRIORITY,
-              &test_delegate_));
+              &test_delegate_, TRAFFIC_ANNOTATION_FOR_TESTS));
       StartRequest(request.get(), content::RESOURCE_TYPE_SUB_FRAME);
       EXPECT_EQ(net::ERR_BLOCKED_BY_CLIENT, test_delegate_.request_status());
     }
   }
 
   // And subresource types, such as media, should fail.
   {
     std::unique_ptr<net::URLRequest> request(
         resource_context_.GetRequestContext()->CreateRequest(
             extension->GetResourceURL("test.dat"), net::DEFAULT_PRIORITY,
-            &test_delegate_));
+            &test_delegate_, TRAFFIC_ANNOTATION_FOR_TESTS));
     StartRequest(request.get(), content::RESOURCE_TYPE_MEDIA);
     EXPECT_EQ(net::ERR_BLOCKED_BY_CLIENT, test_delegate_.request_status());
   }
 }
 
 TEST_F(ExtensionProtocolsTest, MetadataFolder) {
   SetProtocolHandler(false);
 
   base::FilePath extension_dir = GetTestPath("metadata_folder");
   std::string error;
@@ skipping 113 matching lines @@
   content_verifier_->OnExtensionLoaded(testing_profile_.get(), extension.get());
   // Wait for PostTask to ContentVerifierIOData::AddData() to finish.
   content::RunAllPendingInMessageLoop();
 
   // Request foo.js.
   EXPECT_EQ(net::OK, DoRequest(*extension, kFooJs));
   test_job_delegate.WaitForDoneReading(extension->id());
 }
 
 }  // namespace extensions