Home | History | Annotate | Download | only in extensions 
      1 // Copyright (c) 2012 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/strings/stringprintf.h"
      6 #include "chrome/browser/extensions/extension_apitest.h"
      7 #include "chrome/browser/ui/browser.h"
      8 #include "chrome/browser/ui/tabs/tab_strip_model.h"
      9 #include "chrome/test/base/ui_test_utils.h"
     10 #include "content/public/browser/web_contents.h"
     11 #include "content/public/test/browser_test_utils.h"
     12 #include "net/dns/mock_host_resolver.h"
     13 #include "net/test/embedded_test_server/embedded_test_server.h"
     14 #include "url/gurl.h"
     15 
     16 namespace {
     17 
     18 class ClipboardApiTest : public ExtensionApiTest {
     19  public:
     20   bool LoadHostedApp(const std::string& app_name,
     21                      const std::string& launch_page);
     22   bool ExecuteCopyInSelectedTab();
     23   bool ExecutePasteInSelectedTab();
     24   bool ExecuteCommandInIframeInSelectedTab(const char* command);
     25 
     26  private:
     27   bool ExecuteScriptInSelectedTab(const std::string& script);
     28 };
     29 
     30 bool ClipboardApiTest::LoadHostedApp(const std::string& app_name,
     31                                      const std::string& launch_page) {
     32   host_resolver()->AddRule("*", "127.0.0.1");
     33 
     34   if (!StartEmbeddedTestServer()) {
     35     message_ = "Failed to start test server.";
     36     return false;
     37   }
     38 
     39   if (!LoadExtension(test_data_dir_.AppendASCII("clipboard")
     40                                    .AppendASCII(app_name))) {
     41     message_ = "Failed to load hosted app.";
     42     return false;
     43   }
     44 
     45   GURL base_url = embedded_test_server()->GetURL(
     46       "/extensions/api_test/clipboard/");
     47   GURL::Replacements replace_host;
     48   std::string host_str("localhost");  // Must stay in scope with replace_host.
     49   replace_host.SetHostStr(host_str);
     50   base_url = base_url.ReplaceComponents(replace_host);
     51 
     52   std::string launch_page_path =
     53       base::StringPrintf("%s/%s", app_name.c_str(), launch_page.c_str());
     54   ui_test_utils::NavigateToURL(browser(), base_url.Resolve(launch_page_path));
     55 
     56   return true;
     57 }
     58 
     59 bool ClipboardApiTest::ExecuteCopyInSelectedTab() {
     60   const char kScript[] =
     61       "window.domAutomationController.send(document.execCommand('copy'))";
     62   return ExecuteScriptInSelectedTab(kScript);
     63 }
     64 
     65 bool ClipboardApiTest::ExecutePasteInSelectedTab() {
     66   const char kScript[] =
     67       "window.domAutomationController.send(document.execCommand('paste'))";
     68   return ExecuteScriptInSelectedTab(kScript);
     69 }
     70 
     71 bool ClipboardApiTest::ExecuteCommandInIframeInSelectedTab(
     72     const char* command) {
     73   const char kScript[] =
     74       "var ifr = document.createElement('iframe');\n"
     75       "document.body.appendChild(ifr);\n"
     76       "ifr.contentDocument.write('<script>parent.domAutomationController.send("
     77           "document.execCommand(\"%s\"))</script>');";
     78   return ExecuteScriptInSelectedTab(base::StringPrintf(kScript, command));
     79 }
     80 
     81 bool ClipboardApiTest::ExecuteScriptInSelectedTab(const std::string& script) {
     82   bool result;
     83   CHECK(content::ExecuteScriptAndExtractBool(
     84         browser()->tab_strip_model()->GetActiveWebContents(),
     85         script,
     86         &result));
     87   return result;
     88 }
     89 
     90 }  // namespace
     91 
     92 IN_PROC_BROWSER_TEST_F(ClipboardApiTest, Extension) {
     93   ASSERT_TRUE(StartEmbeddedTestServer());
     94   ASSERT_TRUE(RunExtensionTest("clipboard/extension")) << message_;
     95 }
     96 
     97 IN_PROC_BROWSER_TEST_F(ClipboardApiTest, ExtensionNoPermission) {
     98   ASSERT_TRUE(StartEmbeddedTestServer());
     99   ASSERT_TRUE(RunExtensionTest("clipboard/extension_no_permission"))
    100       << message_;
    101 }
    102 
    103 IN_PROC_BROWSER_TEST_F(ClipboardApiTest, HostedApp) {
    104   ASSERT_TRUE(LoadHostedApp("hosted_app", "main.html")) << message_;
    105 
    106   EXPECT_TRUE(ExecuteCopyInSelectedTab()) << message_;
    107   EXPECT_TRUE(ExecutePasteInSelectedTab()) << message_;
    108   EXPECT_TRUE(ExecuteCommandInIframeInSelectedTab("copy")) << message_;
    109   EXPECT_TRUE(ExecuteCommandInIframeInSelectedTab("paste")) << message_;
    110 }
    111 
    112 IN_PROC_BROWSER_TEST_F(ClipboardApiTest, HostedAppNoPermission) {
    113   ASSERT_TRUE(LoadHostedApp("hosted_app_no_permission", "main.html"))
    114       << message_;
    115 
    116   EXPECT_FALSE(ExecuteCopyInSelectedTab()) << message_;
    117   EXPECT_FALSE(ExecutePasteInSelectedTab()) << message_;
    118   EXPECT_FALSE(ExecuteCommandInIframeInSelectedTab("copy")) << message_;
    119   EXPECT_FALSE(ExecuteCommandInIframeInSelectedTab("paste")) << message_;
    120 }
    121 
    122