mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-15 09:31:45 +00:00
Add test for FairMQMessage::Rebuild
This commit is contained in:
committed by
Dennis Klein
parent
c605cbc3f6
commit
a9619a06d0
@@ -74,14 +74,14 @@ add_testsuite(FairMQ.Parts
|
||||
TIMEOUT 5
|
||||
)
|
||||
|
||||
add_testsuite(FairMQ.MessageResize
|
||||
add_testsuite(FairMQ.Message
|
||||
SOURCES
|
||||
${CMAKE_CURRENT_BINARY_DIR}/runner.cxx
|
||||
message_resize/_message_resize.cxx
|
||||
message/_message.cxx
|
||||
|
||||
LINKS FairMQ
|
||||
INCLUDES ${CMAKE_CURRENT_SOURCE_DIR}
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/message_resize
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/message
|
||||
${CMAKE_CURRENT_BINARY_DIR}
|
||||
TIMEOUT 5
|
||||
${definitions}
|
||||
|
@@ -21,8 +21,8 @@ namespace
|
||||
|
||||
using namespace std;
|
||||
|
||||
auto RunPushPullWithMsgResize(string transport, string address) -> void {
|
||||
|
||||
void RunPushPullWithMsgResize(const string& transport, const string& address)
|
||||
{
|
||||
size_t session{fair::mq::tools::UuidHash()};
|
||||
|
||||
FairMQProgOptions config;
|
||||
@@ -53,6 +53,29 @@ auto RunPushPullWithMsgResize(string transport, string address) -> void {
|
||||
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");
|
||||
@@ -70,4 +93,21 @@ TEST(MessageResize, nanomsg)
|
||||
}
|
||||
#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
|
Reference in New Issue
Block a user