Added ability to set ChromeDownloadManagerDelegate for testing.

chrome/browser/profiles/off_the_record_profile_impl.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 "chrome/browser/profiles/off_the_record_profile_impl.h"
 
 #include "base/command_line.h"
 #include "base/compiler_specific.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
@@ skipping 308 matching lines @@
 PrefService* OffTheRecordProfileImpl::GetOffTheRecordPrefs() {
   return prefs_;
 }
 
 TemplateURLFetcher* OffTheRecordProfileImpl::GetTemplateURLFetcher() {
   return profile_->GetTemplateURLFetcher();
 }
 
 DownloadManager* OffTheRecordProfileImpl::GetDownloadManager() {
   if (!download_manager_.get()) {
-    download_manager_delegate_ = new ChromeDownloadManagerDelegate(this);
+    // In case the delegate has already been set by
+    // SetDownloadManagerDelegate.
+    if (!download_manager_delegate_.get())
+      download_manager_delegate_ = new ChromeDownloadManagerDelegate(this);
     scoped_refptr<DownloadManager> dlm(
         new DownloadManager(download_manager_delegate_,
                             g_browser_process->download_status_updater()));
     dlm->Init(this);
     download_manager_delegate_->SetDownloadManager(dlm);
     download_manager_.swap(dlm);
   }
   return download_manager_.get();
 }
 
@@ skipping 268 matching lines @@
                      const NotificationDetails& details) {
   if (type == content::NOTIFICATION_ZOOM_LEVEL_CHANGED) {
     const std::string& host = *(Details<const std::string>(details).ptr());
     if (!host.empty()) {
       double level = profile_->GetHostZoomMap()->GetZoomLevel(host);
       GetHostZoomMap()->SetZoomLevel(host, level);
     }
   }
 }
 
+void OffTheRecordProfileImpl::SetDownloadManagerDelegate(
+    ChromeDownloadManagerDelegate* delegate) {
+  download_manager_delegate_ = delegate;
+}
+
 void OffTheRecordProfileImpl::CreateQuotaManagerAndClients() {
   if (quota_manager_.get()) {
     DCHECK(file_system_context_.get());
     DCHECK(db_tracker_.get());
     DCHECK(webkit_context_.get());
     return;
   }
 
   // All of the clients have to be created and registered with the
   // QuotaManager prior to the QuotaManger being used. So we do them
@@ skipping 58 matching lines @@
 
 Profile* Profile::CreateOffTheRecordProfile() {
 #if defined(OS_CHROMEOS)
   if (Profile::IsGuestSession())
     return new GuestSessionProfile(this);
 #endif
   OffTheRecordProfileImpl* profile = new OffTheRecordProfileImpl(this);
   profile->Init();
   return profile;
 }