Mirror/FairMQ
1
0
Fork 0
mirror of https://github.com/FairRootGroup/FairMQ.git synced 2025-10-13 16:46:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

FairMQ: Add test for setting used message size functionality

Browse Source
This commit is contained in:
Alexey Rybalchenko 2017-12-07 14:36:11 +01:00 committed by Mohammad Al-Turany
parent ea7ae3ded9
commit 848ebf2ff3
3 changed files with 101 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
fairmq/test/CMakeLists.txt
View File

@ -64,6 +64,16 @@ add_testsuite(FairMQ.Parts
TIMEOUT 5
)
add_testsuite(FairMQ.MessageResize
SOURCES
message_resize/runner.cxx
message_resize/_message_resize.cxx
LINKS FairMQ ZeroMQ
INCLUDES ${CMAKE_CURRENT_SOURCE_DIR}/message_resize
TIMEOUT 5
)
add_testsuite(FairMQ.Device
SOURCES
device/TestSender.h

75
fairmq/test/message_resize/_message_resize.cxx Normal file
View File

@ -0,0 +1,75 @@
/********************************************************************************
* 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;
auto RunPushPullWithMsgResize(string transport, string address) -> void {
size_t session{fair::mq::tools::UuidHash()};
FairMQProgOptions config;
config.SetValue<string>("session", to_string(session));
config.SetValue<int>("io-threads", 1);
config.SetValue<size_t>("shm-segment-size", 20000000);
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);
}
TEST(MessageResize, ZeroMQ)
{
RunPushPullWithMsgResize("zeromq", "ipc://test_message_resize");
}
TEST(MessageResize, shmem)
{
RunPushPullWithMsgResize("shmem", "ipc://test_message_resize");
}
#ifdef NANOMSG_FOUND
TEST(MessageResize, nanomsg)
{
RunPushPullWithMsgResize("nanomsg", "ipc://test_message_resize");
}
#endif /* NANOMSG_FOUND */
} // namespace

16
fairmq/test/message_resize/runner.cxx Normal file
View File

@ -0,0 +1,16 @@
/********************************************************************************
* 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 <gtest/gtest.h>
auto main(int argc, char** argv) -> int
{
::testing::InitGoogleTest(&argc, argv);
::testing::FLAGS_gtest_death_test_style = "threadsafe";
return RUN_ALL_TESTS();
}