Network traffic annotation added to OAuth2ApiCallFlow and its subclasses.

components/cryptauth/mock_cryptauth_client.h

 // Copyright 2015 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.
 
 #ifndef COMPONENTS_CRYPTAUTH_MOCK_CRYPTAUTH_CLIENT_H_
 #define COMPONENTS_CRYPTAUTH_MOCK_CRYPTAUTH_CLIENT_H_
 
 #include "base/macros.h"
 #include "base/observer_list.h"
 #include "components/cryptauth/cryptauth_client.h"
 #include "components/cryptauth/proto/cryptauth_api.pb.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 namespace cryptauth {
 
 class MockCryptAuthClient : public CryptAuthClient {
  public:
   MockCryptAuthClient();
   ~MockCryptAuthClient() override;
 
   // CryptAuthClient:
-  MOCK_METHOD3(GetMyDevices,
+  MOCK_METHOD4(GetMyDevices,
                void(const GetMyDevicesRequest& request,
                     const GetMyDevicesCallback& callback,
-                    const ErrorCallback& error_callback));
+                    const ErrorCallback& error_callback,
+                    const net::PartialNetworkTrafficAnnotationTag&
+                        partial_traffic_annotation));
   MOCK_METHOD3(FindEligibleUnlockDevices,
                void(const FindEligibleUnlockDevicesRequest& request,
                     const FindEligibleUnlockDevicesCallback& callback,
                     const ErrorCallback& error_callback));
-  MOCK_METHOD3(SendDeviceSyncTickle,
+  MOCK_METHOD4(SendDeviceSyncTickle,
                void(const SendDeviceSyncTickleRequest& request,
                     const SendDeviceSyncTickleCallback& callback,
-                    const ErrorCallback& error_callback));
+                    const ErrorCallback& error_callback,
+                    const net::PartialNetworkTrafficAnnotationTag&
+                        partial_traffic_annotation));
   MOCK_METHOD3(ToggleEasyUnlock,
                void(const ToggleEasyUnlockRequest& request,
                     const ToggleEasyUnlockCallback& callback,
                     const ErrorCallback& error_callback));
   MOCK_METHOD3(SetupEnrollment,
                void(const SetupEnrollmentRequest& request,
                     const SetupEnrollmentCallback& callback,
                     const ErrorCallback& error_callback));
   MOCK_METHOD3(FinishEnrollment,
                void(const FinishEnrollmentRequest& request,
@@ skipping 35 matching lines @@
 
   // Observers of the factory.
   base::ObserverList<Observer> observer_list_;
 
   DISALLOW_COPY_AND_ASSIGN(MockCryptAuthClientFactory);
 };
 
 }  // namespace cryptauth
 
 #endif  // COMPONENTS_CRYPTAUTH_MOCK_CRYPTAUTH_CLIENT_H_