Add setup for integration tests in FakeChromeIdentityService.

ios/public/provider/chrome/browser/signin/fake_chrome_identity_service.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.
 
 #import "ios/public/provider/chrome/browser/signin/fake_chrome_identity_service.h"
 
 #import <Foundation/Foundation.h>
 
+#include "base/mac/scoped_block.h"
 #include "base/strings/sys_string_conversions.h"
+#include "google_apis/gaia/gaia_auth_util.h"
 #include "ios/public/provider/chrome/browser/chrome_browser_provider.h"
 #import "ios/public/provider/chrome/browser/signin/fake_chrome_identity.h"
+#include "ios/public/provider/chrome/browser/signin/signin_resources_provider.h"
+
+using ::testing::_;
+using ::testing::Invoke;
+
+namespace {
+
+void FakeGetAccessToken(ChromeIdentity*,
+                        const std::string&,
+                        const std::string&,
+                        const std::set<std::string>&,
+                        const ios::AccessTokenCallback& callback) {
+  base::mac::ScopedBlock<ios::AccessTokenCallback> safe_callback(
+      [callback copy]);
+
+  // |GetAccessToken| is normally an asynchronous operation (that requires some
+  // network calls), this is replicated here by dispatching it.
+  dispatch_async(dispatch_get_main_queue(), ^{
+    // Token and expiration date. It should be larger than typical test
+    // execution because tests usually setup mock to expect one token request
+    // and then rely on access token being served from cache.
+    NSTimeInterval expiration = 60.0;
+    NSDate* expiresDate = [NSDate dateWithTimeIntervalSinceNow:expiration];
+    NSString* token = [expiresDate description];
+
+    safe_callback.get()(token, expiresDate, nil);
+  });
+}
+
+UIImage* FakeGetCachedAvatarForIdentity(ChromeIdentity*) {
+  ios::SigninResourcesProvider* provider =
+      ios::GetChromeBrowserProvider()->GetSigninResourcesProvider();
+  return provider ? provider->GetDefaultAvatar() : nil;
+}
+
+void FakeGetAvatarForIdentity(ChromeIdentity* identity,
+                              ios::GetAvatarCallback callback) {
+  // |GetAvatarForIdentity| is normally an asynchronous operation, this is
+  // replicated here by dispatching it.
+  dispatch_async(dispatch_get_main_queue(), ^{
+    callback(FakeGetCachedAvatarForIdentity(identity));
+  });
+}
+
+void FakeGetHostedDomainForIdentity(ChromeIdentity* identity,
+                                    ios::GetHostedDomainCallback callback) {
+  NSString* domain = base::SysUTF8ToNSString(gaia::ExtractDomainName(
+      gaia::CanonicalizeEmail(base::SysNSStringToUTF8(identity.userEmail))));
+
+  // |GetHostedDomainForIdentity| is normally an asynchronous operation , this
+  // is replicated here by dispatching it.
+  dispatch_async(dispatch_get_main_queue(), ^{
+    callback(domain, nil);
+  });
+}
+}
 
 namespace ios {
 
 NSString* const kIdentityEmailFormat = @"%@@foo.com";
 NSString* const kIdentityGaiaIDFormat = @"%@ID";
 
 FakeChromeIdentityService::FakeChromeIdentityService()
     : identities_([[NSMutableArray alloc] init]) {}
 
 FakeChromeIdentityService::~FakeChromeIdentityService() {}
@@ skipping 43 matching lines @@
   FireIdentityListChanged();
   if (callback) {
     // Forgetting an identity is normally an asynchronous operation (that
     // require some network calls), this is replicated here by dispatching it.
     dispatch_async(dispatch_get_main_queue(), ^{
       callback(nil);
     });
   }
 }
 
+void FakeChromeIdentityService::SetUpForIntegrationTests() {
+  ON_CALL(*this, GetAccessToken(_, _, _, _, _))
+      .WillByDefault(Invoke(FakeGetAccessToken));
+
+  ON_CALL(*this, GetAvatarForIdentity(_, _))
+      .WillByDefault(Invoke(FakeGetAvatarForIdentity));
+
+  ON_CALL(*this, GetCachedAvatarForIdentity(_))
+      .WillByDefault(Invoke(FakeGetCachedAvatarForIdentity));
+
+  ON_CALL(*this, GetHostedDomainForIdentity(_, _))
+      .WillByDefault(Invoke(FakeGetHostedDomainForIdentity));
+}
+
 void FakeChromeIdentityService::AddIdentities(NSArray* identitiesNames) {
   for (NSString* name in identitiesNames) {
     NSString* email = [NSString stringWithFormat:kIdentityEmailFormat, name];
     NSString* gaiaID = [NSString stringWithFormat:kIdentityGaiaIDFormat, name];
     [identities_ addObject:[FakeChromeIdentity identityWithEmail:email
                                                           gaiaID:gaiaID
                                                             name:name]];
   }
 }
 
 void FakeChromeIdentityService::AddIdentity(ChromeIdentity* identity) {
   [identities_ addObject:identity];
   FireIdentityListChanged();
 }
 
 }  // namespace ios