Basic tests for new webview based signin flow

google_apis/gaia/fake_gaia.cc

 // 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.
 
 #include "google_apis/gaia/fake_gaia.h"
 
 #include <vector>
 
 #include "base/base_paths.h"
 #include "base/bind.h"
@@ skipping 38 matching lines @@
 const char kTestSessionLSIDCookie[] = "fake-session-LSID-cookie";
 const char kTestOAuthLoginSID[] = "fake-oauth-SID-cookie";
 const char kTestOAuthLoginLSID[] = "fake-oauth-LSID-cookie";
 const char kTestOAuthLoginAuthCode[] = "fake-oauth-auth-code";
 
 const char kDefaultGaiaId[]  ="12345";
 
 const base::FilePath::CharType kServiceLogin[] =
     FILE_PATH_LITERAL("google_apis/test/service_login.html");
 
+const base::FilePath::CharType kEmbeddedSetupChromeos[] =
+    FILE_PATH_LITERAL("google_apis/test/embedded_setup_chromeos.html");
+
 // OAuth2 Authentication header value prefix.
 const char kAuthHeaderBearer[] = "Bearer ";
 const char kAuthHeaderOAuth[] = "OAuth ";
 
 const char kListAccountsResponseFormat[] =
     "[\"gaia.l.a.r\",[[\"gaia.l.a\",1,\"\",\"%s\",\"\",1,1,0]]]";
 
 typedef std::map<std::string, std::string> CookieMap;
 
 // Parses cookie name-value map our of |request|.
@@ skipping 63 matching lines @@
 
 FakeGaia::MergeSessionParams::~MergeSessionParams() {
 }
 
 FakeGaia::FakeGaia() : issue_oauth_code_cookie_(false) {
   base::FilePath source_root_dir;
   PathService::Get(base::DIR_SOURCE_ROOT, &source_root_dir);
   CHECK(base::ReadFileToString(
       source_root_dir.Append(base::FilePath(kServiceLogin)),
       &service_login_response_));
+  CHECK(base::ReadFileToString(
+      source_root_dir.Append(base::FilePath(kEmbeddedSetupChromeos)),
+      &embedded_setup_chromeos_response_));
 }
 
 FakeGaia::~FakeGaia() {}
 
 void FakeGaia::SetFakeMergeSessionParams(
     const std::string& email,
     const std::string& auth_sid_cookie,
     const std::string& auth_lsid_cookie) {
   FakeGaia::MergeSessionParams params;
   params.auth_sid_cookie = auth_sid_cookie;
@@ skipping 54 matching lines @@
 
   // Handles /o/oauth2/programmatic_auth GAIA call.
   REGISTER_RESPONSE_HANDLER(
       gaia_urls->client_login_to_oauth2_url(), HandleProgramaticAuth);
 
   // Handles /ServiceLogin GAIA call.
   REGISTER_RESPONSE_HANDLER(
       gaia_urls->service_login_url(), HandleServiceLogin);
 
   // Handles /embedded/setup/chromeos GAIA call.
-  // Same handler as for /ServiceLogin is used for now.
-  REGISTER_RESPONSE_HANDLER(
-      gaia_urls->embedded_setup_chromeos_url(), HandleServiceLogin);
+  REGISTER_RESPONSE_HANDLER(gaia_urls->embedded_setup_chromeos_url(),
+                            HandleEmbeddedSetupChromeos);
 
   // Handles /OAuthLogin GAIA call.
   REGISTER_RESPONSE_HANDLER(
       gaia_urls->oauth1_login_url(), HandleOAuthLogin);
 
   // Handles /ServiceLoginAuth GAIA call.
   REGISTER_RESPONSE_HANDLER(
       gaia_urls->service_login_auth_url(), HandleServiceLoginAuth);
 
+  // Handles /_/embedded/lookup/accountlookup for /embedded/setup/chromeos
+  // authentication request.
+  REGISTER_PATH_RESPONSE_HANDLER("/_/embedded/lookup/accountlookup",
+                                 HandleEmbeddedLookupAccountLookup);
+
+  // Handles /_/embedded/signin/challenge for /embedded/setup/chromeos
+  // authentication request.
+  REGISTER_PATH_RESPONSE_HANDLER("/_/embedded/signin/challenge",
+                                 HandleEmbeddedSigninChallenge);
+
   // Handles /SSO GAIA call (not GAIA, made up for SAML tests).
   REGISTER_PATH_RESPONSE_HANDLER("/SSO", HandleSSO);
 
   // Handles /o/oauth2/token GAIA call.
   REGISTER_RESPONSE_HANDLER(
       gaia_urls->oauth2_token_url(), HandleAuthToken);
 
   // Handles /oauth2/v2/tokeninfo GAIA call.
   REGISTER_RESPONSE_HANDLER(
       gaia_urls->oauth2_token_info_url(), HandleTokenInfo);
@@ skipping 179 matching lines @@
   return NULL;
 }
 
 void FakeGaia::HandleServiceLogin(const HttpRequest& request,
                                   BasicHttpResponse* http_response) {
   http_response->set_code(net::HTTP_OK);
   http_response->set_content(service_login_response_);
   http_response->set_content_type("text/html");
 }
 
+void FakeGaia::HandleEmbeddedSetupChromeos(const HttpRequest& request,
+                                           BasicHttpResponse* http_response) {
+  http_response->set_code(net::HTTP_OK);
+  http_response->set_content(embedded_setup_chromeos_response_);
+  http_response->set_content_type("text/html");
+}
+
 void FakeGaia::HandleOAuthLogin(const HttpRequest& request,
                                 BasicHttpResponse* http_response) {
   http_response->set_code(net::HTTP_UNAUTHORIZED);
   if (merge_session_params_.gaia_uber_token.empty()) {
     http_response->set_code(net::HTTP_FORBIDDEN);
     return;
   }
 
   std::string access_token;
   if (!GetAccessToken(request, kAuthHeaderBearer, &access_token) &&
@@ skipping 57 matching lines @@
 
   // SAML sign-ins complete in HandleSSO().
   if (is_saml)
     return;
 
   AddGoogleAccountsSigninHeader(http_response, email);
   if (issue_oauth_code_cookie_)
     SetOAuthCodeCookie(http_response);
 }
 
+void FakeGaia::HandleEmbeddedLookupAccountLookup(
+    const net::test_server::HttpRequest& request,
+    net::test_server::BasicHttpResponse* http_response) {
+  std::string email;
+  const bool is_saml =
+      GetQueryParameter(request.content, "identifier", &email) &&
+      saml_account_idp_map_.find(email) != saml_account_idp_map_.end();
+
+  if (!is_saml)
+    return;
+
+  GURL url(saml_account_idp_map_[email]);
+  url = net::AppendQueryParameter(url, "SAMLRequest", "fake_request");
+  url = net::AppendQueryParameter(
+      url, "RelayState",
+      "chrome-extension://mfffpogegjflfpflabcdkioaeobkgjik/success.html");
+  std::string redirect_url = url.spec();
+  http_response->AddCustomHeader("Google-Accounts-SAML", "Start");
+
+  http_response->AddCustomHeader("continue", redirect_url);
+}
+
+void FakeGaia::HandleEmbeddedSigninChallenge(const HttpRequest& request,
+                                             BasicHttpResponse* http_response) {
+  std::string email;
+  GetQueryParameter(request.content, "identifier", &email);
+
+  if (!merge_session_params_.auth_sid_cookie.empty() &&
+      !merge_session_params_.auth_lsid_cookie.empty()) {
+    SetCookies(http_response, merge_session_params_.auth_sid_cookie,
+               merge_session_params_.auth_lsid_cookie);
+  }
+
+  AddGoogleAccountsSigninHeader(http_response, email);
+
+  if (issue_oauth_code_cookie_)
+    SetOAuthCodeCookie(http_response);
+}
+
 void FakeGaia::HandleSSO(const HttpRequest& request,
                          BasicHttpResponse* http_response) {
   if (!merge_session_params_.auth_sid_cookie.empty() &&
       !merge_session_params_.auth_lsid_cookie.empty()) {
     SetCookies(http_response,
                merge_session_params_.auth_sid_cookie,
                merge_session_params_.auth_lsid_cookie);
   }
   std::string relay_state;
   GetQueryParameter(request.content, "RelayState", &relay_state);
@@ skipping 137 matching lines @@
   if (token_info) {
     base::DictionaryValue response_dict;
     response_dict.SetString("id", GetGaiaIdOfEmail(token_info->email));
     response_dict.SetString("email", token_info->email);
     response_dict.SetString("verified_email", token_info->email);
     FormatJSONResponse(response_dict, http_response);
   } else {
     http_response->set_code(net::HTTP_BAD_REQUEST);
   }
 }