| // 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 |