cpp/bindings/tests/synchronous_connector_unittest.cc

// Copyright 2016 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 "mojo/public/cpp/bindings/lib/synchronous_connector.h"

#include <string.h>

#include <string>
#include <utility>

#include "gtest/gtest.h"
#include "mojo/public/cpp/bindings/lib/message_builder.h"
#include "mojo/public/cpp/bindings/message.h"

namespace mojo {
namespace test {
namespace {

void AllocMessage(const std::string& text, Message* message) {
  size_t payload_size = text.length() + 1;  // Plus null terminator.
  MessageBuilder builder(1, payload_size);
  memcpy(builder.buffer()->Allocate(payload_size), text.c_str(), payload_size);

  builder.message()->MoveTo(message);
}

// Simple success case.
TEST(SynchronousConnectorTest, Basic) {
  MessagePipe pipe;
  internal::SynchronousConnector connector0(std::move(pipe.handle0));
  internal::SynchronousConnector connector1(std::move(pipe.handle1));

  const std::string kText("hello world");
  Message message;
  AllocMessage(kText, &message);

  EXPECT_TRUE(connector0.Write(&message));

  Message message_received;
  EXPECT_TRUE(connector1.BlockingRead(&message_received));

  EXPECT_EQ(
      kText,
      std::string(reinterpret_cast<const char*>(message_received.payload())));
}

// Writing to a closed pipe should fail.
TEST(SynchronousConnectorTest, WriteToClosedPipe) {
  MessagePipe pipe;
  internal::SynchronousConnector connector0(std::move(pipe.handle0));

  // Close the other end of the pipe.
  pipe.handle1.reset();

  Message message;
  EXPECT_FALSE(connector0.Write(&message));
}

// Reading from a closed pipe should fail (while waiting on it).
TEST(SynchronousConnectorTest, ReadFromClosedPipe) {
  MessagePipe pipe;
  internal::SynchronousConnector connector0(std::move(pipe.handle0));

  // Close the other end of the pipe.
  pipe.handle1.reset();

  Message message;
  EXPECT_FALSE(connector0.BlockingRead(&message));
}

}  // namespace
}  // namespace test
}  // namespace mojo