unittests/ADT/RangeAdapterTest.cpp

   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 public:
  26   ReverseOnlyVector(std::initializer_list<int> list) : Vec(list) { }
  27
  28   typedef std::vector<int>::reverse_iterator reverse_iterator;
  29   reverse_iterator rbegin() { return Vec.rbegin(); }
  30   reverse_iterator rend() { return Vec.rend(); }
  31 };
  32
  33 // A wrapper around vector which exposes begin(), end(), rbegin() and rend().
  34 // begin() and end() don't have implementations as this ensures that we will
  35 // get a linker error if reverse() chooses begin()/end() over rbegin(), rend().
  36 class BidirectionalVector {
  37   std::vector<int> Vec;
  38
  39 public:
  40   BidirectionalVector(std::initializer_list<int> list) : Vec(list) { }
  41
  42   typedef std::vector<int>::iterator iterator;
  43   iterator begin();
  44   iterator end();
  45
  46   typedef std::vector<int>::reverse_iterator reverse_iterator;
  47   reverse_iterator rbegin() { return Vec.rbegin(); }
  48   reverse_iterator rend() { return Vec.rend(); }
  49 };
  50
  51 // Test fixture
  52 template <typename T>
  53 class RangeAdapterTest : public ::testing::Test { };
  54
  55 typedef ::testing::Types<std::vector<int>,
  56                          std::list<int>,
  57                          int[4],
  58                          ReverseOnlyVector,
  59                          BidirectionalVector,
  60                          const std::vector<int>,
  61                          const std::list<int>,
  62                          const int[4]> RangeAdapterTestTypes;
  63 TYPED_TEST_CASE(RangeAdapterTest, RangeAdapterTestTypes);
  64
  65 TYPED_TEST(RangeAdapterTest, TrivialOperation) {
  66   TypeParam v = { 0, 1, 2, 3 };
  67
  68   int counter = 3;
  69   for (int i : reverse(v))
  70     EXPECT_EQ(i, counter--);
  71
  72   counter = 0;
  73   for (int i : reverse(reverse(v)))
  74     EXPECT_EQ(i, counter++);
  75 }
  76
  77 } // anonymous namespace