mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 08:41:16 +00:00
99 lines
3.1 KiB
C++
99 lines
3.1 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 <gtest/gtest.h>
|
|
#include <FairMQChannel.h>
|
|
#include <FairMQParts.h>
|
|
#include <FairMQLogger.h>
|
|
#include <FairMQTransportFactory.h>
|
|
#include <fairmq/Tools.h>
|
|
#include <options/FairMQProgOptions.h>
|
|
|
|
#include <algorithm>
|
|
#include <memory>
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <thread>
|
|
|
|
namespace
|
|
{
|
|
|
|
using namespace std;
|
|
|
|
void CheckOldOptionInterface(FairMQChannel& channel, const string& option, const string& transport)
|
|
{
|
|
int value = 500;
|
|
channel.GetSocket().SetOption(option, &value, sizeof(value));
|
|
value = 0;
|
|
size_t valueSize = sizeof(value);
|
|
channel.GetSocket().GetOption(option, &value, &valueSize);
|
|
if (transport == "nanomsg" && (option == "snd-hwm" || option == "rcv-hwm")) {
|
|
ASSERT_EQ(value, -1);
|
|
} else {
|
|
ASSERT_EQ(value, 500);
|
|
}
|
|
}
|
|
|
|
void RunOptionsTest(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);
|
|
FairMQChannel channel("Push", "push", factory);
|
|
|
|
CheckOldOptionInterface(channel, "linger", transport);
|
|
CheckOldOptionInterface(channel, "snd-hwm", transport);
|
|
CheckOldOptionInterface(channel, "rcv-hwm", transport);
|
|
CheckOldOptionInterface(channel, "snd-size", transport);
|
|
CheckOldOptionInterface(channel, "rcv-size", transport);
|
|
|
|
channel.GetSocket().SetLinger(300);
|
|
ASSERT_EQ(channel.GetSocket().GetLinger(), 300);
|
|
|
|
channel.GetSocket().SetSndBufSize(500);
|
|
if (transport == "nanomsg") { // nanomsg doesn't use this option and the getter always returns -1
|
|
ASSERT_EQ(channel.GetSocket().GetSndBufSize(), -1);
|
|
} else {
|
|
ASSERT_EQ(channel.GetSocket().GetSndBufSize(), 500);
|
|
}
|
|
|
|
channel.GetSocket().SetRcvBufSize(500);
|
|
if (transport == "nanomsg") { // nanomsg doesn't use this option and the getter always returns -1
|
|
ASSERT_EQ(channel.GetSocket().GetRcvBufSize(), -1);
|
|
} else {
|
|
ASSERT_EQ(channel.GetSocket().GetRcvBufSize(), 500);
|
|
}
|
|
|
|
channel.GetSocket().SetSndKernelSize(8000);
|
|
ASSERT_EQ(channel.GetSocket().GetSndKernelSize(), 8000);
|
|
|
|
channel.GetSocket().SetRcvKernelSize(8000);
|
|
ASSERT_EQ(channel.GetSocket().GetRcvKernelSize(), 8000);
|
|
}
|
|
|
|
TEST(Options, zeromq)
|
|
{
|
|
RunOptionsTest("zeromq");
|
|
}
|
|
|
|
TEST(Options, shmem)
|
|
{
|
|
RunOptionsTest("shmem");
|
|
}
|
|
|
|
#ifdef BUILD_NANOMSG_TRANSPORT
|
|
TEST(Options, nanomsg)
|
|
{
|
|
RunOptionsTest("nanomsg");
|
|
}
|
|
#endif /* BUILD_NANOMSG_TRANSPORT */
|
|
|
|
} // namespace
|