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
3 // found in the LICENSE file.
4
5 #include "base/base_paths.h"
6 #include "base/file_util.h"
7 #include "base/path_service.h"
8 #include "chrome/browser/extensions/component_loader.h"
9 #include "chrome/browser/extensions/extension_apitest.h"
10 #include "chrome/browser/extensions/extension_service.h"
11 #include "chrome/browser/ui/tabs/tab_strip_model.h"
12 #include "chrome/common/chrome_switches.h"
13 #include "chrome/common/extensions/manifest_handlers/mime_types_handler.h"
14 #include "chrome/test/base/ui_test_utils.h"
15 #include "content/public/test/browser_test_utils.h"
16 #include "grit/browser_resources.h"
17 #include "net/test/embedded_test_server/embedded_test_server.h"
18
19 class PDFExtensionTest : public ExtensionApiTest {
20 public:
~PDFExtensionTest()21 virtual ~PDFExtensionTest() {}
22
SetUpCommandLine(CommandLine * command_line)23 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
24 ExtensionApiTest::SetUpCommandLine(command_line);
25 command_line->AppendSwitch(switches::kOutOfProcessPdf);
26 }
27
SetUpOnMainThread()28 virtual void SetUpOnMainThread() OVERRIDE {
29 ExtensionApiTest::SetUpOnMainThread();
30 ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady());
31 }
32
33
CleanUpOnMainThread()34 virtual void CleanUpOnMainThread() OVERRIDE {
35 ASSERT_TRUE(embedded_test_server()->ShutdownAndWaitUntilComplete());
36 ExtensionApiTest::CleanUpOnMainThread();
37 }
38
RunTestsInFile(std::string filename,bool requiresPlugin)39 void RunTestsInFile(std::string filename, bool requiresPlugin) {
40 base::FilePath pdf_plugin_src;
41 PathService::Get(base::DIR_SOURCE_ROOT, &pdf_plugin_src);
42 pdf_plugin_src = pdf_plugin_src.AppendASCII("pdf");
43 if (requiresPlugin && !base::DirectoryExists(pdf_plugin_src)) {
44 LOG(WARNING) << "Not running " << filename <<
45 " because it requires the PDF plugin which is not available.";
46 return;
47 }
48 ExtensionService* service = extensions::ExtensionSystem::Get(
49 profile())->extension_service();
50 service->component_loader()->Add(IDR_PDF_MANIFEST,
51 base::FilePath(FILE_PATH_LITERAL("pdf")));
52 const extensions::Extension* extension =
53 service->extensions()->GetByID("mhjfbmdgcfjbbpaeojofohoefgiehjai");
54 ASSERT_TRUE(extension);
55 ASSERT_TRUE(MimeTypesHandler::GetHandler(
56 extension)->CanHandleMIMEType("application/pdf"));
57
58 ResultCatcher catcher;
59
60 GURL url(embedded_test_server()->GetURL("/pdf/test.pdf"));
61 GURL extension_url(
62 "chrome-extension://mhjfbmdgcfjbbpaeojofohoefgiehjai/index.html?" +
63 url.spec());
64 ui_test_utils::NavigateToURL(browser(), extension_url);
65 content::WebContents* contents =
66 browser()->tab_strip_model()->GetActiveWebContents();
67 content::WaitForLoadStop(contents);
68
69 base::FilePath test_data_dir;
70 PathService::Get(base::DIR_SOURCE_ROOT, &test_data_dir);
71 test_data_dir = test_data_dir.Append(
72 FILE_PATH_LITERAL("chrome/test/data/pdf"));
73 test_data_dir = test_data_dir.AppendASCII(filename);
74
75 std::string test_js;
76 ASSERT_TRUE(base::ReadFileToString(test_data_dir, &test_js));
77 ASSERT_TRUE(content::ExecuteScript(contents, test_js));
78
79 if (!catcher.GetNextResult())
80 FAIL() << catcher.message();
81 }
82 };
83
IN_PROC_BROWSER_TEST_F(PDFExtensionTest,Basic)84 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, Basic) {
85 RunTestsInFile("basic_test.js", false);
86 }
87
88 // TODO(raymes): This fails with component builds on linux because the plugin
89 // plugin crashes due to something related to how the plugin DLL is
90 // compiled/linked. crbug.com/386436.
91 #if defined(LINUX) && defined(COMPONENT_BUILD)
92 #define MAYBE_BasicPlugin DISABLED_BasicPlugin
93 #else
94 #define MAYBE_BasicPlugin BasicPlugin
95 #endif
96
IN_PROC_BROWSER_TEST_F(PDFExtensionTest,MAYBE_BasicPlugin)97 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, MAYBE_BasicPlugin) {
98 RunTestsInFile("basic_plugin_test.js", true);
99 }
100
IN_PROC_BROWSER_TEST_F(PDFExtensionTest,Viewport)101 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, Viewport) {
102 RunTestsInFile("viewport_test.js", false);
103 }
104