1 //===------ MappedIteratorTest.cpp - Unit tests for mapped_iterator -------===//
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/STLExtras.h"
11 #include "gtest/gtest.h"
12
13 using namespace llvm;
14
15 namespace {
16
TEST(MappedIteratorTest,ApplyFunctionOnDereference)17 TEST(MappedIteratorTest, ApplyFunctionOnDereference) {
18 std::vector<int> V({0});
19
20 auto I = map_iterator(V.begin(), [](int X) { return X + 1; });
21
22 EXPECT_EQ(*I, 1) << "should have applied function in dereference";
23 }
24
TEST(MappedIteratorTest,ApplyFunctionOnArrow)25 TEST(MappedIteratorTest, ApplyFunctionOnArrow) {
26 struct S {
27 int Z = 0;
28 };
29
30 std::vector<int> V({0});
31 S Y;
32 S* P = &Y;
33
34 auto I = map_iterator(V.begin(), [&](int X) -> S& { return *(P + X); });
35
36 I->Z = 42;
37
38 EXPECT_EQ(Y.Z, 42) << "should have applied function during arrow";
39 }
40
TEST(MappedIteratorTest,FunctionPreservesReferences)41 TEST(MappedIteratorTest, FunctionPreservesReferences) {
42 std::vector<int> V({1});
43 std::map<int, int> M({ {1, 1} });
44
45 auto I = map_iterator(V.begin(), [&](int X) -> int& { return M[X]; });
46 *I = 42;
47
48 EXPECT_EQ(M[1], 42) << "assignment should have modified M";
49 }
50
51 } // anonymous namespace
52