chrome/browser/signin/fake_gaia_cookie_manager_service.cc

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/signin/fake_gaia_cookie_manager_service.h"
   6
   7 #include "base/strings/stringprintf.h"
   8 #include "chrome/browser/profiles/profile.h"
   9 #include "chrome/browser/signin/chrome_signin_client_factory.h"
  10 #include "chrome/browser/signin/profile_oauth2_token_service_factory.h"
  11 #include "components/signin/core/browser/profile_oauth2_token_service.h"
  12 #include "google_apis/gaia/gaia_constants.h"
  13 #include "google_apis/gaia/gaia_urls.h"
  14
  15 FakeGaiaCookieManagerService::FakeGaiaCookieManagerService(
  16     OAuth2TokenService* token_service,
  17     const std::string& source,
  18     SigninClient* client) :
  19         GaiaCookieManagerService(token_service, source, client),
  20         url_fetcher_factory_(NULL) {}
  21
  22 void FakeGaiaCookieManagerService::Init(
  23     net::FakeURLFetcherFactory* url_fetcher_factory) {
  24   url_fetcher_factory_ = url_fetcher_factory;
  25 }
  26
  27 void FakeGaiaCookieManagerService::SetListAccountsResponseHttpNotFound() {
  28   DCHECK(url_fetcher_factory_);
  29   url_fetcher_factory_->SetFakeResponse(
  30       GaiaUrls::GetInstance()->ListAccountsURLWithSource(
  31           GaiaConstants::kChromeSource),
  32       "",
  33       net::HTTP_NOT_FOUND,
  34       net::URLRequestStatus::SUCCESS);
  35 }
  36
  37 void FakeGaiaCookieManagerService::SetListAccountsResponseWebLoginRequired() {
  38   DCHECK(url_fetcher_factory_);
  39   url_fetcher_factory_->SetFakeResponse(
  40       GaiaUrls::GetInstance()->ListAccountsURLWithSource(
  41           GaiaConstants::kChromeSource),
  42       "Info=WebLoginRequired",
  43       net::HTTP_OK,
  44       net::URLRequestStatus::SUCCESS);
  45 }
  46
  47 void FakeGaiaCookieManagerService::SetListAccountsResponseNoAccounts() {
  48   DCHECK(url_fetcher_factory_);
  49   url_fetcher_factory_->SetFakeResponse(
  50       GaiaUrls::GetInstance()->ListAccountsURLWithSource(
  51           GaiaConstants::kChromeSource),
  52       "[\"f\", []]",
  53       net::HTTP_OK,
  54       net::URLRequestStatus::SUCCESS);
  55 }
  56
  57 void FakeGaiaCookieManagerService::SetListAccountsResponseOneAccount(
  58     const char* email, const char* gaia_id) {
  59   DCHECK(url_fetcher_factory_);
  60   url_fetcher_factory_->SetFakeResponse(
  61       GaiaUrls::GetInstance()->ListAccountsURLWithSource(
  62           GaiaConstants::kChromeSource),
  63       base::StringPrintf(
  64           "[\"f\", [[\"b\", 0, \"n\", \"%s\", \"p\", 0, 0, 0, 0, 1, \"%s\"]]]",
  65           email, gaia_id),
  66       net::HTTP_OK,
  67       net::URLRequestStatus::SUCCESS);
  68 }
  69
  70 void FakeGaiaCookieManagerService::SetListAccountsResponseOneAccountWithExpiry(
  71     const char* email, const char* gaia_id, bool expired) {
  72   DCHECK(url_fetcher_factory_);
  73   url_fetcher_factory_->SetFakeResponse(
  74       GaiaUrls::GetInstance()->ListAccountsURLWithSource(
  75           GaiaConstants::kChromeSource),
  76       base::StringPrintf(
  77           "[\"f\", [[\"b\", 0, \"n\", \"%s\", \"p\", 0, 0, 0, 0, %d, \"%s\"]]]",
  78           email, expired ? 0 : 1, gaia_id),
  79       net::HTTP_OK,
  80       net::URLRequestStatus::SUCCESS);
  81 }
  82
  83 void FakeGaiaCookieManagerService::SetListAccountsResponseTwoAccounts(
  84     const char* email1, const char* gaia_id1,
  85     const char* email2, const char* gaia_id2) {
  86   DCHECK(url_fetcher_factory_);
  87   url_fetcher_factory_->SetFakeResponse(
  88       GaiaUrls::GetInstance()->ListAccountsURLWithSource(
  89           GaiaConstants::kChromeSource),
  90       base::StringPrintf(
  91           "[\"f\", [[\"b\", 0, \"n\", \"%s\", \"p\", 0, 0, 0, 0, 1, \"%s\"], "
  92           "[\"b\", 0, \"n\", \"%s\", \"p\", 0, 0, 0, 0, 1, \"%s\"]]]",
  93           email1, gaia_id1, email2, gaia_id2),
  94       net::HTTP_OK,
  95       net::URLRequestStatus::SUCCESS);
  96 }
  97
  98 void FakeGaiaCookieManagerService::SetListAccountsResponseTwoAccountsWithExpiry(
  99     const char* account1, bool account1_expired,
 100     const char* account2, bool account2_expired) {
 101   DCHECK(url_fetcher_factory_);
 102   url_fetcher_factory_->SetFakeResponse(
 103       GaiaUrls::GetInstance()->ListAccountsURLWithSource(
 104           GaiaConstants::kChromeSource),
 105       base::StringPrintf(
 106           "[\"f\", [[\"b\", 0, \"n\", \"%s\", \"p\", 0, 0, 0, 0, %d, \"28\"], "
 107           "[\"b\", 0, \"n\", \"%s\", \"p\", 0, 0, 0, 0, %d, \"29\"]]]",
 108           account1, account1_expired ? 0 : 1,
 109           account2, account2_expired ? 0 : 1),
 110       net::HTTP_OK,
 111       net::URLRequestStatus::SUCCESS);
 112 }
 113
 114 // static
 115 scoped_ptr<KeyedService> FakeGaiaCookieManagerService::Build(
 116     content::BrowserContext* context) {
 117   Profile* profile = Profile::FromBrowserContext(context);
 118   return make_scoped_ptr(new FakeGaiaCookieManagerService(
 119       ProfileOAuth2TokenServiceFactory::GetForProfile(profile),
 120       GaiaConstants::kChromeSource,
 121       ChromeSigninClientFactory::GetForProfile(profile)));
 122 }