Add support for distilling current WebContents

chrome/browser/dom_distiller/tab_utils_browsertest.cc

Index: chrome/browser/dom_distiller/tab_utils_browsertest.cc
diff --git a/chrome/browser/dom_distiller/tab_utils_browsertest.cc b/chrome/browser/dom_distiller/tab_utils_browsertest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..37bd4f293b607ccbbc0b0c22f97713961941f7a6
--- /dev/null
+++ b/chrome/browser/dom_distiller/tab_utils_browsertest.cc
@@ -0,0 +1,91 @@
+// Copyright 2014 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.h>
+
+#include "base/command_line.h"
+#include "base/strings/utf_string_conversions.h"
+#include "chrome/browser/dom_distiller/dom_distiller_service_factory.h"
+#include "chrome/browser/dom_distiller/tab_utils.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/tabs/tab_strip_model.h"
+#include "chrome/common/chrome_switches.h"
+#include "chrome/common/url_constants.h"
+#include "chrome/test/base/in_process_browser_test.h"
+#include "chrome/test/base/ui_test_utils.h"
+#include "components/dom_distiller/content/web_contents_main_frame_observer.h"
+#include "components/dom_distiller/core/dom_distiller_service.h"
+#include "components/dom_distiller/core/task_tracker.h"
+#include "components/dom_distiller/core/url_utils.h"
+#include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/web_contents.h"
+#include "content/public/browser/web_contents_observer.h"
+#include "net/test/embedded_test_server/embedded_test_server.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace dom_distiller {
+
+namespace {
+const char* kSimpleArticlePath = "/dom_distiller/simple_article.html";
+}  // namespace
+
+class DomDistillerTabUtilsBrowserTest : public InProcessBrowserTest {
+ public:
+  virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
+    command_line->AppendSwitch(switches::kEnableDomDistiller);
+  }
+};
+
+class WebContentsMainFrameHelper : public content::WebContentsObserver {
+ public:
+  WebContentsMainFrameHelper(content::WebContents* web_contents,
+                             const base::Closure& callback)
+      : callback_(callback) {
+    content::WebContentsObserver::Observe(web_contents);
+  }
+
+  virtual void DidFinishLoad(
+      int64 frame_id,
+      const GURL& validated_url,
+      bool is_main_frame,
+      content::RenderViewHost* render_view_host) OVERRIDE {
+    if (is_main_frame && validated_url.scheme() == chrome::kDomDistillerScheme)
+      callback_.Run();
+  }
+
+ private:
+  base::Closure callback_;
+};
+
+IN_PROC_BROWSER_TEST_F(DomDistillerTabUtilsBrowserTest, TestSwapWebContents) {
+  ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady());
+
+  content::WebContents* initial_web_contents =
+      browser()->tab_strip_model()->GetActiveWebContents();
+  const GURL& article_url = embedded_test_server()->GetURL(kSimpleArticlePath);
+
+  // This blocks until the navigation has completely finished.
+  ui_test_utils::NavigateToURL(browser(), article_url);
+
+  DistillCurrentPageAndView(initial_web_contents);
+
+  // Wait until the new WebContents has fully navigated.
+  content::WebContents* after_web_contents =
+      browser()->tab_strip_model()->GetActiveWebContents();
+  ASSERT_TRUE(after_web_contents != NULL);
+  base::RunLoop new_url_loaded_runner;
+  scoped_ptr<WebContentsMainFrameHelper> distilled_page_loaded(
+      new WebContentsMainFrameHelper(after_web_contents,
+                                     new_url_loaded_runner.QuitClosure()));
+  new_url_loaded_runner.Run();
+
+  // Verify the new URL is showing distilled content in a new WebContents.
+  EXPECT_NE(initial_web_contents, after_web_contents);
+  EXPECT_TRUE(after_web_contents->GetLastCommittedURL().SchemeIs(
+      chrome::kDomDistillerScheme));
+  EXPECT_EQ("Test Page Title",
+            base::UTF16ToUTF8(after_web_contents->GetTitle()));
+}
+
+}  // namespace dom_distiller