Network traffic annotation stored in url_request.

google_apis/gaia/oauth2_api_call_flow_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.
 //
 // A complete set of unit tests for OAuth2MintTokenFlow.
 
 #include "google_apis/gaia/oauth2_api_call_flow.h"
 
 #include <memory>
 #include <string>
 #include <vector>
 
 #include "base/message_loop/message_loop.h"
 #include "base/time/time.h"
 #include "google_apis/gaia/gaia_urls.h"
 #include "google_apis/gaia/google_service_auth_error.h"
 #include "google_apis/gaia/oauth2_access_token_consumer.h"
 #include "google_apis/gaia/oauth2_access_token_fetcher_impl.h"
 #include "net/base/net_errors.h"
 #include "net/http/http_request_headers.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.h"
 #include "net/url_request/url_fetcher_delegate.h"
 #include "net/url_request/url_fetcher_factory.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_status.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
@@ skipping 24 matching lines @@
 // Replaces the global URLFetcher factory so the test can return a custom
 // URLFetcher to complete requests.
 class MockUrlFetcherFactory : public ScopedURLFetcherFactory,
                               public URLFetcherFactory {
  public:
   MockUrlFetcherFactory()
       : ScopedURLFetcherFactory(this) {
   }
   virtual ~MockUrlFetcherFactory() {}
 
-  MOCK_METHOD4(CreateURLFetcherMock,
-               URLFetcher*(int id,
-                           const GURL& url,
-                           URLFetcher::RequestType request_type,
-                           URLFetcherDelegate* d));
+  MOCK_METHOD5(
+      CreateURLFetcherMock,
+      URLFetcher*(int id,
+                  const GURL& url,
+                  URLFetcher::RequestType request_type,
+                  URLFetcherDelegate* d,
+                  net::NetworkTrafficAnnotationTag traffic_annotation));
 
   std::unique_ptr<URLFetcher> CreateURLFetcher(
       int id,
       const GURL& url,
       URLFetcher::RequestType request_type,
-      URLFetcherDelegate* d) override {
+      URLFetcherDelegate* d,
+      net::NetworkTrafficAnnotationTag traffic_annotation) override {
     return std::unique_ptr<URLFetcher>(
-        CreateURLFetcherMock(id, url, request_type, d));
+        CreateURLFetcherMock(id, url, request_type, d, traffic_annotation));
   }
 };
 
 class MockApiCallFlow : public OAuth2ApiCallFlow {
  public:
   MockApiCallFlow() {}
   ~MockApiCallFlow() {}
 
   MOCK_METHOD0(CreateApiCallUrl, GURL ());
   MOCK_METHOD0(CreateApiCallBody, std::string ());
@@ skipping 31 matching lines @@
     return url_fetcher;
   }
 
   TestURLFetcher* SetupApiCall(bool succeeds, net::HttpStatusCode status) {
     std::string body(CreateBody());
     GURL url(CreateApiUrl());
     EXPECT_CALL(flow_, CreateApiCallBody()).WillOnce(Return(body));
     EXPECT_CALL(flow_, CreateApiCallUrl()).WillOnce(Return(url));
     TestURLFetcher* url_fetcher =
         CreateURLFetcher(url, succeeds, status, std::string());
-    EXPECT_CALL(factory_, CreateURLFetcherMock(_, url, _, _))
+    EXPECT_CALL(factory_, CreateURLFetcherMock(_, url, _, _, _))
         .WillOnce(Return(url_fetcher));
     return url_fetcher;
   }
 
   base::MessageLoop message_loop_;
   scoped_refptr<net::TestURLRequestContextGetter> request_context_getter_;
   StrictMock<MockApiCallFlow> flow_;
   MockUrlFetcherFactory factory_;
 };
 
@@ skipping 25 matching lines @@
   TestURLFetcher* url_fetcher = SetupApiCall(true, net::HTTP_OK);
   flow_.Start(request_context_getter_.get(), kAccessToken);
   HttpRequestHeaders headers;
   url_fetcher->GetExtraRequestHeaders(&headers);
   std::string auth_header;
   EXPECT_TRUE(headers.GetHeader("Authorization", &auth_header));
   EXPECT_EQ("Bearer access_token", auth_header);
   EXPECT_EQ(url, url_fetcher->GetOriginalURL());
   EXPECT_EQ(body, url_fetcher->upload_data());
 }