PrerenderingLoader initial integration with PrerenderManager/PrerenderHandle and make it unit-testa…

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) {
+  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);
+  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;
+}
+
+content::WebContents* PrerenderAdapter::GetWebContents() const {
+  DCHECK(IsActive());
+  DCHECK(active_handle_->contents());
+  // Note: the prerender stack maintains ownership of these contents
+  // and PrerenderingLoader::StopLoading() must be called to report
+  // the Loader is done with the contents.
+  return active_handle_->contents()->prerender_contents();
+}
+
+prerender::FinalStatus PrerenderAdapter::GetFinalStatus() const {
+  DCHECK(IsActive());
+  DCHECK(active_handle_->contents());
+  return active_handle_->contents()->final_status();
+}
+
+bool PrerenderAdapter::IsActive() const {
+  return active_handle_.get();
+}
+
+void PrerenderAdapter::DestroyActive() {
+  DCHECK(IsActive());
+  active_handle_->OnCancel();
+  active_handle_.reset(nullptr);
+}
+
+void PrerenderAdapter::OnPrerenderStart(prerender::PrerenderHandle* handle) {
+  DCHECK(active_handle_.get() == handle);
+  observer_->OnPrerenderStart();
+}
+
+void PrerenderAdapter::OnPrerenderStopLoading(
+    prerender::PrerenderHandle* handle) {
+  DCHECK(active_handle_.get() == handle);
+  observer_->OnPrerenderDomContentLoaded();
+}
+
+void PrerenderAdapter::OnPrerenderDomContentLoaded(
+    prerender::PrerenderHandle* handle) {
+  DCHECK(active_handle_.get() == handle);
+  observer_->OnPrerenderDomContentLoaded();
+}
+
+void PrerenderAdapter::OnPrerenderStop(prerender::PrerenderHandle* handle) {
+  DCHECK(active_handle_.get() == handle);
+  observer_->OnPrerenderStop();
+}
+
+}  // namespace offline_pages