| 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 "base/base_paths.h" | 5 #include "base/base_paths.h" |
| 6 #include "base/files/file_util.h" | 6 #include "base/files/file_util.h" |
| 7 #include "base/path_service.h" | 7 #include "base/path_service.h" |
| 8 #include "chrome/browser/extensions/component_loader.h" | 8 #include "chrome/browser/extensions/component_loader.h" |
| 9 #include "chrome/browser/extensions/extension_apitest.h" | 9 #include "chrome/browser/extensions/extension_apitest.h" |
| 10 #include "chrome/browser/extensions/extension_service.h" | 10 #include "chrome/browser/extensions/extension_service.h" |
| (...skipping 20 matching lines...) Expand all Loading... |
| 31 ExtensionApiTest::SetUpOnMainThread(); | 31 ExtensionApiTest::SetUpOnMainThread(); |
| 32 ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady()); | 32 ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady()); |
| 33 } | 33 } |
| 34 | 34 |
| 35 | 35 |
| 36 virtual void TearDownOnMainThread() override { | 36 virtual void TearDownOnMainThread() override { |
| 37 ASSERT_TRUE(embedded_test_server()->ShutdownAndWaitUntilComplete()); | 37 ASSERT_TRUE(embedded_test_server()->ShutdownAndWaitUntilComplete()); |
| 38 ExtensionApiTest::TearDownOnMainThread(); | 38 ExtensionApiTest::TearDownOnMainThread(); |
| 39 } | 39 } |
| 40 | 40 |
| 41 void RunTestsInFile(std::string filename, | 41 void RunTestsInFile(std::string filename, bool requiresPlugin) { |
| 42 std::string pdf_filename) { | |
| 43 base::FilePath pdf_plugin_src; | 42 base::FilePath pdf_plugin_src; |
| 44 PathService::Get(base::DIR_SOURCE_ROOT, &pdf_plugin_src); | 43 PathService::Get(base::DIR_SOURCE_ROOT, &pdf_plugin_src); |
| 45 pdf_plugin_src = pdf_plugin_src.AppendASCII("pdf"); | 44 pdf_plugin_src = pdf_plugin_src.AppendASCII("pdf"); |
| 46 if (!base::DirectoryExists(pdf_plugin_src)) { | 45 if (requiresPlugin && !base::DirectoryExists(pdf_plugin_src)) { |
| 47 LOG(WARNING) << "Not running " << filename << | 46 LOG(WARNING) << "Not running " << filename << |
| 48 " because it requires the PDF plugin which is not available."; | 47 " because it requires the PDF plugin which is not available."; |
| 49 return; | 48 return; |
| 50 } | 49 } |
| 51 ExtensionService* service = extensions::ExtensionSystem::Get( | 50 ExtensionService* service = extensions::ExtensionSystem::Get( |
| 52 profile())->extension_service(); | 51 profile())->extension_service(); |
| 53 service->component_loader()->Add(IDR_PDF_MANIFEST, | 52 service->component_loader()->Add(IDR_PDF_MANIFEST, |
| 54 base::FilePath(FILE_PATH_LITERAL("pdf"))); | 53 base::FilePath(FILE_PATH_LITERAL("pdf"))); |
| 55 const extensions::Extension* extension = | 54 const extensions::Extension* extension = |
| 56 extensions::ExtensionRegistry::Get(profile()) | 55 extensions::ExtensionRegistry::Get(profile()) |
| 57 ->enabled_extensions() | 56 ->enabled_extensions() |
| 58 .GetByID("mhjfbmdgcfjbbpaeojofohoefgiehjai"); | 57 .GetByID("mhjfbmdgcfjbbpaeojofohoefgiehjai"); |
| 59 ASSERT_TRUE(extension); | 58 ASSERT_TRUE(extension); |
| 60 ASSERT_TRUE(MimeTypesHandler::GetHandler( | 59 ASSERT_TRUE(MimeTypesHandler::GetHandler( |
| 61 extension)->CanHandleMIMEType("application/pdf")); | 60 extension)->CanHandleMIMEType("application/pdf")); |
| 62 | 61 |
| 63 extensions::ResultCatcher catcher; | 62 extensions::ResultCatcher catcher; |
| 64 | 63 |
| 65 GURL url(embedded_test_server()->GetURL("/pdf/" + pdf_filename)); | 64 GURL url(embedded_test_server()->GetURL("/pdf/test.pdf")); |
| 66 GURL extension_url( | 65 GURL extension_url( |
| 67 "chrome-extension://mhjfbmdgcfjbbpaeojofohoefgiehjai/index.html?" + | 66 "chrome-extension://mhjfbmdgcfjbbpaeojofohoefgiehjai/index.html?" + |
| 68 url.spec()); | 67 url.spec()); |
| 69 ui_test_utils::NavigateToURL(browser(), extension_url); | 68 ui_test_utils::NavigateToURL(browser(), extension_url); |
| 70 content::WebContents* contents = | 69 content::WebContents* contents = |
| 71 browser()->tab_strip_model()->GetActiveWebContents(); | 70 browser()->tab_strip_model()->GetActiveWebContents(); |
| 72 content::WaitForLoadStop(contents); | 71 content::WaitForLoadStop(contents); |
| 73 | 72 |
| 74 base::FilePath test_data_dir; | 73 base::FilePath test_data_dir; |
| 75 PathService::Get(base::DIR_SOURCE_ROOT, &test_data_dir); | 74 PathService::Get(base::DIR_SOURCE_ROOT, &test_data_dir); |
| 76 test_data_dir = test_data_dir.Append( | 75 test_data_dir = test_data_dir.Append( |
| 77 FILE_PATH_LITERAL("chrome/test/data/pdf")); | 76 FILE_PATH_LITERAL("chrome/test/data/pdf")); |
| 78 test_data_dir = test_data_dir.AppendASCII(filename); | 77 test_data_dir = test_data_dir.AppendASCII(filename); |
| 79 | 78 |
| 80 std::string test_js; | 79 std::string test_js; |
| 81 ASSERT_TRUE(base::ReadFileToString(test_data_dir, &test_js)); | 80 ASSERT_TRUE(base::ReadFileToString(test_data_dir, &test_js)); |
| 82 ASSERT_TRUE(content::ExecuteScript(contents, test_js)); | 81 ASSERT_TRUE(content::ExecuteScript(contents, test_js)); |
| 83 | 82 |
| 84 if (!catcher.GetNextResult()) | 83 if (!catcher.GetNextResult()) |
| 85 FAIL() << catcher.message(); | 84 FAIL() << catcher.message(); |
| 86 } | 85 } |
| 87 }; | 86 }; |
| 88 | 87 |
| 89 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, Basic) { | 88 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, Basic) { |
| 90 RunTestsInFile("basic_test.js", "test.pdf"); | 89 RunTestsInFile("basic_test.js", false); |
| 91 } | 90 } |
| 92 | 91 |
| 93 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, BasicPlugin) { | 92 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, BasicPlugin) { |
| 94 RunTestsInFile("basic_plugin_test.js", "test.pdf"); | 93 RunTestsInFile("basic_plugin_test.js", true); |
| 95 } | 94 } |
| 96 | 95 |
| 97 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, Viewport) { | 96 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, Viewport) { |
| 98 RunTestsInFile("viewport_test.js", "test.pdf"); | 97 RunTestsInFile("viewport_test.js", false); |
| 99 } | 98 } |
| 100 | |
| 101 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, Bookmark) { | |
| 102 RunTestsInFile("bookmarks_test.js", "test-bookmarks.pdf"); | |
| 103 } | |
| OLD | NEW |