ipc/mojo/ipc_mojo_bootstrap_unittest.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "ipc/mojo/ipc_mojo_bootstrap.h"
   6
   7 #include "base/base_paths.h"
   8 #include "base/files/file.h"
   9 #include "base/message_loop/message_loop.h"
  10 #include "ipc/ipc_test_base.h"
  11
  12 #if defined(OS_POSIX)
  13 #include "base/file_descriptor_posix.h"
  14 #endif
  15
  16 namespace {
  17
  18 class IPCMojoBootstrapTest : public IPCTestBase {
  19  protected:
  20 };
  21
  22 class TestingDelegate : public IPC::MojoBootstrap::Delegate {
  23  public:
  24   TestingDelegate() : passed_(false) {}
  25
  26   void OnPipeAvailable(mojo::embedder::ScopedPlatformHandle handle) override;
  27   void OnBootstrapError() override;
  28
  29   bool passed() const { return passed_; }
  30
  31  private:
  32   bool passed_;
  33 };
  34
  35 void TestingDelegate::OnPipeAvailable(
  36     mojo::embedder::ScopedPlatformHandle handle) {
  37   passed_ = true;
  38   base::MessageLoop::current()->Quit();
  39 }
  40
  41 void TestingDelegate::OnBootstrapError() {
  42   base::MessageLoop::current()->Quit();
  43 }
  44
  45 TEST_F(IPCMojoBootstrapTest, Connect) {
  46   Init("IPCMojoBootstrapTestClient");
  47
  48   TestingDelegate delegate;
  49   scoped_ptr<IPC::MojoBootstrap> bootstrap = IPC::MojoBootstrap::Create(
  50       GetTestChannelHandle(), IPC::Channel::MODE_SERVER, &delegate);
  51
  52   ASSERT_TRUE(bootstrap->Connect());
  53 #if defined(OS_POSIX)
  54   ASSERT_TRUE(StartClientWithFD(bootstrap->GetClientFileDescriptor()));
  55 #else
  56   ASSERT_TRUE(StartClient());
  57 #endif
  58   bootstrap->OnClientLaunched(client_process().Handle());
  59
  60   base::MessageLoop::current()->Run();
  61
  62   EXPECT_TRUE(delegate.passed());
  63   EXPECT_TRUE(WaitForClientShutdown());
  64 }
  65
  66 // A long running process that connects to us.
  67 MULTIPROCESS_IPC_TEST_CLIENT_MAIN(IPCMojoBootstrapTestClient) {
  68   base::MessageLoopForIO main_message_loop;
  69
  70   TestingDelegate delegate;
  71   scoped_ptr<IPC::MojoBootstrap> bootstrap = IPC::MojoBootstrap::Create(
  72       IPCTestBase::GetChannelName("IPCMojoBootstrapTestClient"),
  73       IPC::Channel::MODE_CLIENT,
  74       &delegate);
  75
  76   bootstrap->Connect();
  77
  78   base::MessageLoop::current()->Run();
  79
  80   EXPECT_TRUE(delegate.passed());
  81
  82   return 0;
  83 }
  84
  85 }  // namespace