mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 16:46:47 +00:00
114 lines
3.2 KiB
C++
114 lines
3.2 KiB
C++
/********************************************************************************
|
|
* 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(Resize, zeromq)
|
|
{
|
|
RunPushPullWithMsgResize("zeromq", "ipc://test_message_resize");
|
|
}
|
|
|
|
TEST(Resize, shmem)
|
|
{
|
|
RunPushPullWithMsgResize("shmem", "ipc://test_message_resize");
|
|
}
|
|
|
|
#ifdef BUILD_NANOMSG_TRANSPORT
|
|
TEST(Resize, nanomsg)
|
|
{
|
|
RunPushPullWithMsgResize("nanomsg", "ipc://test_message_resize");
|
|
}
|
|
#endif /* BUILD_NANOMSG_TRANSPORT */
|
|
|
|
TEST(Rebuild, zeromq)
|
|
{
|
|
RunMsgRebuild("zeromq");
|
|
}
|
|
|
|
TEST(Rebuild, shmem)
|
|
{
|
|
RunMsgRebuild("shmem");
|
|
}
|
|
|
|
#ifdef BUILD_NANOMSG_TRANSPORT
|
|
TEST(Rebuild, nanomsg)
|
|
{
|
|
RunMsgRebuild("nanomsg");
|
|
}
|
|
#endif /* BUILD_NANOMSG_TRANSPORT */
|
|
|
|
} // namespace
|