Adds unit tests for offline_pages::PrerenderAdapter calling PrerenderManager.

chrome/browser/android/offline_pages/prerender_adapter.cc

 // Copyright 2016 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/android/offline_pages/prerender_adapter.h"
 
 #include "chrome/browser/prerender/prerender_manager.h"
 #include "chrome/browser/prerender/prerender_manager_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/web_contents.h"
 #include "ui/gfx/geometry/size.h"
 
 namespace offline_pages {
 
 PrerenderAdapter::Observer::Observer() {}
 PrerenderAdapter::Observer::~Observer() {}
 
 PrerenderAdapter::PrerenderAdapter(PrerenderAdapter::Observer* observer)
-    : observer_(observer) {
+    : observer_(observer), test_prerender_manager_(nullptr) {
   DCHECK(observer);
 }
 
 PrerenderAdapter::~PrerenderAdapter() {
   if (IsActive())
     DestroyActive();
 }
 
 bool PrerenderAdapter::CanPrerender() const {
   return prerender::PrerenderManager::ActuallyPrerendering();
 }
 
 bool PrerenderAdapter::StartPrerender(
     content::BrowserContext* browser_context,
     const GURL& url,
     content::SessionStorageNamespace* session_storage_namespace,
     const gfx::Size& size) {
   DCHECK(!IsActive());
   DCHECK(CanPrerender());
 
   Profile* profile = Profile::FromBrowserContext(browser_context);
   prerender::PrerenderManager* manager =
       prerender::PrerenderManagerFactory::GetForProfile(profile);
+  if (test_prerender_manager_) {
+    DVLOG(1) << "Using test PrerenderManager";
+    manager = test_prerender_manager_;
+  }
   DCHECK(manager);
 
   // Start prerendering the url and capture the handle for the prerendering.
   active_handle_.reset(
       manager->AddPrerenderForOffline(url, session_storage_namespace, size));
   if (!active_handle_)
     return false;
 
   active_handle_->SetObserver(this);
   return true;
@@ skipping 39 matching lines @@
     prerender::PrerenderHandle* handle) {
   DCHECK(active_handle_.get() == handle);
   observer_->OnPrerenderDomContentLoaded();
 }
 
 void PrerenderAdapter::OnPrerenderStop(prerender::PrerenderHandle* handle) {
   DCHECK(active_handle_.get() == handle);
   observer_->OnPrerenderStop();
 }
 
+void PrerenderAdapter::SetPrerenderManagerForTest(
+    prerender::PrerenderManager* test_prerender_manager) {
+  test_prerender_manager_ = test_prerender_manager;
+}
+
 }  // namespace offline_pages