Network traffic annotation stored in url_request.

ios/chrome/test/app/signin_test_util.mm

 // 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 "ios/chrome/test/app/signin_test_util.h"
 
 #include "base/logging.h"
 #include "base/strings/stringprintf.h"
 #import "base/test/ios/wait_util.h"
 #include "components/prefs/pref_service.h"
 #include "components/signin/core/browser/account_tracker_service.h"
 #include "components/signin/core/common/signin_pref_names.h"
 #include "google_apis/gaia/gaia_constants.h"
 #include "ios/chrome/browser/browser_state/chrome_browser_state.h"
 #include "ios/chrome/browser/signin/account_tracker_service_factory.h"
 #import "ios/chrome/browser/signin/authentication_service.h"
 #include "ios/chrome/browser/signin/authentication_service_factory.h"
 #include "ios/chrome/browser/signin/gaia_auth_fetcher_ios.h"
 #import "ios/chrome/test/app/chrome_test_util.h"
 #import "ios/public/provider/chrome/browser/chrome_browser_provider.h"
 #import "ios/public/provider/chrome/browser/signin/fake_chrome_identity_service.h"
 #include "net/http/http_status_code.h"
+#include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
 #include "net/url_request/test_url_fetcher_factory.h"
 #include "net/url_request/url_fetcher_delegate.h"
 #include "net/url_request/url_request_status.h"
 
 namespace {
 
 net::FakeURLFetcherFactory* gFakeURLFetcherFactory = nullptr;
 
 // Specialization of the FakeURLFetcherFactory that will recognize GET requests
 // for MergeSession and answer those requests correctly.
 class MergeSessionFakeURLFetcherFactory : public net::FakeURLFetcherFactory {
  public:
   explicit MergeSessionFakeURLFetcherFactory(URLFetcherFactory* default_factory)
       : net::FakeURLFetcherFactory(default_factory) {}
   std::unique_ptr<net::URLFetcher> CreateURLFetcher(
       int id,
       const GURL& url,
       net::URLFetcher::RequestType request_type,
-      net::URLFetcherDelegate* d) override {
+      net::URLFetcherDelegate* d,
+      net::NetworkTrafficAnnotationTag traffic_annotation) override {
     const GURL kMergeSessionURL =
         GURL("https://accounts.google.com/MergeSession");
     url::Replacements<char> replacements;
     replacements.ClearRef();
     replacements.ClearQuery();
     if (url.ReplaceComponents(replacements) != kMergeSessionURL) {
       // URL is not a MergeSession GET. Use the default method.
-      return net::FakeURLFetcherFactory::CreateURLFetcher(id, url, request_type,
-                                                          d);
+      return net::FakeURLFetcherFactory::CreateURLFetcher(
+          id, url, request_type, d, traffic_annotation);
     }
     // Actual MergeSession request. Answer is ignored by the AccountReconcilor,
     // so it can also be empty.
     return std::unique_ptr<net::FakeURLFetcher>(new net::FakeURLFetcher(
         url, d, "", net::HTTP_OK, net::URLRequestStatus::SUCCESS));
   }
 };
 
 }  // namespace
 
@@ skipping 100 matching lines @@
   NSDate* deadline = [NSDate dateWithTimeIntervalSinceNow:10.0];
   while (identity_service->HasIdentities() &&
          [[NSDate date] compare:deadline] != NSOrderedDescending) {
     base::test::ios::SpinRunLoopWithMaxDelay(
         base::TimeDelta::FromSecondsD(0.01));
   }
   return !identity_service->HasIdentities();
 }
 
 }  // namespace chrome_test_util