Function TEST_F

tensorflow/core/common_runtime/buf_rendezvous_test.cc:79–109

const uint64 BufRendezvousTest::kDefaultIncarnation = 12345;

TEST_F(BufRendezvousTest, CorrectUseProducerFirst) {
  Status prod_status;
  Status cons_status;
  bool prod_callback_called = false;
  bool cons_callback_called = false;
  Notification note;
  br_->ProvideBuf(
      *kDefaultKey, default_device_, fake_device_context_, &a_, aa_,
      [&note, &prod_status, &prod_callback_called](const Status& s) {
        prod_status = s;
        prod_callback_called = true;
        note.Notify();
      });
  EXPECT_FALSE(prod_callback_called);
  br_->ConsumeBuf(*kDefaultKey, *kDefaultDeviceName, kDefaultIncarnation,
                  [this, &cons_status, &cons_callback_called](
                      const Status& s, BufRendezvous::Hook* h) {
                    cons_status = s;
                    cons_callback_called = true;
                    ASSERT_TRUE(h != nullptr);
                    EXPECT_EQ(h->prod_dev, default_device_);
                    EXPECT_EQ(h->prod_ctx, fake_device_context_);
                    EXPECT_EQ(h->prod_value, &a_);
                    br_->DoneWithHook(h);
                  });
  EXPECT_TRUE(cons_callback_called);
  note.WaitForNotification();
  EXPECT_TRUE(prod_callback_called);
  TF_EXPECT_OK(cons_status);
  TF_EXPECT_OK(prod_status);
}

TEST_F(BufRendezvousTest, CorrectUseConsumerFirst) {
  Status prod_status;