diff options
author | Chandler Carruth <chandlerc@gmail.com> | 2014-04-24 21:10:35 +0000 |
---|---|---|
committer | Chandler Carruth <chandlerc@gmail.com> | 2014-04-24 21:10:35 +0000 |
commit | 01f21349f26fc4765f90dc4b1b9e335bfc2550dc (patch) | |
tree | 491fcbca20b1385386cb96841fdc10ca31f143b2 /unittests/Support | |
parent | fda5e19b96afdc33d16b9b006e703d8fe91801fa (diff) | |
download | llvm-01f21349f26fc4765f90dc4b1b9e335bfc2550dc.tar.gz llvm-01f21349f26fc4765f90dc4b1b9e335bfc2550dc.tar.bz2 llvm-01f21349f26fc4765f90dc4b1b9e335bfc2550dc.tar.xz |
[ADT] Generalize pointee_iterator to smart pointers by using decltype.
Based on review feedback from Dave on the original patch.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@207146 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'unittests/Support')
-rw-r--r-- | unittests/Support/IteratorTest.cpp | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/unittests/Support/IteratorTest.cpp b/unittests/Support/IteratorTest.cpp index 3a16406d6f..83848328c0 100644 --- a/unittests/Support/IteratorTest.cpp +++ b/unittests/Support/IteratorTest.cpp @@ -8,6 +8,7 @@ //===----------------------------------------------------------------------===// #include "llvm/ADT/iterator.h" +#include "llvm/ADT/STLExtras.h" #include "llvm/ADT/SmallVector.h" #include "gtest/gtest.h" @@ -56,4 +57,45 @@ TEST(PointeeIteratorTest, Basic) { EXPECT_EQ(End, I); } +TEST(PointeeIteratorTest, SmartPointer) { + SmallVector<std::unique_ptr<int>, 4> V; + V.push_back(make_unique<int>(1)); + V.push_back(make_unique<int>(2)); + V.push_back(make_unique<int>(3)); + V.push_back(make_unique<int>(4)); + + typedef pointee_iterator< + SmallVectorImpl<std::unique_ptr<int>>::const_iterator> test_iterator; + + test_iterator Begin, End; + Begin = V.begin(); + End = test_iterator(V.end()); + + test_iterator I = Begin; + for (int i = 0; i < 4; ++i) { + EXPECT_EQ(*V[i], *I); + + EXPECT_EQ(I, Begin + i); + EXPECT_EQ(I, std::next(Begin, i)); + test_iterator J = Begin; + J += i; + EXPECT_EQ(I, J); + EXPECT_EQ(*V[i], Begin[i]); + + EXPECT_NE(I, End); + EXPECT_GT(End, I); + EXPECT_LT(I, End); + EXPECT_GE(I, Begin); + EXPECT_LE(Begin, I); + + EXPECT_EQ(i, I - Begin); + EXPECT_EQ(i, std::distance(Begin, I)); + EXPECT_EQ(Begin, I - i); + + test_iterator K = I++; + EXPECT_EQ(K, std::prev(I)); + } + EXPECT_EQ(End, I); +} + } // anonymous namespace |