#2212: tests: add a simple test for message properties

DARMA-tasking · Nov 9, 2023 · 5547342 · 5547342
1 parent 96689ec
commit 5547342
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/src/vt/messaging/pending_send.h b/src/vt/messaging/pending_send.h
@@ -153,6 +153,15 @@ struct PendingSend final {
    */
   void release();
 
+  /**
+   * \internal \brief Get the message stored in the pending send
+   *
+   * \note Used for testing purposes
+   *
+   * \return a reference to the message
+   */
+  MsgPtr<BaseMsgType>& getMsg() { return msg_; }
+
 private:
 
   /**

diff --git a/tests/unit/active/test_active_send.cc b/tests/unit/active/test_active_send.cc
@@ -242,4 +242,21 @@ TEST_F(TestActiveSend, test_active_message_serialization) {
   EXPECT_EQ(handler_count, num_msg_sent);
 }
 
+void testPropertiesHandler(int a, double b) {
+  // do nothing
+}
+
+TEST_F(TestActiveSend, test_active_message_properties) {
+  auto const this_node = theContext()->getNode();
+  auto const num_nodes = theContext()->getNumNodes();
+  NodeType const next_node = (this_node + 1) % num_nodes;
+
+  if (num_nodes > 1) {
+    auto ps = theMsg()->send<testPropertiesHandler>(
+      vt::Node{next_node}, MsgProps().asTerminationMsg(), 10, 20.0
+    );
+    EXPECT_TRUE(messaging::envelopeIsTerm(ps.getMsg()->env));
+  }
+}
+
 }}}} // end namespace vt::tests::unit::send