1 //===- RangeAdapterTest.cpp - Unit tests for range adapters --------------===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include "llvm/ADT/iterator_range.h"
11 #include "llvm/ADT/STLExtras.h"
12 #include "gtest/gtest.h"
13
14 #include <iterator>
15 #include <list>
16 #include <vector>
17
18 using namespace llvm;
19
20 namespace {
21
22 // A wrapper around vector which exposes rbegin(), rend().
23 class ReverseOnlyVector {
24 std::vector<int> Vec;
25
26 public:
ReverseOnlyVector(std::initializer_list<int> list)27 ReverseOnlyVector(std::initializer_list<int> list) : Vec(list) {}
28
29 typedef std::vector<int>::reverse_iterator reverse_iterator;
rbegin()30 reverse_iterator rbegin() { return Vec.rbegin(); }
rend()31 reverse_iterator rend() { return Vec.rend(); }
32 };
33
34 // A wrapper around vector which exposes begin(), end(), rbegin() and rend().
35 // begin() and end() don't have implementations as this ensures that we will
36 // get a linker error if reverse() chooses begin()/end() over rbegin(), rend().
37 class BidirectionalVector {
38 std::vector<int> Vec;
39
40 public:
BidirectionalVector(std::initializer_list<int> list)41 BidirectionalVector(std::initializer_list<int> list) : Vec(list) {}
42
43 typedef std::vector<int>::iterator iterator;
44 iterator begin();
45 iterator end();
46
47 typedef std::vector<int>::reverse_iterator reverse_iterator;
rbegin()48 reverse_iterator rbegin() { return Vec.rbegin(); }
rend()49 reverse_iterator rend() { return Vec.rend(); }
50 };
51
TestRev(const R & r)52 template <typename R> void TestRev(const R &r) {
53 int counter = 3;
54 for (int i : r)
55 EXPECT_EQ(i, counter--);
56 }
57
58 // Test fixture
59 template <typename T> class RangeAdapterLValueTest : public ::testing::Test {};
60
61 typedef ::testing::Types<std::vector<int>, std::list<int>, int[4]>
62 RangeAdapterLValueTestTypes;
63 TYPED_TEST_CASE(RangeAdapterLValueTest, RangeAdapterLValueTestTypes);
64
TYPED_TEST(RangeAdapterLValueTest,TrivialOperation)65 TYPED_TEST(RangeAdapterLValueTest, TrivialOperation) {
66 TypeParam v = {0, 1, 2, 3};
67 TestRev(reverse(v));
68
69 const TypeParam c = {0, 1, 2, 3};
70 TestRev(reverse(c));
71 }
72
73 template <typename T> struct RangeAdapterRValueTest : testing::Test {};
74
75 typedef ::testing::Types<std::vector<int>, std::list<int>, ReverseOnlyVector,
76 BidirectionalVector> RangeAdapterRValueTestTypes;
77 TYPED_TEST_CASE(RangeAdapterRValueTest, RangeAdapterRValueTestTypes);
78
TYPED_TEST(RangeAdapterRValueTest,TrivialOperation)79 TYPED_TEST(RangeAdapterRValueTest, TrivialOperation) {
80 TestRev(reverse(TypeParam({0, 1, 2, 3})));
81 }
82
83 } // anonymous namespace
84