src/lib/fidl/llcpp/tests/transport/waiter_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/llcpp/internal/transport.h>

#include <zxtest/zxtest.h>

TEST(Transport, AnyTransportWaiter) {
  struct TestTransportWaiter : fidl::internal::TransportWaiter {
    TestTransportWaiter(zx_status_t begin_result, zx_status_t cancel_result, size_t* destruct_count)
        : begin_result(begin_result),
          cancel_result(cancel_result),
          destruct_count(destruct_count) {}
    ~TestTransportWaiter() override { (*destruct_count)++; }

    zx_status_t begin_result;
    zx_status_t cancel_result;
    size_t* destruct_count;

    zx_status_t Begin() override { return begin_result; }
    zx_status_t Cancel() override { return cancel_result; }
  };

  size_t destruct_count_a = 0;
  size_t destruct_count_b = 0;
  {
    fidl::internal::AnyTransportWaiter any_waiter;

    TestTransportWaiter& waiter_a =
        any_waiter.emplace<TestTransportWaiter>(1, 2, &destruct_count_a);
    ASSERT_EQ(0, destruct_count_a);
    ASSERT_EQ(1, waiter_a.begin_result);
    ASSERT_EQ(2, waiter_a.cancel_result);
    ASSERT_EQ(1, any_waiter->Begin());
    ASSERT_EQ(2, any_waiter->Cancel());
    TestTransportWaiter& waiter_b =
        any_waiter.emplace<TestTransportWaiter>(3, 4, &destruct_count_b);
    ASSERT_EQ(1, destruct_count_a);
    ASSERT_EQ(0, destruct_count_b);

    ASSERT_EQ(3, waiter_b.begin_result);
    ASSERT_EQ(4, waiter_b.cancel_result);
    ASSERT_EQ(3, any_waiter->Begin());
    ASSERT_EQ(4, any_waiter->Cancel());
    ASSERT_EQ(0, destruct_count_b);
  }

  ASSERT_EQ(1, destruct_count_b);
}