| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "ipc/ipc_mojo_bootstrap.h" | 5 #include "ipc/ipc_mojo_bootstrap.h" |
| 6 | 6 |
| 7 #include <stdint.h> | 7 #include <stdint.h> |
| 8 #include <memory> | 8 #include <memory> |
| 9 | 9 |
| 10 #include "base/base_paths.h" | 10 #include "base/base_paths.h" |
| (...skipping 17 matching lines...) Expand all Loading... |
| 28 class IPCMojoBootstrapTest : public testing::Test { | 28 class IPCMojoBootstrapTest : public testing::Test { |
| 29 protected: | 29 protected: |
| 30 mojo::edk::test::MultiprocessTestHelper helper_; | 30 mojo::edk::test::MultiprocessTestHelper helper_; |
| 31 }; | 31 }; |
| 32 | 32 |
| 33 class TestingDelegate : public IPC::MojoBootstrap::Delegate { | 33 class TestingDelegate : public IPC::MojoBootstrap::Delegate { |
| 34 public: | 34 public: |
| 35 explicit TestingDelegate(const base::Closure& quit_callback) | 35 explicit TestingDelegate(const base::Closure& quit_callback) |
| 36 : passed_(false), quit_callback_(quit_callback) {} | 36 : passed_(false), quit_callback_(quit_callback) {} |
| 37 | 37 |
| 38 void OnPipesAvailable(IPC::mojom::ChannelAssociatedPtrInfo send_channel, | 38 void OnPipesAvailable( |
| 39 IPC::mojom::ChannelAssociatedRequest receive_channel, | 39 IPC::mojom::ChannelAssociatedPtr sender, |
| 40 int32_t peer_pid) override; | 40 IPC::mojom::ChannelAssociatedRequest receiver) override; |
| 41 void OnBootstrapError() override; | |
| 42 | 41 |
| 43 bool passed() const { return passed_; } | 42 bool passed() const { return passed_; } |
| 44 | 43 |
| 45 private: | 44 private: |
| 46 bool passed_; | 45 bool passed_; |
| 47 const base::Closure quit_callback_; | 46 const base::Closure quit_callback_; |
| 48 }; | 47 }; |
| 49 | 48 |
| 50 void TestingDelegate::OnPipesAvailable( | 49 void TestingDelegate::OnPipesAvailable( |
| 51 IPC::mojom::ChannelAssociatedPtrInfo send_channel, | 50 IPC::mojom::ChannelAssociatedPtr sender, |
| 52 IPC::mojom::ChannelAssociatedRequest receive_channel, | 51 IPC::mojom::ChannelAssociatedRequest receiver) { |
| 53 int32_t peer_pid) { | |
| 54 passed_ = true; | 52 passed_ = true; |
| 55 quit_callback_.Run(); | 53 quit_callback_.Run(); |
| 56 } | 54 } |
| 57 | 55 |
| 58 void TestingDelegate::OnBootstrapError() { | |
| 59 quit_callback_.Run(); | |
| 60 } | |
| 61 | |
| 62 TEST_F(IPCMojoBootstrapTest, Connect) { | 56 TEST_F(IPCMojoBootstrapTest, Connect) { |
| 63 base::MessageLoop message_loop; | 57 base::MessageLoop message_loop; |
| 64 base::RunLoop run_loop; | 58 base::RunLoop run_loop; |
| 65 TestingDelegate delegate(run_loop.QuitClosure()); | 59 TestingDelegate delegate(run_loop.QuitClosure()); |
| 66 std::unique_ptr<IPC::MojoBootstrap> bootstrap = IPC::MojoBootstrap::Create( | 60 std::unique_ptr<IPC::MojoBootstrap> bootstrap = IPC::MojoBootstrap::Create( |
| 67 helper_.StartChild("IPCMojoBootstrapTestClient"), | 61 helper_.StartChild("IPCMojoBootstrapTestClient"), |
| 68 IPC::Channel::MODE_SERVER, &delegate); | 62 IPC::Channel::MODE_SERVER, &delegate, |
| 63 base::ThreadTaskRunnerHandle::Get()); |
| 69 | 64 |
| 70 bootstrap->Connect(); | 65 bootstrap->Connect(); |
| 71 run_loop.Run(); | 66 run_loop.Run(); |
| 72 | 67 |
| 73 EXPECT_TRUE(delegate.passed()); | 68 EXPECT_TRUE(delegate.passed()); |
| 74 EXPECT_TRUE(helper_.WaitForChildTestShutdown()); | 69 EXPECT_TRUE(helper_.WaitForChildTestShutdown()); |
| 75 } | 70 } |
| 76 | 71 |
| 77 // A long running process that connects to us. | 72 // A long running process that connects to us. |
| 78 MULTIPROCESS_TEST_MAIN_WITH_SETUP( | 73 MULTIPROCESS_TEST_MAIN_WITH_SETUP( |
| 79 IPCMojoBootstrapTestClientTestChildMain, | 74 IPCMojoBootstrapTestClientTestChildMain, |
| 80 ::mojo::edk::test::MultiprocessTestHelper::ChildSetup) { | 75 ::mojo::edk::test::MultiprocessTestHelper::ChildSetup) { |
| 81 base::MessageLoop message_loop; | 76 base::MessageLoop message_loop; |
| 82 base::RunLoop run_loop; | 77 base::RunLoop run_loop; |
| 83 TestingDelegate delegate(run_loop.QuitClosure()); | 78 TestingDelegate delegate(run_loop.QuitClosure()); |
| 84 std::unique_ptr<IPC::MojoBootstrap> bootstrap = IPC::MojoBootstrap::Create( | 79 std::unique_ptr<IPC::MojoBootstrap> bootstrap = IPC::MojoBootstrap::Create( |
| 85 mojo::edk::CreateChildMessagePipe( | 80 mojo::edk::CreateChildMessagePipe( |
| 86 mojo::edk::test::MultiprocessTestHelper::primordial_pipe_token), | 81 mojo::edk::test::MultiprocessTestHelper::primordial_pipe_token), |
| 87 IPC::Channel::MODE_CLIENT, &delegate); | 82 IPC::Channel::MODE_CLIENT, &delegate, |
| 83 base::ThreadTaskRunnerHandle::Get()); |
| 88 | 84 |
| 89 bootstrap->Connect(); | 85 bootstrap->Connect(); |
| 90 | 86 |
| 91 run_loop.Run(); | 87 run_loop.Run(); |
| 92 | 88 |
| 93 return delegate.passed() ? 0 : 1; | 89 return delegate.passed() ? 0 : 1; |
| 94 } | 90 } |
| 95 | 91 |
| 96 } // namespace | 92 } // namespace |
| OLD | NEW |