Remove the Extensions URLRequestContext

chrome/browser/signin/signin_manager_unittest.cc

 // Copyright (c) 2012 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 "chrome/browser/signin/signin_manager.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/compiler_specific.h"
 #include "base/prefs/pref_service.h"
 #include "base/prefs/testing_pref_service.h"
 #include "base/run_loop.h"
 #include "base/strings/stringprintf.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/prefs/browser_prefs.h"
 #include "chrome/browser/signin/chrome_signin_manager_delegate.h"
 #include "chrome/browser/signin/token_service.h"
 #include "chrome/browser/signin/token_service_unittest.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/webdata/encryptor/encryptor.h"
 #include "content/public/browser/child_process_security_policy.h"
+#include "content/public/browser/cookie_store_map.h"
+#include "content/public/browser/storage_partition.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "google_apis/gaia/gaia_constants.h"
 #include "google_apis/gaia/gaia_urls.h"
 #include "net/cookies/cookie_monster.h"
 #include "net/url_request/test_url_fetcher_factory.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "net/url_request/url_request_status.h"
 
 #include "testing/gmock/include/gmock/gmock.h"
@@ skipping 157 matching lines @@
   void CompleteSigninCallback(const std::string& oauth_token) {
     oauth_tokens_fetched_.push_back(oauth_token);
     manager_->CompletePendingSignin();
   }
 
   void CancelSigninCallback(const std::string& oauth_token) {
     oauth_tokens_fetched_.push_back(oauth_token);
     manager_->SignOut();
   }
 
+  net::CookieMonster* GetCookieMonster(const GURL& origin) {
+    using content::BrowserContext;
+    // Since it's a unittest, assume default StoragePartition.
+    const content::CookieStoreMap& cookie_store_map =
+        BrowserContext::GetDefaultStoragePartition(profile())->
+            GetCookieStoreMap();
+    return cookie_store_map.GetForScheme(origin.scheme())->GetCookieMonster();
+  }
+
   net::TestURLFetcherFactory factory_;
   scoped_ptr<SigninManager> manager_;
   content::TestNotificationTracker google_login_success_;
   content::TestNotificationTracker google_login_failure_;
   std::vector<std::string> oauth_tokens_fetched_;
   scoped_ptr<TestingPrefServiceSimple> prefs_;
   std::vector<std::string> cookies_;
 };
 
 // NOTE: ClientLogin's "StartSignin" is called after collecting credentials
@@ skipping 69 matching lines @@
       SigninManager::OAuthTokenFetchedCallback());
 
   ExpectSignInWithCredentialsFail(true /* requestSent */);
 }
 
 TEST_F(SigninManagerTest, SignInWithCredentialsEmptyPasswordValidCookie) {
   manager_->Initialize(profile(), NULL);
   EXPECT_TRUE(manager_->GetAuthenticatedUsername().empty());
 
   // Set a valid LSID cookie in the test cookie store.
-  scoped_refptr<net::CookieMonster> cookie_monster =
-      profile()->GetCookieMonster();
+  GURL origin("https://accounts.google.com");
+  scoped_refptr<net::CookieMonster> cookie_monster = GetCookieMonster(origin);
   net::CookieOptions options;
   options.set_include_httponly();
   cookie_monster->SetCookieWithOptionsAsync(
-        GURL("https://accounts.google.com"),
-        "LSID=1234; secure; httponly", options,
-        net::CookieMonster::SetCookiesCallback());
+      origin,
+      "LSID=1234; secure; httponly", options,
+      net::CookieMonster::SetCookiesCallback());
 
   // Since the password is empty, will verify the gaia cookies first.
   manager_->StartSignInWithCredentials(
       "0",
       "user@gmail.com",
       std::string(),
       SigninManager::OAuthTokenFetchedCallback());
 
   base::RunLoop().RunUntilIdle();
 
@@ skipping 17 matching lines @@
   // Since the test cookie store is empty, verification should fail and throws
   // a login error.
   ExpectSignInWithCredentialsFail(false /* requestSent */);
 }
 
 TEST_F(SigninManagerTest, SignInWithCredentialsEmptyPasswordInValidCookie) {
   manager_->Initialize(profile(), NULL);
   EXPECT_TRUE(manager_->GetAuthenticatedUsername().empty());
 
   // Set an invalid LSID cookie in the test cookie store.
-  scoped_refptr<net::CookieMonster> cookie_monster =
-      profile()->GetCookieMonster();
+  GURL origin("https://accounts.google.com");
+  scoped_refptr<net::CookieMonster> cookie_monster = GetCookieMonster(origin);
   net::CookieOptions options;
   options.set_include_httponly();
   cookie_monster->SetCookieWithOptionsAsync(
-        GURL("https://accounts.google.com"),
-        "LSID=1234; domain=google.com; secure; httponly", options,
-        net::CookieMonster::SetCookiesCallback());
+      origin,
+      "LSID=1234; domain=google.com; secure; httponly", options,
+      net::CookieMonster::SetCookiesCallback());
 
   // Since the password is empty, must verify the gaia cookies first.
   manager_->StartSignInWithCredentials(
       "0",
       "user@gmail.com",
       std::string(),
       SigninManager::OAuthTokenFetchedCallback());
 
   base::RunLoop().RunUntilIdle();
 
@@ skipping 268 matching lines @@
   EXPECT_EQ("", manager_->GetAuthenticatedUsername());
   EXPECT_EQ(0u, google_login_success_.size());
 
   manager_->OnExternalSigninCompleted("external@example.com");
   EXPECT_EQ(1u, google_login_success_.size());
   EXPECT_EQ(0u, google_login_failure_.size());
   EXPECT_EQ("external@example.com",
             profile()->GetPrefs()->GetString(prefs::kGoogleServicesUsername));
   EXPECT_EQ("external@example.com", manager_->GetAuthenticatedUsername());
 }