Allow extension APIs to be called from WebUI.

chrome/browser/extensions/extension_webui_apitest.cc

+// 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 "base/file_util.h"
+#include "base/files/file_path.h"
+#include "base/path_service.h"
+#include "base/values.h"
+#include "chrome/browser/extensions/extension_apitest.h"
+#include "chrome/browser/extensions/extension_test_message_listener.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/tabs/tab_strip_model.h"
+#include "chrome/common/chrome_paths.h"
+#include "chrome/test/base/ui_test_utils.h"
+#include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/web_contents.h"
+#include "content/public/test/browser_test_utils.h"
+#include "extensions/browser/event_router.h"
+#include "extensions/common/api/test.h"
+#include "extensions/common/extension.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace extensions {
+
+namespace OnMessage = core_api::test::OnMessage;
+
+namespace {
+
+void FindFrame(const GURL& url,
+               content::RenderFrameHost** out,
+               content::RenderFrameHost* frame) {
+  if (frame->GetLastCommittedURL() == url) {
+    if (*out != NULL) {
+      ADD_FAILURE() << "Found multiple frames at " << url;
+    }
+    *out = frame;
+  }
+}
+
+// Tests running extension APIs on WebUI.
+class ExtensionWebUITest : public ExtensionApiTest {
+ protected:
+  testing::AssertionResult RunTest(const char* name) {
+    // Tests are located in chrome/test/data/extensions/webui/$(name).
+    base::FilePath path;
+    PathService::Get(chrome::DIR_TEST_DATA, &path);
+    path =
+        path.AppendASCII("extensions").AppendASCII("webui").AppendASCII(name);
+
+    // Read the test.
+    if (!base::PathExists(path))
+      return testing::AssertionFailure() << "Couldn't find " << path.value();
+    std::string script;
+    base::ReadFileToString(path, &script);
+    script = "(function(){'use strict';" + script + "}());";
+
+    // Run the test.
+    bool result = false;
+    content::RenderFrameHost* webui = NavigateToWebUI();
+    if (!webui)
+      return testing::AssertionFailure() << "Failed to navigate to WebUI";
+    CHECK(content::ExecuteScriptAndExtractBool(webui, script, &result));
+    return result ? testing::AssertionSuccess()
+                  : (testing::AssertionFailure() << "Check console output");
+  }
+
+ private:
+  // Navigates the browser to a WebUI page and returns the RenderFrameHost for
+  // that page.
+  content::RenderFrameHost* NavigateToWebUI() {
+    // Use the chrome://extensions page, cos, why not.
+    ui_test_utils::NavigateToURL(browser(), GURL("chrome://extensions/"));
+
+    // In the current design the URL of the chrome://extensions page it's
+    // actually chrome://extensions-frame/ -- and it's important we find it,
+    // because the top-level frame doesn't execute any code, so a script
+    // context is never created, so the bindings are never set up, and
+    // apparently the call to ExecuteScriptAndExtractString doesn't adequately
+    // set them up either.
+    content::RenderFrameHost* frame_host = NULL;
+    browser()->tab_strip_model()->GetActiveWebContents()->ForEachFrame(
+        base::Bind(
+            &FindFrame, GURL("chrome://extensions-frame/"), &frame_host));
+    return frame_host;
+  }
+};
+
+IN_PROC_BROWSER_TEST_F(ExtensionWebUITest, SanityCheckAvailableAPIs) {
+  ASSERT_TRUE(RunTest("sanity_check_available_apis.js"));
+}
+
+// Tests chrome.test.sendMessage, which exercises WebUI making a
+// function call and receiving a response.
+IN_PROC_BROWSER_TEST_F(ExtensionWebUITest, SendMessage) {
+  scoped_ptr<ExtensionTestMessageListener> listener(
+      new ExtensionTestMessageListener("ping", true));
+
+  ASSERT_TRUE(RunTest("send_message.js"));
+
+  ASSERT_TRUE(listener->WaitUntilSatisfied());
+  listener->Reply("pong");
+
+  listener.reset(new ExtensionTestMessageListener(false));
+  ASSERT_TRUE(listener->WaitUntilSatisfied());
+  EXPECT_EQ("true", listener->message());
+}
+
+// Tests chrome.runtime.onMessage, which exercises WebUI registering and
+// receiving an event.
+IN_PROC_BROWSER_TEST_F(ExtensionWebUITest, OnMessage) {
+  ASSERT_TRUE(RunTest("on_message.js"));
+
+  OnMessage::Info info;
+  info.data = "hi";
+  info.last_message = true;
+  EventRouter::Get(profile())->BroadcastEvent(make_scoped_ptr(
+      new Event(OnMessage::kEventName, OnMessage::Create(info))));
+
+  scoped_ptr<ExtensionTestMessageListener> listener(
+      new ExtensionTestMessageListener(false));
+  ASSERT_TRUE(listener->WaitUntilSatisfied());
+  EXPECT_EQ("true", listener->message());
+}
+
+// Tests chrome.runtime.lastError, which exercises WebUI accessing a property
+// on an API which it doesn't actually have access to. A bindings test really.
+IN_PROC_BROWSER_TEST_F(ExtensionWebUITest, RuntimeLastError) {
+  scoped_ptr<ExtensionTestMessageListener> listener(
+      new ExtensionTestMessageListener("ping", true));
+
+  ASSERT_TRUE(RunTest("runtime_last_error.js"));
+
+  ASSERT_TRUE(listener->WaitUntilSatisfied());
+  listener->ReplyWithError("unknown host");
+
+  listener.reset(new ExtensionTestMessageListener(false));
+  ASSERT_TRUE(listener->WaitUntilSatisfied());
+  EXPECT_EQ("true", listener->message());
+}
+
+}  // namespace
+
+}  // namespace extensions