Making OAuth2TokenService multi-login aware, updating callers, minor fixes

chrome/browser/signin/fake_profile_oauth2_token_service.h

 // Copyright 2013 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 CHROME_BROWSER_SIGNIN_FAKE_PROFILE_OAUTH2_TOKEN_SERVICE_H_
 #define CHROME_BROWSER_SIGNIN_FAKE_PROFILE_OAUTH2_TOKEN_SERVICE_H_
 
 #include "chrome/browser/signin/profile_oauth2_token_service.h"
 
 #include <string>
@@ skipping 25 matching lines @@
 // ...
 // // ...or make them fail...
 // IssueErrorForScope(scopes, GoogleServiceAuthError(INVALID_GAIA_CREDENTIALS));
 //
 class FakeProfileOAuth2TokenService : public ProfileOAuth2TokenService {
  public:
   struct PendingRequest {
     PendingRequest();
     ~PendingRequest();
 
+    std::string account_id;
     std::string client_id;
     std::string client_secret;
     ScopeSet scopes;
     base::WeakPtr<RequestImpl> request;
   };
 
   FakeProfileOAuth2TokenService();
   virtual ~FakeProfileOAuth2TokenService();
 
   // Sets the current refresh token. If |token| is non-empty, this will invoke
   // OnRefreshTokenAvailable() on all Observers, otherwise this will invoke
   // OnRefreshTokenRevoked().
   void IssueRefreshToken(const std::string& token);
 
   // Gets a list of active requests (can be used by tests to validate that the
   // correct request has been issued).
   std::vector<PendingRequest> GetPendingRequests();
 
   // Helper routines to issue tokens for pending requests.
+  // TODO(fgorski): Add account IDs as parameters.
   void IssueTokenForScope(const ScopeSet& scopes,
                           const std::string& access_token,
                           const base::Time& expiration);
 
   void IssueErrorForScope(const ScopeSet& scopes,
                           const GoogleServiceAuthError& error);
 
   void IssueTokenForAllPendingRequests(const std::string& access_token,
                                        const base::Time& expiration);
 
   void IssueErrorForAllPendingRequests(const GoogleServiceAuthError& error);
 
+  virtual void Initialize(Profile* profile) OVERRIDE;
   virtual void Shutdown() OVERRIDE;
 
   // Helper function to be used with
   // BrowserContextKeyedService::SetTestingFactory().
   static BrowserContextKeyedService* Build(content::BrowserContext* profile);
 
  protected:
   // OAuth2TokenService overrides.
   virtual void FetchOAuth2Token(RequestImpl* request,
+                                const std::string& account_id,
                                 net::URLRequestContextGetter* getter,
                                 const std::string& client_id,
                                 const std::string& client_secret,
                                 const ScopeSet& scopes) OVERRIDE;
 
-  virtual std::string GetRefreshToken() OVERRIDE;
+  virtual std::string GetRefreshToken(const std::string& account_id) OVERRIDE;
 
   virtual net::URLRequestContextGetter* GetRequestContext() OVERRIDE;
 
  private:
   // Helper function to complete pending requests - if |all_scopes| is true,
   // then all pending requests are completed, otherwise, only those requests
   // matching |scopes| are completed.
   void CompleteRequests(bool all_scopes,
                         const ScopeSet& scopes,
                         const GoogleServiceAuthError& error,
                         const std::string& access_token,
                         const base::Time& expiration);
 
   std::vector<PendingRequest> pending_requests_;
   std::string refresh_token_;
+  bool is_initialized_;
 
   DISALLOW_COPY_AND_ASSIGN(FakeProfileOAuth2TokenService);
 };
 
 #endif  // CHROME_BROWSER_SIGNIN_FAKE_PROFILE_OAUTH2_TOKEN_SERVICE_H_