unittests/Support/SourceMgrTest.cpp

   1 //===- unittests/Support/SourceMgrTest.cpp - SourceMgr 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/SourceMgr.h"
  11 #include "llvm/Support/MemoryBuffer.h"
  12 #include "llvm/Support/raw_ostream.h"
  13 #include "gtest/gtest.h"
  14
  15 using namespace llvm;
  16
  17 namespace {
  18
  19 class SourceMgrTest : public testing::Test {
  20 public:
  21   SourceMgr SM;
  22   unsigned MainBufferID;
  23   std::string Output;
  24
  25   void setMainBuffer(StringRef Text, StringRef BufferName) {
  26     std::unique_ptr<MemoryBuffer> MainBuffer =
  27         MemoryBuffer::getMemBuffer(Text, BufferName);
  28     MainBufferID = SM.AddNewSourceBuffer(std::move(MainBuffer), llvm::SMLoc());
  29   }
  30
  31   SMLoc getLoc(unsigned Offset) {
  32     return SMLoc::getFromPointer(
  33         SM.getMemoryBuffer(MainBufferID)->getBufferStart() + Offset);
  34   }
  35
  36   SMRange getRange(unsigned Offset, unsigned Length) {
  37     return SMRange(getLoc(Offset), getLoc(Offset + Length));
  38   }
  39
  40   void printMessage(SMLoc Loc, SourceMgr::DiagKind Kind,
  41                     const Twine &Msg, ArrayRef<SMRange> Ranges,
  42                     ArrayRef<SMFixIt> FixIts) {
  43     raw_string_ostream OS(Output);
  44     SM.PrintMessage(OS, Loc, Kind, Msg, Ranges, FixIts);
  45   }
  46 };
  47
  48 } // unnamed namespace
  49
  50 TEST_F(SourceMgrTest, BasicError) {
  51   setMainBuffer("aaa bbb\nccc ddd\n", "file.in");
  52   printMessage(getLoc(4), SourceMgr::DK_Error, "message", None, None);
  53
  54   EXPECT_EQ("file.in:1:5: error: message\n"
  55             "aaa bbb\n"
  56             "    ^\n",
  57             Output);
  58 }
  59
  60 TEST_F(SourceMgrTest, BasicWarning) {
  61   setMainBuffer("aaa bbb\nccc ddd\n", "file.in");
  62   printMessage(getLoc(4), SourceMgr::DK_Warning, "message", None, None);
  63
  64   EXPECT_EQ("file.in:1:5: warning: message\n"
  65             "aaa bbb\n"
  66             "    ^\n",
  67             Output);
  68 }
  69
  70 TEST_F(SourceMgrTest, BasicNote) {
  71   setMainBuffer("aaa bbb\nccc ddd\n", "file.in");
  72   printMessage(getLoc(4), SourceMgr::DK_Note, "message", None, None);
  73
  74   EXPECT_EQ("file.in:1:5: note: message\n"
  75             "aaa bbb\n"
  76             "    ^\n",
  77             Output);
  78 }
  79
  80 TEST_F(SourceMgrTest, LocationAtEndOfLine) {
  81   setMainBuffer("aaa bbb\nccc ddd\n", "file.in");
  82   printMessage(getLoc(6), SourceMgr::DK_Error, "message", None, None);
  83
  84   EXPECT_EQ("file.in:1:7: error: message\n"
  85             "aaa bbb\n"
  86             "      ^\n",
  87             Output);
  88 }
  89
  90 TEST_F(SourceMgrTest, LocationAtNewline) {
  91   setMainBuffer("aaa bbb\nccc ddd\n", "file.in");
  92   printMessage(getLoc(7), SourceMgr::DK_Error, "message", None, None);
  93
  94   EXPECT_EQ("file.in:1:8: error: message\n"
  95             "aaa bbb\n"
  96             "       ^\n",
  97             Output);
  98 }
  99
 100 TEST_F(SourceMgrTest, BasicRange) {
 101   setMainBuffer("aaa bbb\nccc ddd\n", "file.in");
 102   printMessage(getLoc(4), SourceMgr::DK_Error, "message", getRange(4, 3), None);
 103
 104   EXPECT_EQ("file.in:1:5: error: message\n"
 105             "aaa bbb\n"
 106             "    ^~~\n",
 107             Output);
 108 }
 109
 110 TEST_F(SourceMgrTest, RangeWithTab) {
 111   setMainBuffer("aaa\tbbb\nccc ddd\n", "file.in");
 112   printMessage(getLoc(4), SourceMgr::DK_Error, "message", getRange(3, 3), None);
 113
 114   EXPECT_EQ("file.in:1:5: error: message\n"
 115             "aaa     bbb\n"
 116             "   ~~~~~^~\n",
 117             Output);
 118 }
 119
 120 TEST_F(SourceMgrTest, MultiLineRange) {
 121   setMainBuffer("aaa bbb\nccc ddd\n", "file.in");
 122   printMessage(getLoc(4), SourceMgr::DK_Error, "message", getRange(4, 7), None);
 123
 124   EXPECT_EQ("file.in:1:5: error: message\n"
 125             "aaa bbb\n"
 126             "    ^~~\n",
 127             Output);
 128 }
 129
 130 TEST_F(SourceMgrTest, MultipleRanges) {
 131   setMainBuffer("aaa bbb\nccc ddd\n", "file.in");
 132   SMRange Ranges[] = { getRange(0, 3), getRange(4, 3) };
 133   printMessage(getLoc(4), SourceMgr::DK_Error, "message", Ranges, None);
 134
 135   EXPECT_EQ("file.in:1:5: error: message\n"
 136             "aaa bbb\n"
 137             "~~~ ^~~\n",
 138             Output);
 139 }
 140
 141 TEST_F(SourceMgrTest, OverlappingRanges) {
 142   setMainBuffer("aaa bbb\nccc ddd\n", "file.in");
 143   SMRange Ranges[] = { getRange(0, 3), getRange(2, 4) };
 144   printMessage(getLoc(4), SourceMgr::DK_Error, "message", Ranges, None);
 145
 146   EXPECT_EQ("file.in:1:5: error: message\n"
 147             "aaa bbb\n"
 148             "~~~~^~\n",
 149             Output);
 150 }
 151
 152 TEST_F(SourceMgrTest, BasicFixit) {
 153   setMainBuffer("aaa bbb\nccc ddd\n", "file.in");
 154   printMessage(getLoc(4), SourceMgr::DK_Error, "message", None,
 155                makeArrayRef(SMFixIt(getRange(4, 3), "zzz")));
 156
 157   EXPECT_EQ("file.in:1:5: error: message\n"
 158             "aaa bbb\n"
 159             "    ^~~\n"
 160             "    zzz\n",
 161             Output);
 162 }
 163
 164 TEST_F(SourceMgrTest, FixitForTab) {
 165   setMainBuffer("aaa\tbbb\nccc ddd\n", "file.in");
 166   printMessage(getLoc(3), SourceMgr::DK_Error, "message", None,
 167                makeArrayRef(SMFixIt(getRange(3, 1), "zzz")));
 168
 169   EXPECT_EQ("file.in:1:4: error: message\n"
 170             "aaa     bbb\n"
 171             "   ^^^^^\n"
 172             "   zzz\n",
 173             Output);
 174 }
 175