Refactoring DownloadManager::DownloadURL to add proper annotation.

content/browser/background_fetch/background_fetch_job_controller_unittest.cc

 // Copyright 2017 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/background_fetch/background_fetch_job_controller.h"
 
 #include <memory>
 #include <string>
 #include <unordered_map>
 
 #include "base/guid.h"
 #include "base/macros.h"
 #include "base/run_loop.h"
 #include "content/browser/background_fetch/background_fetch_constants.h"
 #include "content/browser/background_fetch/background_fetch_data_manager.h"
 #include "content/browser/background_fetch/background_fetch_registration_id.h"
 #include "content/browser/background_fetch/background_fetch_test_base.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/download_item.h"
 #include "content/public/browser/storage_partition.h"
 #include "content/public/test/fake_download_item.h"
 #include "content/public/test/mock_download_manager.h"
+#include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 using testing::_;
 
 namespace content {
 namespace {
 
 const char kExampleTag[] = "my-example-tag";
 
@@ skipping 108 matching lines @@
   ASSERT_NO_FATAL_FAILURE(CreateRegistrationForRequests(
       &registration_id, &initial_requests,
       {{"https://example.com/funny_cat.png", "GET"}}));
 
   std::unique_ptr<BackgroundFetchJobController> controller =
       CreateJobController(registration_id);
 
   EXPECT_EQ(controller->state(),
             BackgroundFetchJobController::State::INITIALIZED);
 
-  controller->Start(initial_requests /* deliberate copy */);
+  controller->Start(initial_requests /* deliberate copy */,
+                    TRAFFIC_ANNOTATION_FOR_TESTS);
   EXPECT_EQ(controller->state(), BackgroundFetchJobController::State::FETCHING);
 
   // Mark the single download item as finished, completing the job.
   {
     base::RunLoop run_loop;
     job_completed_closure_ = run_loop.QuitClosure();
 
     run_loop.Run();
   }
 
@@ skipping 23 matching lines @@
 
   EXPECT_EQ(controller->state(),
             BackgroundFetchJobController::State::INITIALIZED);
 
   // Continue spinning until the Job Controller has completed all the requests.
   // The Download Manager has been told to automatically mark them as finished.
   {
     base::RunLoop run_loop;
     job_completed_closure_ = run_loop.QuitClosure();
 
-    controller->Start(initial_requests /* deliberate copy */);
+    controller->Start(initial_requests /* deliberate copy */,
+                      TRAFFIC_ANNOTATION_FOR_TESTS);
     EXPECT_EQ(controller->state(),
               BackgroundFetchJobController::State::FETCHING);
 
     run_loop.Run();
   }
 
   EXPECT_EQ(controller->state(),
             BackgroundFetchJobController::State::COMPLETED);
   EXPECT_TRUE(did_complete_job_);
 }
@@ skipping 10 matching lines @@
       CreateJobController(registration_id);
 
   EXPECT_EQ(controller->state(),
             BackgroundFetchJobController::State::INITIALIZED);
 
   // Start the set of |initial_requests|, and abort them immediately after.
   {
     base::RunLoop run_loop;
     job_completed_closure_ = run_loop.QuitClosure();
 
-    controller->Start(initial_requests /* deliberate copy */);
+    controller->Start(initial_requests /* deliberate copy */,
+                      TRAFFIC_ANNOTATION_FOR_TESTS);
     EXPECT_EQ(controller->state(),
               BackgroundFetchJobController::State::FETCHING);
 
     controller->Abort();
 
     run_loop.Run();
   }
 
   // TODO(peter): Verify that the issued download items have had their state
   // updated to be cancelled as well.
 
   EXPECT_EQ(controller->state(), BackgroundFetchJobController::State::ABORTED);
   EXPECT_TRUE(did_complete_job_);
 }
 
 }  // namespace
 }  // namespace content