Modifying prefetch to account for multi-profile.

chrome/browser/profiles/profile_manager_unittest.cc

 // Copyright (c) 2011 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 <string>
 
 #include "base/command_line.h"
 #include "base/message_loop.h"
 #include "base/path_service.h"
 #include "base/scoped_temp_dir.h"
 #include "base/system_monitor/system_monitor.h"
 #include "base/values.h"
 #include "chrome/browser/extensions/extension_event_router_forwarder.h"
 #include "chrome/browser/io_thread.h"
+#include "chrome/browser/net/predictor.h"
 #include "chrome/browser/prefs/browser_prefs.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_browser_process_test.h"
 #include "chrome/test/base/testing_pref_service.h"
 #include "content/browser/browser_thread.h"
 #include "content/common/notification_service.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 // This global variable is used to check that value returned to different
 // observers is the same.
 Profile* g_created_profile;
 
 }  // namespace
 
+class MockPredictor : public chrome_browser_net::Predictor {
+ public:
+  MOCK_METHOD1(InitNetworkPredictor, void(PrefService* prefs));
+  MOCK_METHOD2(AnticipateOmniboxUrl, void(const GURL&, bool));
+  MOCK_METHOD1(PreconnectUrlAndSubresources, void(const GURL&));
+  MOCK_METHOD1(PredictFrameSubresources, void(const GURL&));
+  MOCK_METHOD1(ShutdownOnUIThread, void(PrefService* prefs));
+  MOCK_METHOD0(DiscardAllResults, void());
+  MOCK_METHOD2(ResolveList,
+               void(const chrome_common_net::UrlList&,
+                    chrome_browser_net::UrlInfo::ResolutionMotivation));
+  MOCK_METHOD2(Resolve,
+               void(const chrome_common_net::UrlList&,
+                    chrome_browser_net::UrlInfo::ResolutionMotivation));
+  MOCK_METHOD2(LearnFromNavigation, void(const GURL&, const GURL&));
+  MOCK_METHOD1(GetHtmlReferrerLists, void(std::string*));
+  MOCK_METHOD1(GetHtmlInfo, void(std::string*));
+  MOCK_METHOD0(TrimReferrersNow, void());
+  MOCK_METHOD1(SerializeReferrers, void(base::ListValue*));
+  MOCK_METHOD1(DeserializeReferrers, void(const base::ListValue&));
+  MOCK_METHOD1(DeserializeReferrersThenDelete, void(base::ListValue*));
+  MOCK_METHOD0(DiscardInitialNavigationHistory, void());
+  MOCK_METHOD3(FinalizeInitializationOnIOThread,
+               void(const std::vector<GURL>&, base::ListValue*, IOThread*));
+  MOCK_METHOD1(LearnAboutInitialNavigation, void(const GURL&));
+  MOCK_METHOD1(DnsPrefetchList, void(const chrome_common_net::NameList&));
+  MOCK_METHOD2(DnsPrefetchMotivatedList,
+               void(const chrome_common_net::UrlList&,
+                    chrome_browser_net::UrlInfo::ResolutionMotivation));
+  MOCK_METHOD1(SaveStateForNextStartupAndTrim, void(PrefService* prefs));
+  MOCK_METHOD3(SaveDnsPrefetchStateForNextStartupAndTrim,
+               void(base::ListValue*, base::ListValue*, base::WaitableEvent*));
+  MOCK_METHOD1(EnablePredictor, void(bool enable));
+  MOCK_METHOD1(EnablePredictorOnIOThread, void(bool enable));
+};
+
 class ProfileManagerTest : public TestingBrowserProcessTest {
  protected:
   ProfileManagerTest()
       : local_state_(testing_browser_process_.get()),
         extension_event_router_forwarder_(new ExtensionEventRouterForwarder),
         ui_thread_(BrowserThread::UI, &message_loop_),
         db_thread_(BrowserThread::DB, &message_loop_),
         file_thread_(BrowserThread::FILE, &message_loop_),
         io_thread_(local_state_.Get(), NULL, extension_event_router_forwarder_),
         profile_manager_(new ProfileManagerWithoutInit) {
@@ skipping 28 matching lines @@
   MessageLoopForUI message_loop_;
   BrowserThread ui_thread_;
   BrowserThread db_thread_;
   BrowserThread file_thread_;
   IOThread io_thread_;
 
   scoped_ptr<base::SystemMonitor> system_monitor_dummy_;
 
   // Also will test profile deletion.
   scoped_ptr<ProfileManager> profile_manager_;
+  scoped_ptr<MockPredictor> mock_predictor_;
 };
 
 TEST_F(ProfileManagerTest, GetProfile) {
   FilePath dest_path = temp_dir_.path();
   dest_path = dest_path.Append(FILE_PATH_LITERAL("New Profile"));
 
   Profile* profile;
 
   // Successfully create a profile.
   profile = profile_manager_->GetProfile(dest_path);
@@ skipping 152 matching lines @@
 
   MockObserver mock_observer;
   EXPECT_CALL(mock_observer, OnProfileCreated(
       testing::NotNull(), NotFail())).Times(testing::AtLeast(3));
 
   profile_manager_->CreateProfileAsync(dest_path1, &mock_observer);
   profile_manager_->CreateProfileAsync(dest_path2, &mock_observer);
 
   message_loop_.RunAllPending();
 }