Index: content/browser/mojo_shell_browsertest.cc |
diff --git a/content/browser/mojo_shell_browsertest.cc b/content/browser/mojo_shell_browsertest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..737be5c9a5ae8d049166480ef01a2696341ecaf0 |
--- /dev/null |
+++ b/content/browser/mojo_shell_browsertest.cc |
@@ -0,0 +1,162 @@ |
+// 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) { |
Ken Rockot(use gerrit already)
2015/05/27 00:12:23
I'm not terribly happy with the fact that this act
jam
2015/05/27 16:05:35
i think we want to test the production code path,
|
+ 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 |