OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <string.h> | 5 #include <string.h> |
6 | 6 |
7 #include "base/command_line.h" | 7 #include "base/command_line.h" |
8 #include "base/guid.h" | 8 #include "base/guid.h" |
9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
10 #include "base/strings/utf_string_conversions.h" | 10 #include "base/strings/utf_string_conversions.h" |
11 #include "chrome/browser/dom_distiller/dom_distiller_service_factory.h" | 11 #include "chrome/browser/dom_distiller/dom_distiller_service_factory.h" |
12 #include "chrome/browser/profiles/profile.h" | 12 #include "chrome/browser/profiles/profile.h" |
13 #include "chrome/browser/ui/browser.h" | 13 #include "chrome/browser/ui/browser.h" |
14 #include "chrome/browser/ui/tabs/tab_strip_model.h" | 14 #include "chrome/browser/ui/tabs/tab_strip_model.h" |
15 #include "chrome/common/chrome_isolated_world_ids.h" | |
16 #include "chrome/common/chrome_switches.h" | |
17 #include "chrome/test/base/in_process_browser_test.h" | 15 #include "chrome/test/base/in_process_browser_test.h" |
18 #include "chrome/test/base/ui_test_utils.h" | 16 #include "chrome/test/base/ui_test_utils.h" |
19 #include "components/dom_distiller/content/browser/distiller_javascript_utils.h" | 17 #include "components/dom_distiller/content/browser/distiller_javascript_utils.h" |
20 #include "components/dom_distiller/content/browser/dom_distiller_viewer_source.h
" | 18 #include "components/dom_distiller/content/browser/dom_distiller_viewer_source.h
" |
21 #include "components/dom_distiller/core/article_entry.h" | 19 #include "components/dom_distiller/core/article_entry.h" |
22 #include "components/dom_distiller/core/distilled_page_prefs.h" | 20 #include "components/dom_distiller/core/distilled_page_prefs.h" |
23 #include "components/dom_distiller/core/distiller.h" | 21 #include "components/dom_distiller/core/distiller.h" |
24 #include "components/dom_distiller/core/dom_distiller_service.h" | 22 #include "components/dom_distiller/core/dom_distiller_service.h" |
25 #include "components/dom_distiller/core/dom_distiller_store.h" | 23 #include "components/dom_distiller/core/dom_distiller_store.h" |
26 #include "components/dom_distiller/core/dom_distiller_switches.h" | 24 #include "components/dom_distiller/core/dom_distiller_switches.h" |
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
62 const char kGetContent[] = | 60 const char kGetContent[] = |
63 "window.domAutomationController.send(" | 61 "window.domAutomationController.send(" |
64 "document.getElementById('content').innerHTML)"; | 62 "document.getElementById('content').innerHTML)"; |
65 | 63 |
66 const char kGetBodyClass[] = | 64 const char kGetBodyClass[] = |
67 "window.domAutomationController.send(" | 65 "window.domAutomationController.send(" |
68 "document.body.className)"; | 66 "document.body.className)"; |
69 | 67 |
70 const unsigned kDarkToolbarThemeColor = 0xFF1A1A1A; | 68 const unsigned kDarkToolbarThemeColor = 0xFF1A1A1A; |
71 | 69 |
72 const char kTestDistillerObject[] = | |
73 "window.domAutomationController.send(" | |
74 "typeof distiller == 'object')"; | |
75 | |
76 void AddEntry(const ArticleEntry& e, FakeDB<ArticleEntry>::EntryMap* map) { | 70 void AddEntry(const ArticleEntry& e, FakeDB<ArticleEntry>::EntryMap* map) { |
77 (*map)[e.entry_id()] = e; | 71 (*map)[e.entry_id()] = e; |
78 } | 72 } |
79 | 73 |
80 ArticleEntry CreateEntry(std::string entry_id, std::string page_url) { | 74 ArticleEntry CreateEntry(std::string entry_id, std::string page_url) { |
81 ArticleEntry entry; | 75 ArticleEntry entry; |
82 entry.set_entry_id(entry_id); | 76 entry.set_entry_id(entry_id); |
83 if (!page_url.empty()) { | 77 if (!page_url.empty()) { |
84 ArticleEntryPage* page = entry.add_pages(); | 78 ArticleEntryPage* page = entry.add_pages(); |
85 page->set_url(page_url); | 79 page->set_url(page_url); |
(...skipping 233 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
319 distiller->RunDistillerUpdateCallback(update); | 313 distiller->RunDistillerUpdateCallback(update); |
320 | 314 |
321 content::WaitForLoadStop(contents); | 315 content::WaitForLoadStop(contents); |
322 | 316 |
323 EXPECT_TRUE(content::ExecuteScriptAndExtractString( | 317 EXPECT_TRUE(content::ExecuteScriptAndExtractString( |
324 contents, kGetContent , &result)); | 318 contents, kGetContent , &result)); |
325 EXPECT_THAT(result, HasSubstr("content")); | 319 EXPECT_THAT(result, HasSubstr("content")); |
326 } | 320 } |
327 | 321 |
328 IN_PROC_BROWSER_TEST_F(DomDistillerViewerSourceBrowserTest, | 322 IN_PROC_BROWSER_TEST_F(DomDistillerViewerSourceBrowserTest, |
329 DistillerJavaScriptExposed) { | 323 MultiPageArticle) { |
330 // Navigate to a distiller URL. | |
331 GURL url(std::string(kDomDistillerScheme) + "://url"); | |
332 chrome::NavigateParams params(browser(), url, ui::PAGE_TRANSITION_TYPED); | |
333 chrome::Navigate(¶ms); | |
334 | |
335 content::WebContents* contents = | |
336 browser()->tab_strip_model()->GetActiveWebContents(); | |
337 | |
338 // Wait for the page load to complete (this will be a distiller error page). | |
339 content::WaitForLoadStop(contents); | |
340 | |
341 bool result; | |
342 // Execute in isolated world; where all distiller scripts are run. | |
343 EXPECT_TRUE(content::ExecuteScriptInIsolatedWorldAndExtractBool( | |
344 contents, chrome::ISOLATED_WORLD_ID_CHROME_INTERNAL, kTestDistillerObject, | |
345 &result)); | |
346 EXPECT_TRUE(result); | |
347 } | |
348 | |
349 IN_PROC_BROWSER_TEST_F(DomDistillerViewerSourceBrowserTest, | |
350 DistillerJavaScriptNotInMainWorld) { | |
351 // Navigate to a distiller URL. | |
352 GURL url(std::string(kDomDistillerScheme) + "://url"); | |
353 chrome::NavigateParams params(browser(), url, ui::PAGE_TRANSITION_TYPED); | |
354 chrome::Navigate(¶ms); | |
355 | |
356 content::WebContents* contents = | |
357 browser()->tab_strip_model()->GetActiveWebContents(); | |
358 | |
359 // Wait for the page load to complete (this will be a distiller error page). | |
360 content::WaitForLoadStop(contents); | |
361 | |
362 bool result; | |
363 // Execute in main world, the distiller object should not be here. | |
364 EXPECT_TRUE(content::ExecuteScriptAndExtractBool( | |
365 contents, kTestDistillerObject, &result)); | |
366 EXPECT_FALSE(result); | |
367 } | |
368 | |
369 IN_PROC_BROWSER_TEST_F(DomDistillerViewerSourceBrowserTest, | |
370 DistillerJavaScriptNotExposed) { | |
371 // Navigate to a non-distiller URL. | |
372 GURL url("http://url"); | |
373 chrome::NavigateParams params(browser(), url, ui::PAGE_TRANSITION_TYPED); | |
374 chrome::Navigate(¶ms); | |
375 | |
376 content::WebContents* contents = | |
377 browser()->tab_strip_model()->GetActiveWebContents(); | |
378 | |
379 // Wait for the page load to complete. | |
380 content::WaitForLoadStop(contents); | |
381 | |
382 bool result; | |
383 EXPECT_TRUE(content::ExecuteScriptAndExtractBool( | |
384 contents, kTestDistillerObject, &result)); | |
385 EXPECT_FALSE(result); | |
386 } | |
387 | |
388 IN_PROC_BROWSER_TEST_F(DomDistillerViewerSourceBrowserTest, MultiPageArticle) { | |
389 expect_distillation_ = false; | 324 expect_distillation_ = false; |
390 expect_distiller_page_ = true; | 325 expect_distiller_page_ = true; |
391 dom_distiller::DomDistillerServiceFactory::GetInstance() | 326 dom_distiller::DomDistillerServiceFactory::GetInstance() |
392 ->SetTestingFactoryAndUse(browser()->profile(), &Build); | 327 ->SetTestingFactoryAndUse(browser()->profile(), &Build); |
393 | 328 |
394 scoped_refptr<content::MessageLoopRunner> distillation_done_runner = | 329 scoped_refptr<content::MessageLoopRunner> distillation_done_runner = |
395 new content::MessageLoopRunner; | 330 new content::MessageLoopRunner; |
396 | 331 |
397 FakeDistiller* distiller = new FakeDistiller( | 332 FakeDistiller* distiller = new FakeDistiller( |
398 false, | 333 false, |
(...skipping 111 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
510 EXPECT_EQ(kDarkToolbarThemeColor, contents->GetThemeColor()); | 445 EXPECT_EQ(kDarkToolbarThemeColor, contents->GetThemeColor()); |
511 | 446 |
512 distilled_page_prefs->SetFontFamily(DistilledPagePrefs::SERIF); | 447 distilled_page_prefs->SetFontFamily(DistilledPagePrefs::SERIF); |
513 base::RunLoop().RunUntilIdle(); | 448 base::RunLoop().RunUntilIdle(); |
514 EXPECT_TRUE( | 449 EXPECT_TRUE( |
515 content::ExecuteScriptAndExtractString(contents, kGetBodyClass, &result)); | 450 content::ExecuteScriptAndExtractString(contents, kGetBodyClass, &result)); |
516 EXPECT_EQ("dark serif", result); | 451 EXPECT_EQ("dark serif", result); |
517 } | 452 } |
518 | 453 |
519 } // namespace dom_distiller | 454 } // namespace dom_distiller |
OLD | NEW |