80 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			80 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			C++
		
	
	
	
//===- unittests/Basic/LineOffsetMappingTest.cpp - Test LineOffsetMapping -===//
 | 
						|
//
 | 
						|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 | 
						|
// See https://llvm.org/LICENSE.txt for license information.
 | 
						|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 | 
						|
//
 | 
						|
//===----------------------------------------------------------------------===//
 | 
						|
 | 
						|
#include "clang/Basic/SourceManager.h"
 | 
						|
#include "gtest/gtest.h"
 | 
						|
 | 
						|
using namespace clang;
 | 
						|
using namespace clang::SrcMgr;
 | 
						|
using namespace llvm;
 | 
						|
 | 
						|
namespace {
 | 
						|
 | 
						|
TEST(LineOffsetMappingTest, empty) {
 | 
						|
  LineOffsetMapping Mapping;
 | 
						|
  EXPECT_FALSE(Mapping);
 | 
						|
 | 
						|
#if !defined(NDEBUG) && GTEST_HAS_DEATH_TEST
 | 
						|
  EXPECT_DEATH((void)Mapping.getLines(), "Storage");
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
TEST(LineOffsetMappingTest, construct) {
 | 
						|
  BumpPtrAllocator Alloc;
 | 
						|
  unsigned Offsets[] = {0, 10, 20};
 | 
						|
  LineOffsetMapping Mapping(Offsets, Alloc);
 | 
						|
  EXPECT_EQ(3u, Mapping.size());
 | 
						|
  EXPECT_EQ(0u, Mapping[0]);
 | 
						|
  EXPECT_EQ(10u, Mapping[1]);
 | 
						|
  EXPECT_EQ(20u, Mapping[2]);
 | 
						|
 | 
						|
#if !defined(NDEBUG) && GTEST_HAS_DEATH_TEST
 | 
						|
  EXPECT_DEATH((void)Mapping[3], "Invalid index");
 | 
						|
#endif
 | 
						|
}
 | 
						|
 | 
						|
TEST(LineOffsetMappingTest, constructTwo) {
 | 
						|
  // Confirm allocation size is big enough, convering an off-by-one bug.
 | 
						|
  BumpPtrAllocator Alloc;
 | 
						|
  unsigned Offsets1[] = {0, 10};
 | 
						|
  unsigned Offsets2[] = {0, 20};
 | 
						|
  LineOffsetMapping Mapping1(Offsets1, Alloc);
 | 
						|
  LineOffsetMapping Mapping2(Offsets2, Alloc);
 | 
						|
 | 
						|
  // Need to check Mapping1 *after* building Mapping2.
 | 
						|
  EXPECT_EQ(2u, Mapping1.size());
 | 
						|
  EXPECT_EQ(0u, Mapping1[0]);
 | 
						|
  EXPECT_EQ(10u, Mapping1[1]);
 | 
						|
  EXPECT_EQ(2u, Mapping2.size());
 | 
						|
  EXPECT_EQ(0u, Mapping2[0]);
 | 
						|
  EXPECT_EQ(20u, Mapping2[1]);
 | 
						|
}
 | 
						|
 | 
						|
TEST(LineOffsetMappingTest, get) {
 | 
						|
  BumpPtrAllocator Alloc;
 | 
						|
  StringRef Source = "first line\n"
 | 
						|
                     "second line\n";
 | 
						|
  auto Mapping = LineOffsetMapping::get(MemoryBufferRef(Source, ""), Alloc);
 | 
						|
  EXPECT_EQ(3u, Mapping.size());
 | 
						|
  EXPECT_EQ(0u, Mapping[0]);
 | 
						|
  EXPECT_EQ(11u, Mapping[1]);
 | 
						|
  EXPECT_EQ(23u, Mapping[2]);
 | 
						|
}
 | 
						|
 | 
						|
TEST(LineOffsetMappingTest, getMissingFinalNewline) {
 | 
						|
  BumpPtrAllocator Alloc;
 | 
						|
  StringRef Source = "first line\n"
 | 
						|
                     "second line";
 | 
						|
  auto Mapping = LineOffsetMapping::get(MemoryBufferRef(Source, ""), Alloc);
 | 
						|
  EXPECT_EQ(2u, Mapping.size());
 | 
						|
  EXPECT_EQ(0u, Mapping[0]);
 | 
						|
  EXPECT_EQ(11u, Mapping[1]);
 | 
						|
}
 | 
						|
 | 
						|
} // end namespace
 |