Network traffic annotation added to safe_browsing

chrome/browser/safe_browsing/download_feedback_unittest.cc

 // Copyright 2013 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 "chrome/browser/safe_browsing/download_feedback.h"
 
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "chrome/browser/safe_browsing/two_phase_uploader.h"
 #include "chrome/common/safe_browsing/csd.pb.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/test/test_browser_thread_bundle.h"
+#include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace safe_browsing {
 
 namespace {
 
 class FakeUploader : public TwoPhaseUploader {
  public:
   FakeUploader(net::URLRequestContextGetter* url_request_context_getter,
@@ skipping 41 matching lines @@
   FakeUploaderFactory() : uploader_(nullptr) {}
   ~FakeUploaderFactory() override {}
 
   std::unique_ptr<TwoPhaseUploader> CreateTwoPhaseUploader(
       net::URLRequestContextGetter* url_request_context_getter,
       base::TaskRunner* file_task_runner,
       const GURL& base_url,
       const std::string& metadata,
       const base::FilePath& file_path,
       const TwoPhaseUploader::ProgressCallback& progress_callback,
-      const TwoPhaseUploader::FinishCallback& finish_callback) override;
+      const TwoPhaseUploader::FinishCallback& finish_callback,
+      const net::NetworkTrafficAnnotationTag& traffic_annotation) override;
 
   FakeUploader* uploader_;
 };
 
 std::unique_ptr<TwoPhaseUploader> FakeUploaderFactory::CreateTwoPhaseUploader(
     net::URLRequestContextGetter* url_request_context_getter,
     base::TaskRunner* file_task_runner,
     const GURL& base_url,
     const std::string& metadata,
     const base::FilePath& file_path,
     const TwoPhaseUploader::ProgressCallback& progress_callback,
-    const TwoPhaseUploader::FinishCallback& finish_callback) {
+    const TwoPhaseUploader::FinishCallback& finish_callback,
+    const net::NetworkTrafficAnnotationTag& traffic_annotation) {
   EXPECT_FALSE(uploader_);
 
   uploader_ = new FakeUploader(url_request_context_getter, file_task_runner,
                                base_url, metadata, file_path, progress_callback,
                                finish_callback);
   return base::WrapUnique(uploader_);
 }
 
 }  // namespace
 
@@ skipping 113 matching lines @@
   EXPECT_TRUE(base::PathExists(upload_file_path_));
 
   feedback.reset();
   EXPECT_FALSE(feedback_finish_called_);
 
   base::RunLoop().RunUntilIdle();
   EXPECT_FALSE(base::PathExists(upload_file_path_));
 }
 
 }  // namespace safe_browsing