Move code duplicated in two tests up into the TestingProfile.

chrome/test/testing_profile.h

-// Copyright (c) 2009 The Chromium Authors. All rights reserved.
+// Copyright (c) 2009-2010 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_TEST_TESTING_PROFILE_H_
 #define CHROME_TEST_TESTING_PROFILE_H_
 
 #include "base/base_paths.h"
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "chrome/browser/bookmarks/bookmark_model.h"
 #include "chrome/browser/browser_prefs.h"
 #include "chrome/browser/browser_theme_provider.h"
 #include "chrome/browser/favicon_service.h"
 #include "chrome/browser/history/history.h"
+#include "chrome/browser/net/url_request_context_getter.h"
 #include "chrome/browser/profile.h"
 #include "chrome/browser/search_engines/template_url_model.h"
 #include "chrome/browser/sessions/session_service.h"
 #include "chrome/common/pref_service.h"
+#include "net/base/cookie_monster.h"
 
 class TestingProfile : public Profile {
  public:
   TestingProfile();
 
   // Creates a new profile by adding |count| to the end of the path. Use this
   // when you need to have more than one TestingProfile running at the same
   // time.
   explicit TestingProfile(int count);
 
@@ skipping 63 matching lines @@
   }
   virtual HistoryService* GetHistoryService(ServiceAccessType access) {
     return history_service_.get();
   }
   virtual HistoryService* GetHistoryServiceWithoutCreating() {
     return history_service_.get();
   }
   void set_has_history_service(bool has_history_service) {
     has_history_service_ = has_history_service;
   }
+  net::CookieMonster* GetCookieMonster() {

[John Grabowski, 2010/01/08 04:01:55]

Add comment with reference to CreateRequestContext() being needed first if you
want this, and that it's not done on-demand because....

+    if (!GetRequestContext())
+      return NULL;
+    return GetRequestContext()->GetCookieStore()->GetCookieMonster();
+  }
   virtual SearchVersusNavigateClassifier* GetSearchVersusNavigateClassifier() {
     return NULL;
   }
   virtual WebDataService* GetWebDataService(ServiceAccessType access) {
     return NULL;
   }
   virtual WebDataService* GetWebDataServiceWithoutCreating() {
     return NULL;
   }
   virtual PasswordStore* GetPasswordStore(ServiceAccessType access) {
@@ skipping 19 matching lines @@
   virtual PersonalDataManager* GetPersonalDataManager() { return NULL; }
   virtual void InitThemes();
   virtual void SetTheme(Extension* extension) {}
   virtual void SetNativeTheme() {}
   virtual void ClearTheme() {}
   virtual Extension* GetTheme() { return NULL; }
   virtual BrowserThemeProvider* GetThemeProvider() {
     InitThemes();
     return theme_provider_.get();
   }
-  virtual URLRequestContextGetter* GetRequestContext() { return NULL; }
+
+  // Returns a testing ContextGetter if one has been created (this can return
+  // NULL). This getter is currently only capable of returning a Context that
+  // helps test the CookieMonster. See implementation comments for more
+  // details.
+  virtual URLRequestContextGetter* GetRequestContext();
+  void CreateRequestContext();
+
   virtual URLRequestContextGetter* GetRequestContextForMedia() { return NULL; }
   virtual URLRequestContextGetter* GetRequestContextForExtensions() {
       return NULL;
   }
+
   virtual net::SSLConfigService* GetSSLConfigService() { return NULL; }
   virtual BlacklistManager* GetBlacklistManager() { return NULL; }
   virtual HostZoomMap* GetHostZoomMap() { return NULL; }
   void set_session_service(SessionService* session_service) {
     session_service_ = session_service;
   }
   virtual SessionService* GetSessionService() { return session_service_.get(); }
   virtual void ShutdownSessionService() {}
   virtual bool HasSessionService() const {
     return (session_service_.get() != NULL);
@@ skipping 66 matching lines @@
 
   scoped_ptr<NTPResourceCache> ntp_resource_cache_;
 
   // The SessionService. Defaults to NULL, but can be set using the setter.
   scoped_refptr<SessionService> session_service_;
 
   // The theme provider. Created lazily by GetThemeProvider()/InitThemes().
   scoped_ptr<BrowserThemeProvider> theme_provider_;
   bool created_theme_provider_;
 
+  // Internally, this is a TestURLRequestContextGetter that creates a dummy
+  // request context. Currently, only the CookieMonster is hooked up.
+  scoped_refptr<URLRequestContextGetter> request_context_;
+
   // Do we have a history service? This defaults to the value of
   // history_service, but can be explicitly set.
   bool has_history_service_;
 
   std::wstring id_;
 
   bool off_the_record_;
 
   // Did the last session exit cleanly? Default is true.
   bool last_session_exited_cleanly_;
@@ skipping 13 matching lines @@
 
   virtual ProfileId GetRuntimeId() {
     return original_profile_->GetRuntimeId();
   }
 
  protected:
   Profile* original_profile_;
 };
 
 #endif  // CHROME_TEST_TESTING_PROFILE_H_