Network traffic annotation added to drive/base_requests.

google_apis/drive/base_requests_unittest.cc

 // Copyright (c) 2012 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 "google_apis/drive/base_requests.h"
 
 #include <stdint.h>
 
 #include <memory>
 #include <utility>
 
 #include "base/bind.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/values.h"
 #include "google_apis/drive/drive_api_parser.h"
 #include "google_apis/drive/drive_api_requests.h"
 #include "google_apis/drive/dummy_auth_service.h"
 #include "google_apis/drive/request_sender.h"
 #include "google_apis/drive/test_util.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
 #include "net/test/embedded_test_server/http_request.h"
 #include "net/test/embedded_test_server/http_response.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 google_apis {
 
 namespace {
 
 const char kValidJsonString[] = "{ \"test\": 123 }";
 const char kInvalidJsonString[] = "$$$";
 
@@ skipping 73 matching lines @@
 }  // namespace
 
 class BaseRequestsTest : public testing::Test {
  public:
   BaseRequestsTest() : response_code_(net::HTTP_OK) {}
 
   void SetUp() override {
     request_context_getter_ = new net::TestURLRequestContextGetter(
         message_loop_.task_runner());
 
-    sender_.reset(new RequestSender(new DummyAuthService,
-                                    request_context_getter_.get(),
-                                    message_loop_.task_runner(),
-                                    std::string() /* custom user agent */));
+    sender_.reset(new RequestSender(
+        new DummyAuthService, request_context_getter_.get(),
+        message_loop_.task_runner(), std::string(), /* custom user agent */
+        TRAFFIC_ANNOTATION_FOR_TESTS));
 
     test_server_.RegisterRequestHandler(
         base::Bind(&BaseRequestsTest::HandleRequest, base::Unretained(this)));
     ASSERT_TRUE(test_server_.Start());
   }
 
   std::unique_ptr<net::test_server::HttpResponse> HandleRequest(
       const net::test_server::HttpRequest& request) {
     std::unique_ptr<net::test_server::BasicHttpResponse> response(
         new net::test_server::BasicHttpResponse);
@@ skipping 93 matching lines @@
       "\n"
       "This is a sample file. I like chocolate and chips.\n"
       "\n"
       "--TESTBOUNDARY--",
       upload_content_data);
   ASSERT_EQ(HTTP_SUCCESS, error);
   EXPECT_EQ("file_id", file->file_id());
 }
 
 }  // Namespace google_apis