src/lib/fidl/llcpp/tests/transport/context_test.cc

// Copyright 2021 The Fuchsia 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 <lib/fidl/cpp/wire/internal/transport.h>

#include <zxtest/zxtest.h>

struct TestTransport1 {
  using OutgoingTransportContextType = int64_t;
  static constexpr const fidl::internal::TransportVTable VTable = {
      .type = fidl::internal::fidl_transport_type::kTest,
  };
};

struct TestTransport2 {
  using OutgoingTransportContextType = void;
  static constexpr const fidl::internal::TransportVTable VTable = {
      .type = fidl::internal::fidl_transport_type::kChannel,
  };
};

namespace {

TEST(OutgoingContext, CreateAndReceiveSameType) {
  int64_t input = 123ll;
  fidl::internal::OutgoingTransportContext context =
      fidl::internal::OutgoingTransportContext::Create<TestTransport1>(&input);
  fidl::internal::OutgoingTransportContext moved_context = std::move(context);
  ASSERT_DEATH([&]() { context.release<TestTransport1>(); });

  int64_t* value = moved_context.release<TestTransport1>();
  ASSERT_EQ(&input, value);
  ASSERT_DEATH([&]() { moved_context.release<TestTransport1>(); });
}

TEST(OutgoingContext, CreateAndReceiveDifferentType) {
  int64_t input = 123ll;
  fidl::internal::OutgoingTransportContext context =
      fidl::internal::OutgoingTransportContext::Create<TestTransport1>(&input);
  ASSERT_DEATH([&]() { context.release<TestTransport2>(); });
}

}  // namespace

struct ClosingTestTransport {
  using OutgoingTransportContextType = uint16_t;
  static constexpr const fidl::internal::TransportVTable VTable = {
      .type = fidl::internal::fidl_transport_type::kTest,
      .close_outgoing_transport_context =
          [](void* value) {
            uint16_t* uintval = static_cast<uint16_t*>(value);
            (*uintval)--;
          },
  };
};

namespace {

TEST(OutgoingContext, Closing) {
  uint16_t input = 1;
  { auto unused = fidl::internal::OutgoingTransportContext::Create<ClosingTestTransport>(&input); }
  ASSERT_EQ(0, input);
}

}  // namespace