FairMQ/fairmq/test/parts/_iterator_interface.cxx

79 lines
2.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 <FairMQParts.h>
#include <FairMQDevice.h>
#include <zeromq/FairMQTransportFactoryZMQ.h>
#include <string>
#include <sstream>
#include <algorithm>
#include <memory>
namespace
{
using namespace std;
class RandomAccessIterator : public ::testing::Test {
protected:
FairMQParts mParts;
shared_ptr<FairMQTransportFactory> mFactory;
const string mS1, mS2, mS3;
RandomAccessIterator()
: mParts(FairMQParts{}),
mFactory(FairMQTransportFactoryZMQ{}),
mS1("1"),
mS2("2"),
mS3("3")
{
mParts.AddPart(mFactory->NewSimpleMessage(mS1));
mParts.AddPart(mFactory->NewSimpleMessage(mS2));
mParts.AddPart(mFactory->NewSimpleMessage(mS3));
}
};
TEST_F(RandomAccessIterator, RangeForLoopConst)
{
stringstream out;
for (const auto& part : mParts) {
out << string{static_cast<char*>(part->GetData()), part->GetSize()};
}
ASSERT_EQ(out.str(), "123");
}
TEST_F(RandomAccessIterator, RangeForLoopMutation)
{
auto s = string{"4"};
for (auto&& part : mParts) {
part = mFactory->NewSimpleMessage(s);
}
stringstream out;
for (const auto& part : mParts) {
out << string{static_cast<char*>(part->GetData()), part->GetSize()};
}
ASSERT_EQ(out.str(), "444");
}
TEST_F(RandomAccessIterator, ForEachConst)
{
stringstream out;
for_each(mParts.cbegin(), mParts.cend(), [&out](const FairMQMessagePtr& part) {
out << string{static_cast<char*>(part->GetData()), part->GetSize()};
});
ASSERT_EQ(out.str(), "123");
}
} // namespace