From 99237a7c1f86b89b932a796be641dc8072c8726f Mon Sep 17 00:00:00 2001 From: Yossi Itigin Date: Fri, 27 Nov 2020 18:11:43 +0200 Subject: [PATCH] TEST/TAG: Fix tag test to check inline data flag --- test/gtest/uct/test_tag.cc | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/test/gtest/uct/test_tag.cc b/test/gtest/uct/test_tag.cc index 7a04b6ebf34..5c48ea5ed22 100644 --- a/test/gtest/uct/test_tag.cc +++ b/test/gtest/uct/test_tag.cc @@ -371,14 +371,19 @@ class test_tag : public uct_test { user_ctx->consumed = true; } + static void verify_completed(recv_ctx *user_ctx, uct_tag_t stag, size_t length) + { + EXPECT_EQ(user_ctx->tag, (stag & user_ctx->tmask)); + EXPECT_EQ(user_ctx->mbuf->length(), length); + } + static void completed(uct_tag_context_t *self, uct_tag_t stag, uint64_t imm, size_t length, void *inline_data, ucs_status_t status) { recv_ctx *user_ctx = ucs_container_of(self, recv_ctx, uct_ctx); user_ctx->comp = true; user_ctx->status = status; - EXPECT_EQ(user_ctx->tag, (stag & user_ctx->tmask)); - EXPECT_EQ(user_ctx->mbuf->length(), length); + verify_completed(user_ctx, stag, length); } static void sw_rndv_completed(uct_tag_context_t *self, uct_tag_t stag, @@ -388,8 +393,10 @@ class test_tag : public uct_test { recv_ctx *user_ctx = ucs_container_of(self, recv_ctx, uct_ctx); user_ctx->sw_rndv = true; user_ctx->status = status; - EXPECT_EQ(user_ctx->tag, (stag & user_ctx->tmask)); - EXPECT_EQ(user_ctx->mbuf->length(), header_length); + if (flags & UCT_TAG_RECV_CB_INLINE_DATA) { + memcpy(user_ctx->mbuf->ptr(), header, header_length); + } + verify_completed(user_ctx, stag, header_length); } static ucs_status_t unexp_eager(void *arg, void *data, size_t length,