Added CrOS-specific OAuth2 browser test.

google_apis/gaia/fake_gaia.h

 // Copyright (c) 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 GOOGLE_APIS_GAIA_FAKE_GAIA_H_
 #define GOOGLE_APIS_GAIA_FAKE_GAIA_H_
 
 #include <map>
 #include <set>
 #include <string>
 
 #include "base/basictypes.h"
+#include "base/callback.h"
 #include "base/memory/scoped_ptr.h"
 #include "url/gurl.h"
 
 namespace base {
 class DictionaryValue;
 }
 
 namespace net {
 namespace test_server {
 class BasicHttpResponse;
@@ skipping 19 matching lines @@
     std::string audience;
     std::string user_id;
     ScopeSet scopes;
     int expires_in;
     std::string email;
   };
 
   FakeGaia();
   ~FakeGaia();
 
+  // Sets the initial value of tokens and cookies.
+  void SetAuthTokens(const std::string& auth_code,
+                     const std::string& refresh_token,
+                     const std::string& access_token,
+                     const std::string& gaia_uber_token,
+                     const std::string& session_sid_cookie,
+                     const std::string& session_lsid_cookie);
+
+  // Initializes HTTP request handlers. Should be called after switches
+  // for tweaking GaiaUrls are in place.
+  void Initialize();
+
   // Handles a request and returns a response if the request was recognized as a
   // GAIA request. Note that this respects the switches::kGaiaUrl and friends so
   // that this can used with EmbeddedTestServer::RegisterRequestHandler().
   scoped_ptr<net::test_server::HttpResponse> HandleRequest(
       const net::test_server::HttpRequest& request);
 
   // Configures an OAuth2 token that'll be returned when a client requests an
   // access token for the given auth token, which can be a refresh token or an
   // login-scoped access token for the token minting endpoint. Note that the
   // scope and audience requested by the client need to match the token_info.
@@ skipping 12 matching lines @@
                                 std::string* value);
 
  private:
   typedef std::multimap<std::string, AccessTokenInfo> AccessTokenInfoMap;
   typedef std::map<std::string, GURL> SamlAccountIdpMap;
 
   // Formats a JSON response with the data in |response_dict|.
   void FormatJSONResponse(const base::DictionaryValue& response_dict,
                           net::test_server::BasicHttpResponse* http_response);
 
+  typedef base::Callback<void(
+      const net::test_server::HttpRequest& request,
+      net::test_server::BasicHttpResponse* http_response)>
+          HttpRequestHandlerCallback;
+  typedef std::map<std::string, HttpRequestHandlerCallback> RequestHandlerMap;
+
+  // HTTP request handlers.
+  void HandleProgramaticAuth(
+      const net::test_server::HttpRequest& request,
+      net::test_server::BasicHttpResponse* http_response);
+  void HandleServiceLogin(const net::test_server::HttpRequest& request,
+                          net::test_server::BasicHttpResponse* http_response);
+  void HandleOAuthLogin(const net::test_server::HttpRequest& request,
+                        net::test_server::BasicHttpResponse* http_response);
+  void HandleSSO(const net::test_server::HttpRequest& request,
+                 net::test_server::BasicHttpResponse* http_response);
+  void HandleMergeSession(const net::test_server::HttpRequest& request,
+                          net::test_server::BasicHttpResponse* http_response);
+  void HandleServiceLoginAuth(
+      const net::test_server::HttpRequest& request,
+      net::test_server::BasicHttpResponse* http_response);
+  void HandleAuthToken(const net::test_server::HttpRequest& request,
+                       net::test_server::BasicHttpResponse* http_response);
+  void HandleTokenInfo(const net::test_server::HttpRequest& request,
+                       net::test_server::BasicHttpResponse* http_response);
+  void HandleIssueToken(const net::test_server::HttpRequest& request,
+                        net::test_server::BasicHttpResponse* http_response);
+
   // Returns the access token associated with |auth_token| that matches the
   // given |client_id| and |scope_string|. If |scope_string| is empty, the first
   // token satisfying the other criteria is returned. Returns NULL if no token
   // matches.
-  const AccessTokenInfo* GetAccessTokenInfo(const std::string& auth_token,
-                                            const std::string& client_id,
-                                            const std::string& scope_string)
+  const AccessTokenInfo* FindAccessTokenInfo(const std::string& auth_token,
+                                             const std::string& client_id,
+                                             const std::string& scope_string)
       const;
 
+  // Extracts the |access_token| from authorization header of |request|.
+  static bool GetAccessToken(const net::test_server::HttpRequest& request,
+                             const char* auth_token_prefix,
+                             std::string* access_token);
+
+  // auth_code cookie value response for /o/oauth2/programmatic_auth call.
+  std::string fake_auth_code_;
+
+  // refresh_token field value response for the initial /o/oauth2/token call
+  // with ...&grant_type=authorization_code.
+  std::string fake_refresh_token_;
+  std::string fake_access_token_;
+  std::string fake_gaia_uber_token_;
+  std::string fake_session_sid_cookie_;
+  std::string fake_session_lsid_cookie_;
+
   AccessTokenInfoMap access_token_info_map_;
+  RequestHandlerMap request_handlers_;
   std::string service_login_response_;
   SamlAccountIdpMap saml_account_idp_map_;
 
   DISALLOW_COPY_AND_ASSIGN(FakeGaia);
 };
 
 #endif  // GOOGLE_APIS_GAIA_FAKE_GAIA_H_