1 files changed, 124 insertions, 0 deletions
diff --git a/chromium/extensions/renderer/api_test_base.h b/chromium/extensions/renderer/api_test_base.h
new file mode 100644
index 00000000000..8b62049924c
--- /dev/null
+++ b/chromium/extensions/renderer/api_test_base.h
@@ -0,0 +1,124 @@
+// 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.
+
+#ifndef EXTENSIONS_RENDERER_API_TEST_BASE_H_
+#define EXTENSIONS_RENDERER_API_TEST_BASE_H_
+
+#include <map>
+#include <string>
+#include <utility>
+
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
+#include "extensions/renderer/module_system_test.h"
+#include "extensions/renderer/v8_schema_registry.h"
+#include "gin/handle.h"
+#include "gin/modules/module_registry.h"
+#include "gin/object_template_builder.h"
+#include "gin/wrappable.h"
+#include "mojo/public/cpp/bindings/interface_request.h"
+#include "mojo/public/cpp/system/core.h"
+
+namespace extensions {
+
+class V8SchemaRegistry;
+
+// A ServiceProvider that provides access from JS modules to services registered
+// by AddService() calls.
+class TestServiceProvider : public gin::Wrappable<TestServiceProvider> {
+ public:
+  static gin::Handle<TestServiceProvider> Create(v8::Isolate* isolate);
+  ~TestServiceProvider() override;
+
+  gin::ObjectTemplateBuilder GetObjectTemplateBuilder(
+      v8::Isolate* isolate) override;
+
+  template <typename Interface>
+  void AddService(const base::Callback<void(mojo::InterfaceRequest<Interface>)>
+                      service_factory) {
+    service_factories_.insert(std::make_pair(
+        Interface::Name_,
+        base::Bind(ForwardToServiceFactory<Interface>, service_factory)));
+  }
+
+  // Ignore requests for the Interface service.
+  template <typename Interface>
+  void IgnoreServiceRequests() {
+    service_factories_.insert(std::make_pair(
+        Interface::Name_, base::Bind(&TestServiceProvider::IgnoreHandle)));
+  }
+
+  static gin::WrapperInfo kWrapperInfo;
+
+ private:
+  TestServiceProvider();
+
+  mojo::Handle ConnectToService(const std::string& service_name);
+
+  template <typename Interface>
+  static void ForwardToServiceFactory(
+      const base::Callback<void(mojo::InterfaceRequest<Interface>)>
+          service_factory,
+      mojo::ScopedMessagePipeHandle handle) {
+    service_factory.Run(mojo::MakeRequest<Interface>(std::move(handle)));
+  }
+
+  static void IgnoreHandle(mojo::ScopedMessagePipeHandle handle);
+
+  std::map<std::string, base::Callback<void(mojo::ScopedMessagePipeHandle)> >
+      service_factories_;
+};
+
+// An environment for unit testing apps/extensions API custom bindings
+// implemented on Mojo services. This augments a ModuleSystemTestEnvironment
+// with a TestServiceProvider and other modules available in a real extensions
+// environment.
+class ApiTestEnvironment {
+ public:
+  explicit ApiTestEnvironment(ModuleSystemTestEnvironment* environment);
+  ~ApiTestEnvironment();
+  void RunTest(const std::string& file_name, const std::string& test_name);
+  TestServiceProvider* service_provider() { return service_provider_; }
+  ModuleSystemTestEnvironment* env() { return env_; }
+
+ private:
+  void RegisterModules();
+  void InitializeEnvironment();
+  void RunTestInner(const std::string& test_name,
+                    const base::Closure& quit_closure);
+  void RunPromisesAgain();
+
+  ModuleSystemTestEnvironment* env_;
+  TestServiceProvider* service_provider_;
+  scoped_ptr<V8SchemaRegistry> v8_schema_registry_;
+};
+
+// A base class for unit testing apps/extensions API custom bindings implemented
+// on Mojo services. To use:
+// 1. Register test Mojo service implementations on service_provider().
+// 2. Write JS tests in extensions/test/data/test_file.js.
+// 3. Write one C++ test function for each JS test containing
+//    RunTest("test_file.js", "testFunctionName").
+// See extensions/renderer/api_test_base_unittest.cc and
+// extensions/test/data/api_test_base_unittest.js for sample usage.
+class ApiTestBase : public ModuleSystemTest {
+ protected:
+  ApiTestBase();
+  ~ApiTestBase() override;
+  void SetUp() override;
+  void RunTest(const std::string& file_name, const std::string& test_name);
+
+  ApiTestEnvironment* api_test_env() { return test_env_.get(); }
+  TestServiceProvider* service_provider() {
+    return test_env_->service_provider();
+  }
+
+ private:
+  base::MessageLoop message_loop_;
+  scoped_ptr<ApiTestEnvironment> test_env_;
+};
+
+}  // namespace extensions
+
+#endif  // EXTENSIONS_RENDERER_API_TEST_BASE_H_