Add test for FairMQMessage::Rebuild

test/message/_message.cxx

@@ -0,0 +1,113 @@
+/********************************************************************************
+ *    Copyright (C) 2017 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
+ *                                                                              *
+ *              This software is distributed under the terms of the             *
+ *              GNU Lesser General Public Licence (LGPL) version 3,             *
+ *                  copied verbatim in the file "LICENSE"                       *
+ ********************************************************************************/
+
+#include <FairMQChannel.h>
+#include <FairMQLogger.h>
+#include <FairMQTransportFactory.h>
+#include <options/FairMQProgOptions.h>
+#include <fairmq/Tools.h>
+
+#include <gtest/gtest.h>
+
+#include <string>
+
+namespace
+{
+
+using namespace std;
+
+void RunPushPullWithMsgResize(const string& transport, const string& address)
+{
+    size_t session{fair::mq::tools::UuidHash()};
+
+    FairMQProgOptions config;
+    config.SetValue<string>("session", to_string(session));
+
+    auto factory = FairMQTransportFactory::CreateTransportFactory(transport, fair::mq::tools::Uuid(), &config);
+
+    FairMQChannel push{"Push", "push", factory};
+    push.Bind(address);
+
+    FairMQChannel pull{"Pull", "pull", factory};
+    pull.Connect(address);
+
+    FairMQMessagePtr outMsg(push.NewMessage(1000));
+    ASSERT_EQ(outMsg->GetSize(), 1000);
+    outMsg->SetUsedSize(500);
+    ASSERT_EQ(outMsg->GetSize(), 500);
+    outMsg->SetUsedSize(250);
+    ASSERT_EQ(outMsg->GetSize(), 250);
+    FairMQMessagePtr msgCopy(push.NewMessage());
+    msgCopy->Copy(*outMsg);
+    ASSERT_EQ(msgCopy->GetSize(), 250);
+
+    ASSERT_EQ(push.Send(outMsg), 250);
+
+    FairMQMessagePtr inMsg(pull.NewMessage());
+    ASSERT_EQ(pull.Receive(inMsg), 250);
+    ASSERT_EQ(inMsg->GetSize(), 250);
+}
+
+void RunMsgRebuild(const string& transport)
+{
+    size_t session{fair::mq::tools::UuidHash()};
+
+    FairMQProgOptions config;
+    config.SetValue<string>("session", to_string(session));
+
+    auto factory = FairMQTransportFactory::CreateTransportFactory(transport, fair::mq::tools::Uuid(), &config);
+
+    FairMQMessagePtr msg(factory->CreateMessage());
+    EXPECT_EQ(msg->GetSize(), 0);
+    msg->Rebuild(100);
+    EXPECT_EQ(msg->GetSize(), 100);
+    string* str = new string("asdf");
+    msg->Rebuild(const_cast<char*>(str->c_str()),
+                 str->length(),
+                 [](void* /*data*/, void* obj) { delete static_cast<string*>(obj); },
+                 str);
+    EXPECT_NE(msg->GetSize(), 100);
+    EXPECT_EQ(msg->GetSize(), string("asdf").length());
+    EXPECT_EQ(string(static_cast<char*>(msg->GetData()), msg->GetSize()), string("asdf"));
+}
+
+TEST(MessageResize, ZeroMQ)
+{
+    RunPushPullWithMsgResize("zeromq", "ipc://test_message_resize");
+}
+
+TEST(MessageResize, shmem)
+{
+    RunPushPullWithMsgResize("shmem", "ipc://test_message_resize");
+}
+
+#ifdef BUILD_NANOMSG_TRANSPORT
+TEST(MessageResize, nanomsg)
+{
+    RunPushPullWithMsgResize("nanomsg", "ipc://test_message_resize");
+}
+#endif /* BUILD_NANOMSG_TRANSPORT */
+
+TEST(MessageRebuild, ZeroMQ)
+{
+    RunMsgRebuild("zeromq");
+}
+
+TEST(MessageRebuild, shmem)
+{
+    RunMsgRebuild("shmem");
+}
+
+#ifdef BUILD_NANOMSG_TRANSPORT
+TEST(MessageRebuild, nanomsg)
+{
+    RunMsgRebuild("nanomsg");
+}
+#endif /* BUILD_NANOMSG_TRANSPORT */
+
+} // namespace