1 //===- LineIterator.cpp - Unit tests --------------------------------------===// 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/Support/LineIterator.h" 11 #include "llvm/ADT/OwningPtr.h" 12 #include "llvm/Support/MemoryBuffer.h" 13 #include "gtest/gtest.h" 14 15 using namespace llvm; 16 using namespace llvm::sys; 17 18 namespace { 19 20 TEST(LineIteratorTest, Basic) { 21 OwningPtr<MemoryBuffer> Buffer(MemoryBuffer::getMemBuffer("line 1\n" 22 "line 2\n" 23 "line 3")); 24 25 line_iterator I = line_iterator(*Buffer), E; 26 27 EXPECT_FALSE(I.is_at_eof()); 28 EXPECT_NE(E, I); 29 30 EXPECT_EQ("line 1", *I); 31 EXPECT_EQ(1, I.line_number()); 32 ++I; 33 EXPECT_EQ("line 2", *I); 34 EXPECT_EQ(2, I.line_number()); 35 ++I; 36 EXPECT_EQ("line 3", *I); 37 EXPECT_EQ(3, I.line_number()); 38 ++I; 39 40 EXPECT_TRUE(I.is_at_eof()); 41 EXPECT_EQ(E, I); 42 } 43 44 TEST(LineIteratorTest, CommentSkipping) { 45 OwningPtr<MemoryBuffer> Buffer(MemoryBuffer::getMemBuffer("line 1\n" 46 "line 2\n" 47 "# Comment 1\n" 48 "line 4\n" 49 "# Comment 2")); 50 51 line_iterator I = line_iterator(*Buffer, '#'), E; 52 53 EXPECT_FALSE(I.is_at_eof()); 54 EXPECT_NE(E, I); 55 56 EXPECT_EQ("line 1", *I); 57 EXPECT_EQ(1, I.line_number()); 58 ++I; 59 EXPECT_EQ("line 2", *I); 60 EXPECT_EQ(2, I.line_number()); 61 ++I; 62 EXPECT_EQ("line 4", *I); 63 EXPECT_EQ(4, I.line_number()); 64 ++I; 65 66 EXPECT_TRUE(I.is_at_eof()); 67 EXPECT_EQ(E, I); 68 } 69 70 TEST(LineIteratorTest, BlankSkipping) { 71 OwningPtr<MemoryBuffer> Buffer(MemoryBuffer::getMemBuffer("\n\n\n" 72 "line 1\n" 73 "\n\n\n" 74 "line 2\n" 75 "\n\n\n")); 76 77 line_iterator I = line_iterator(*Buffer), E; 78 79 EXPECT_FALSE(I.is_at_eof()); 80 EXPECT_NE(E, I); 81 82 EXPECT_EQ("line 1", *I); 83 EXPECT_EQ(4, I.line_number()); 84 ++I; 85 EXPECT_EQ("line 2", *I); 86 EXPECT_EQ(8, I.line_number()); 87 ++I; 88 89 EXPECT_TRUE(I.is_at_eof()); 90 EXPECT_EQ(E, I); 91 } 92 93 TEST(LineIteratorTest, EmptyBuffers) { 94 OwningPtr<MemoryBuffer> Buffer(MemoryBuffer::getMemBuffer("")); 95 EXPECT_TRUE(line_iterator(*Buffer).is_at_eof()); 96 EXPECT_EQ(line_iterator(), line_iterator(*Buffer)); 97 98 Buffer.reset(MemoryBuffer::getMemBuffer("\n\n\n")); 99 EXPECT_TRUE(line_iterator(*Buffer).is_at_eof()); 100 EXPECT_EQ(line_iterator(), line_iterator(*Buffer)); 101 102 Buffer.reset(MemoryBuffer::getMemBuffer("# foo\n" 103 "\n" 104 "# bar")); 105 EXPECT_TRUE(line_iterator(*Buffer, '#').is_at_eof()); 106 EXPECT_EQ(line_iterator(), line_iterator(*Buffer, '#')); 107 108 Buffer.reset(MemoryBuffer::getMemBuffer("\n" 109 "# baz\n" 110 "\n")); 111 EXPECT_TRUE(line_iterator(*Buffer, '#').is_at_eof()); 112 EXPECT_EQ(line_iterator(), line_iterator(*Buffer, '#')); 113 } 114 115 } // anonymous namespace 116