fly316/PX4-Autopilot
7
0
Fork 0
mirror of https://gitee.com/mirrors_PX4/PX4-Autopilot.git synced 2026-10-23 04:53:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Multiset<>::forEach()

Browse Source
This commit is contained in:
Pavel Kirienko
2015-05-16 13:38:42 +03:00
parent be5bcf9084
commit 713ec48ce9
2 changed files with 115 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libuavcan/test/util/multiset.cpp
+56
View File
@@ -49,6 +49,20 @@ struct NoncopyableWithCounter : uavcan::Noncopyable
int NoncopyableWithCounter::num_objects = 0;
template <typename T>
struct SummationOperator : uavcan::Noncopyable
{
T accumulator;
SummationOperator() : accumulator() { }
void operator()(const T& x) { accumulator += x; }
};
struct ClearingOperator
{
template <typename T>
void operator()(T& x) const { x = T(); }
};
TEST(Multiset, Basic)
{
@@ -62,6 +76,8 @@ TEST(Multiset, Basic)
typedef Multiset<std::string, 2> MultisetType;
std::auto_ptr<MultisetType> mset(new MultisetType(poolmgr));
typedef SummationOperator<std::string> StringConcatenationOperator;
// Empty
mset->removeFirst("foo");
ASSERT_EQ(0, pool.getNumUsedBlocks());
@@ -80,6 +96,12 @@ TEST(Multiset, Basic)
ASSERT_TRUE(*mset->getByIndex(0) == "1");
ASSERT_TRUE(*mset->getByIndex(1) == "2");
{
StringConcatenationOperator op;
mset->forEach<StringConcatenationOperator&>(op);
ASSERT_EQ("12", op.accumulator);
}
// Dynamic addition
ASSERT_EQ("3", *mset->emplace("3"));
ASSERT_EQ("3", *mset->getByIndex(2));
@@ -107,6 +129,13 @@ TEST(Multiset, Basic)
ASSERT_EQ("4", *mset->find(FindPredicate("4")));
ASSERT_FALSE(mset->find(FindPredicate("nonexistent")));
{
StringConcatenationOperator op;
mset->forEach<StringConcatenationOperator&>(op);
std::cout << "Accumulator: " << op.accumulator << std::endl;
ASSERT_EQ(4, op.accumulator.size());
}
// Removing one static; ordering will be preserved
mset->removeFirst("1");
mset->removeFirst("foo"); // There's no such thing anyway
@@ -165,6 +194,20 @@ TEST(Multiset, Basic)
}
}
// Clearing all strings
{
StringConcatenationOperator op;
mset->forEach<StringConcatenationOperator&>(op);
std::cout << "Accumulator before clearing: " << op.accumulator << std::endl;
}
mset->forEach(ClearingOperator());
{
StringConcatenationOperator op;
mset->forEach<StringConcatenationOperator&>(op);
std::cout << "Accumulator after clearing: " << op.accumulator << std::endl;
ASSERT_TRUE(op.accumulator.empty());
}
// Making sure the memory will be released
mset.reset();
ASSERT_EQ(0, pool.getNumUsedBlocks());
@@ -241,6 +284,19 @@ TEST(Multiset, PrimitiveKey)
ASSERT_FALSE(mset->getByIndex(5));
ASSERT_FALSE(mset->getByIndex(1000));
#endif
// Summation and clearing
{
SummationOperator<int> summation_operator;
mset->forEach<SummationOperator<int>&>(summation_operator);
ASSERT_EQ(1 + 2 + 3 + 4, summation_operator.accumulator);
}
mset->forEach(ClearingOperator());
{
SummationOperator<int> summation_operator;
mset->forEach<SummationOperator<int>&>(summation_operator);
ASSERT_EQ(0, summation_operator.accumulator);
}
}