Network traffic annotation stored in url_request.

google_apis/gaia/oauth2_access_token_fetcher_impl_unittest.cc

 // Copyright 2014 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 OAuth2AccessTokenFetcherImpl.
 
 #include "google_apis/gaia/oauth2_access_token_fetcher_impl.h"
 
 #include <memory>
 #include <string>
 
 #include "base/run_loop.h"
 #include "base/test/scoped_task_environment.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "build/build_config.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 "net/base/net_errors.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"
 #include "url/gurl.h"
@@ skipping 27 matching lines @@
     "{"
     "  \"error\": \"invalid_grant\""
     "}";
 
 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 MockOAuth2AccessTokenConsumer : public OAuth2AccessTokenConsumer {
  public:
   MockOAuth2AccessTokenConsumer() {}
   ~MockOAuth2AccessTokenConsumer() {}
 
   MOCK_METHOD2(OnGetTokenSuccess,
                void(const std::string& access_token,
@@ skipping 21 matching lines @@
     TestURLFetcher* url_fetcher = new TestURLFetcher(0, url, &fetcher_);
     net::Error error = fetch_succeeds ? net::OK : net::ERR_FAILED;
     url_fetcher->set_status(URLRequestStatus::FromError(error));
 
     if (response_code != 0)
       url_fetcher->set_response_code(response_code);
 
     if (!body.empty())
       url_fetcher->SetResponseString(body);
 
-    EXPECT_CALL(factory_, CreateURLFetcherMock(_, url, _, _))
+    EXPECT_CALL(factory_, CreateURLFetcherMock(_, url, _, _, _))
         .WillOnce(Return(url_fetcher));
     return url_fetcher;
   }
 
  protected:
   base::test::ScopedTaskEnvironment scoped_task_environment_;
   MockUrlFetcherFactory factory_;
   MockOAuth2AccessTokenConsumer consumer_;
   scoped_refptr<net::TestURLRequestContextGetter> request_context_getter_;
   OAuth2AccessTokenFetcherImpl fetcher_;
@@ skipping 135 matching lines @@
     TestURLFetcher url_fetcher(0, GURL("http://www.google.com"), NULL);
     url_fetcher.SetResponseString(kValidFailureTokenResponse);
 
     std::string error;
     EXPECT_TRUE(
         OAuth2AccessTokenFetcherImpl::ParseGetAccessTokenFailureResponse(
             &url_fetcher, &error));
     EXPECT_EQ("invalid_grant", error);
   }
 }