Mirror/FairMQ
1
0
Fork 0
mirror of https://github.com/FairRootGroup/FairMQ.git synced 2025-10-15 17:41:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add new GetNumberOfConnectedPeers() API

Browse Source
This commit is contained in:
Dennis Klein
2021-10-12 18:14:33 +02:00
committed by Alexey Rybalchenko
parent 8796ce5b20
commit fda8126a43
7 changed files with 218 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
test/channel/_channel.cxx
View File

@@ -6,9 +6,14 @@
* copied verbatim in the file "LICENSE" *
********************************************************************************/
#include <chrono>
#include <fairmq/Channel.h>
#include <fairmq/ProgOptions.h>
#include <fairmq/Tools.h>
#include <fairmq/TransportFactory.h>
#include <gtest/gtest.h>
#include <string>
#include <thread>
namespace
{
@@ -91,4 +96,42 @@ TEST(Channel, Validation)
ASSERT_EQ(channel2.Validate(), true);
}
auto testConnectedPeers(std::string const& transport)
{
using namespace std::chrono_literals;
ProgOptions config;
config.SetProperty<string>("session", tools::Uuid());
string const address(tools::ToString("ipc://", config.GetProperty<string>("session")));
unsigned long constexpr zero(0), one(1);
auto factory(TransportFactory::CreateTransportFactory(transport, tools::Uuid(), &config));
Channel ch1("ch1", "pair", factory);
ASSERT_EQ(ch1.GetNumberOfConnectedPeers(), zero);
ch1.Bind(address);
ASSERT_EQ(ch1.GetNumberOfConnectedPeers(), zero);
{
Channel ch2("ch2", "pair", factory);
ASSERT_EQ(ch2.GetNumberOfConnectedPeers(), zero);
ch2.Connect(address);
std::this_thread::sleep_for(10ms);
ASSERT_EQ(ch1.GetNumberOfConnectedPeers(), one);
ASSERT_EQ(ch2.GetNumberOfConnectedPeers(), one);
}
std::this_thread::sleep_for(10ms);
ASSERT_EQ(ch1.GetNumberOfConnectedPeers(), zero);
}
TEST(Channel, GetNumberOfConnectedPeers_zeromq)
{
testConnectedPeers("zeromq");
}
TEST(Channel, GetNumberOfConnectedPeers_shmem)
{
testConnectedPeers("shmem");
}
} /* namespace */