Index: unittests/Support/ReverseIterationTest.cpp =================================================================== --- unittests/Support/ReverseIterationTest.cpp +++ unittests/Support/ReverseIterationTest.cpp @@ -53,3 +53,53 @@ for (auto iter = Map.begin(), end = Map.end(); iter != end; iter++, ++i) ASSERT_EQ(iter->first, IterKeys[i]); } + +// Define a pointer-like int. +struct PtrLikeInt { int value; }; + +template<> struct DenseMapInfo { + static inline PtrLikeInt *getEmptyKey() { + return reinterpret_cast(0x7fffffff); + } + + static inline PtrLikeInt *getTombstoneKey() { + return reinterpret_cast(-0x7fffffff - 1); + } + + static unsigned getHashValue(const PtrLikeInt *PtrVal) { + return (unsigned)(PtrVal->value * 37U); + } + + static bool isEqual(const PtrLikeInt *LHS, const PtrLikeInt *RHS) { + return LHS == RHS; + } +}; + +TEST(ReverseIterationTest, DenseMapTest2) { + static_assert(detail::IsPointerLike::value, + "PtrLikeInt * is pointer-like"); + + PtrLikeInt a = {4}, b = {8}, c = {12}, d = {16}; + PtrLikeInt *Keys[] = { &a, &b, &c, &d }; + + // Insert keys into the DenseMap. + DenseMap Map; + for (auto *Key : Keys) + Map[Key] = Key->value; + + // Note: This is the observed order of keys in the DenseMap. + // If there is any change in the behavior of the DenseMap, this order + // would need to be adjusted accordingly. + PtrLikeInt *IterKeys[] = { &d, &a, &b, &c }; + if (shouldReverseIterate()) + std::reverse(&IterKeys[0], &IterKeys[4]); + + // Check that the DenseMap is iterated in the expected order. + for (const auto &Tuple : zip(Map, IterKeys)) + ASSERT_EQ(std::get<0>(Tuple).second, std::get<1>(Tuple)->value); + + // Check operator++ (post-increment). + int i = 0; + for (auto iter = Map.begin(), end = Map.end(); iter != end; iter++, ++i) + ASSERT_EQ(iter->second, IterKeys[i]->value); +}