Embed a mojo ApplicationManager in content/browser

content/browser/mojo_shell_browsertest.cc

+// Copyright 2015 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/bind.h"
+#include "base/logging.h"
+#include "base/macros.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/run_loop.h"
+#include "base/thread_task_runner_handle.h"
+#include "base/time/time.h"
+#include "content/public/browser/mojo_app_connection.h"
+#include "content/public/browser/mojo_shell_context.h"
+#include "content/public/common/application_registry.h"
+#include "content/public/common/content_client.h"
+#include "content/public/common/static_mojo_application_loader.h"
+#include "content/public/test/content_browser_test.h"
+#include "content/public/test/test_utils.h"
+#include "content/test/test_content_utility_client.h"
+#include "content/test/test_service.mojom.h"
+#include "mojo/application/public/cpp/application_connection.h"
+#include "mojo/application/public/cpp/application_delegate.h"
+#include "mojo/application/public/cpp/application_impl.h"
+#include "third_party/mojo/src/mojo/public/cpp/bindings/binding.h"
+#include "third_party/mojo/src/mojo/public/cpp/bindings/interface_request.h"
+
+namespace content {
+
+// Simple TestApp which provides a TestService impl. The app terminates itself
+// after its TestService fulfills a single DoSomething call.
+class TestApp : public mojo::ApplicationDelegate,
+                public mojo::InterfaceFactory<TestService>,
+                public TestService {
+ public:
+  TestApp() : service_binding_(this), app_(nullptr) {}
+
+  ~TestApp() override {}
+
+ private:
+  // mojo::ApplicationDelegate:
+  void Initialize(mojo::ApplicationImpl* app) override {
+    app_ = app;
+  }
+
+  bool ConfigureIncomingConnection(
+      mojo::ApplicationConnection* connection) override {
+    connection->AddService<TestService>(this);
+    return true;
+  }
+
+  // mojo::InterfaceFactory<TestService>:
+  void Create(mojo::ApplicationConnection* connection,
+              mojo::InterfaceRequest<TestService> request) override {
+    DCHECK(!service_binding_.is_bound());
+    service_binding_.Bind(request.Pass());
+  }
+
+  // TestService:
+  void DoSomething(const DoSomethingCallback& callback) override {
+    callback.Run();
+    DCHECK(app_);
+    app_->Terminate();
+  }
+
+  mojo::Binding<TestService> service_binding_;
+
+  // Not owned.
+  mojo::ApplicationImpl* app_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestApp);
+};
+
+class MojoShellTestUtilityClient : public TestContentUtilityClient {
+ public:
+  MojoShellTestUtilityClient(
+      const base::Callback<void(ApplicationRegistry*)>& registration_callback)
+      : registration_callback_(registration_callback) {}
+
+  // ContentUtilityClient:
+  void RegisterMojoApplications(ApplicationRegistry* registry) override {
+    registration_callback_.Run(registry);
+  }
+
+ private:
+  base::Callback<void(ApplicationRegistry*)> registration_callback_;
+};
+
+class MojoShellTest : public ContentBrowserTest {
+ public:
+  MojoShellTest()
+      : utility_client_(new MojoShellTestUtilityClient(
+            base::Bind(&MojoShellTest::SetUpUtilityApplications,
+                       base::Unretained(this)))) {
+    MojoShellContext::SetApplicationRegistryInitializerForTesting(
+        base::Bind(&MojoShellTest::SetUpBrowserApplications,
+                   base::Unretained(this)));
+  }
+
+ protected:
+  virtual void SetUpBrowserApplications(ApplicationRegistry* registry) {
+    registry->RegisterStaticAppForURL(
+        GURL("https://test.example.com/in-browser"),
+        base::Bind(&CreateTestApp));
+  }
+
+  virtual void SetUpUtilityApplications(ApplicationRegistry* registry) {
+    registry->RegisterStaticAppForURL(
+        GURL("https://test.example.com/in-utility"),
+        base::Bind(&CreateTestApp));
+  }
+
+  static void ValidateResult(int32_t expected_value,
+                             const base::Closure& callback,
+                             int32_t value) {
+    EXPECT_EQ(expected_value, value);
+    callback.Run();
+  }
+
+ private:
+  void SetUpOnMainThread() override {
+    ContentBrowserTest::SetUpOnMainThread();
+    SetUtilityClientForTesting(utility_client_.get());
+    in_process_utility_thread_helper_.reset(new InProcessUtilityThreadHelper);
+  }
+
+  void TearDownOnMainThread() override {
+    in_process_utility_thread_helper_.reset();
+  }
+
+  static scoped_ptr<mojo::ApplicationDelegate> CreateTestApp() {
+    return scoped_ptr<mojo::ApplicationDelegate>(new TestApp);
+  }
+
+  scoped_ptr<InProcessUtilityThreadHelper> in_process_utility_thread_helper_;
+  scoped_ptr<MojoShellTestUtilityClient> utility_client_;
+
+  DISALLOW_COPY_AND_ASSIGN(MojoShellTest);
+};
+
+IN_PROC_BROWSER_TEST_F(MojoShellTest, TestBrowserConnection) {
+  auto test_app = MojoAppConnection::Create(
+      GURL("https://test.example.com/in-browser"));
+  TestServicePtr test_service;
+  test_app->ConnectToService(&test_service);
+
+  base::RunLoop run_loop;
+  test_service->DoSomething(run_loop.QuitClosure());
+  run_loop.Run();
+}
+
+IN_PROC_BROWSER_TEST_F(MojoShellTest, TestUtilityConnection) {
+  auto test_app = MojoAppConnection::Create(
+      GURL("https://test.example.com/in-utility"));
+  TestServicePtr test_service;
+  test_app->ConnectToService(&test_service);
+
+  base::RunLoop run_loop;
+  test_service->DoSomething(run_loop.QuitClosure());
+  run_loop.Run();
+}
+
+}  // namespace content