Support javascript gtests in an extension background page.

chrome/test/base/extension_js_browser_test.cc

Index: chrome/test/base/extension_js_browser_test.cc
diff --git a/chrome/test/base/extension_js_browser_test.cc b/chrome/test/base/extension_js_browser_test.cc
new file mode 100644
index 0000000000000000000000000000000000000000..763574766212b973b885c293991a878d47c3819c
--- /dev/null
+++ b/chrome/test/base/extension_js_browser_test.cc
@@ -0,0 +1,59 @@
+// 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 "chrome/test/base/extension_js_browser_test.h"
+
+#include "base/callback.h"
+#include "base/json/json_reader.h"
+#include "base/strings/string_util.h"
+#include "base/strings/utf_string_conversions.h"
+#include "base/values.h"
+#include "chrome/browser/extensions/browsertest_util.h"
+#include "chrome/browser/profiles/profile.h"
+#include "content/public/browser/web_contents.h"
+#include "content/public/test/browser_test_utils.h"
+
+void ExtensionJSBrowserTest::WaitForExtension(const char* extension_id,
+                                              const base::Closure& load_cb) {
+  load_waiter_.reset(new ExtensionLoadWaiterOneShot());
+  load_waiter_->WaitForExtension(extension_id, load_cb);
+}
+
+bool ExtensionJSBrowserTest::RunJavascriptTestF(bool is_async,
+                                                const std::string& test_fixture,
+                                                const std::string& test_name) {
+  CHECK(load_waiter_->browser_context());

[Paweł Hajdan Jr., 2014/06/13 13:45:23]

Please don't crash the entire test unless really necessary. Why not EXPECT_* and
then return false?

[David Tseng, 2014/06/13 23:23:38]

Done.

+  ConstValueVector args;
+  args.push_back(new base::StringValue(test_fixture));
+  args.push_back(new base::StringValue(test_name));
+  std::vector<base::string16> scripts;
+  if (!libs_loaded_) {
+    BuildJavascriptLibraries(&scripts);
+    libs_loaded_ = true;
+  }
+  scripts.push_back(BuildRunTestJSCall(is_async, "RUN_TEST_F", args));
+
+  base::string16 script_16 = JoinString(scripts, '\n');
+  std::string script = base::UTF16ToUTF8(script_16);
+
+  std::string result =
+      extensions::browsertest_util::ExecuteScriptInBackgroundPage(
+          Profile::FromBrowserContext(load_waiter_->browser_context()),
+          load_waiter_->extension_id(),
+          script);
+
+  base::JSONReader reader;
+  scoped_ptr<base::Value> value_result;
+  value_result.reset(reader.Read(result));
+  CHECK_EQ(base::Value::TYPE_DICTIONARY, value_result->GetType());
+  base::DictionaryValue* dict_value =
+      static_cast<base::DictionaryValue*>(value_result.get());
+  bool test_result;
+  std::string test_result_message;
+  CHECK(dict_value->GetBoolean("result", &test_result));
+  CHECK(dict_value->GetString("message", &test_result_message));
+  if (!test_result_message.empty())
+    ADD_FAILURE() << test_result_message;
+  return test_result;
+}