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/profile.h"
 
 #include <string>
 
 #include "base/command_line.h"
 #include "base/compiler_specific.h"
@@ skipping 298 matching lines @@
   virtual PrefService* GetOffTheRecordPrefs() {
     return prefs_;
   }
 
   virtual TemplateURLFetcher* GetTemplateURLFetcher() {
     return profile_->GetTemplateURLFetcher();
   }
 
   virtual DownloadManager* 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 243 matching lines @@
   }
 
   virtual prerender::PrerenderManager* GetPrerenderManager() {
     // We do not allow prerendering in OTR profiles at this point.
     // TODO(tburkard): Figure out if we want to support this, and how, at some
     // point in the future.
     return NULL;
   }
 
  private:
+  virtual void SetDownloadManagerDelegate(
+      ChromeDownloadManagerDelegate* delegate) {
+    download_manager_delegate_ = delegate;
+  }
+
   void 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 125 matching lines @@
 };
 #endif
 
 Profile* Profile::CreateOffTheRecordProfile() {
 #if defined(OS_CHROMEOS)
   if (Profile::IsGuestSession())
     return new GuestSessionProfile(this);
 #endif
   return new OffTheRecordProfileImpl(this);
 }