averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-21 12:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add some tests for SourceManager routines that compare SourceLocs and SourceRanges

Browse Source 
Swift SVN r7002
This commit is contained in:
Dmitri Hrybenko
2013-08-07 21:32:19 +00:00
parent 49778d9a4b
commit b1f6da2837
2 changed files with 88 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

87
unittests/Basic/SourceManager.cpp Normal file
View File

@@ -0,0 +1,87 @@
#include "swift/Basic/SourceManager.h"
#include "llvm/Support/MemoryBuffer.h"
#include "gtest/gtest.h"
#include <vector>
using namespace swift;
using namespace llvm;
static std::vector<SourceLoc> tokenize(SourceManager &SM, StringRef Source) {
MemoryBuffer *Buf = MemoryBuffer::getMemBuffer(Source);
SM->AddNewSourceBuffer(Buf, llvm::SMLoc());
SourceLoc BeginLoc(SMLoc::getFromPointer(Buf->getBuffer().begin()));
std::vector<SourceLoc> Result;
Result.push_back(BeginLoc);
for (unsigned i = 1, e = Source.size(); i != e; ++i) {
if (Source[i - 1] == ' ')
Result.push_back(BeginLoc.getAdvancedLoc(i));
}
return Result;
}
TEST(SourceManager, IsBeforeInBuffer) {
SourceManager SM;
auto Locs = tokenize(SM, "aaa bbb ccc ddd");
EXPECT_TRUE(SM.isBeforeInBuffer(Locs[0], Locs[1]));
EXPECT_TRUE(SM.isBeforeInBuffer(Locs[1], Locs[2]));
EXPECT_TRUE(SM.isBeforeInBuffer(Locs[2], Locs[3]));
EXPECT_TRUE(SM.isBeforeInBuffer(Locs[0], Locs[3]));
EXPECT_TRUE(SM.isBeforeInBuffer(Locs[0], Locs[0].getAdvancedLoc(1)));
EXPECT_TRUE(SM.isBeforeInBuffer(Locs[0].getAdvancedLoc(1), Locs[1]));
}
TEST(SourceManager, RangeContainsTokenLoc) {
SourceManager SM;
auto Locs = tokenize(SM, "aaa bbb ccc ddd");
SourceRange R_aa(Locs[0], Locs[0]);
SourceRange R_ab(Locs[0], Locs[1]);
SourceRange R_ac(Locs[0], Locs[2]);
SourceRange R_bc(Locs[1], Locs[2]);
EXPECT_TRUE(SM.rangeContainsTokenLoc(R_aa, Locs[0]));
EXPECT_FALSE(SM.rangeContainsTokenLoc(R_aa, Locs[0].getAdvancedLoc(1)));
EXPECT_TRUE(SM.rangeContainsTokenLoc(R_ab, Locs[0]));
EXPECT_TRUE(SM.rangeContainsTokenLoc(R_ab, Locs[0].getAdvancedLoc(1)));
EXPECT_TRUE(SM.rangeContainsTokenLoc(R_ab, Locs[1]));
EXPECT_FALSE(SM.rangeContainsTokenLoc(R_ab, Locs[1].getAdvancedLoc(1)));
EXPECT_TRUE(SM.rangeContainsTokenLoc(R_ac, Locs[0]));
EXPECT_TRUE(SM.rangeContainsTokenLoc(R_ac, Locs[0].getAdvancedLoc(1)));
EXPECT_TRUE(SM.rangeContainsTokenLoc(R_ac, Locs[1]));
EXPECT_TRUE(SM.rangeContainsTokenLoc(R_ac, Locs[1].getAdvancedLoc(1)));
EXPECT_TRUE(SM.rangeContainsTokenLoc(R_ac, Locs[2]));
EXPECT_FALSE(SM.rangeContainsTokenLoc(R_ac, Locs[2].getAdvancedLoc(1)));
EXPECT_FALSE(SM.rangeContainsTokenLoc(R_bc, Locs[0]));
EXPECT_FALSE(SM.rangeContainsTokenLoc(R_bc, Locs[0].getAdvancedLoc(1)));
}
TEST(SourceManager, RangeContains) {
SourceManager SM;
auto Locs = tokenize(SM, "aaa bbb ccc ddd");
SourceRange R_aa(Locs[0], Locs[0]);
SourceRange R_ab(Locs[0], Locs[1]);
SourceRange R_ac(Locs[0], Locs[2]);
SourceRange R_ad(Locs[0], Locs[3]);
SourceRange R_bc(Locs[1], Locs[2]);
EXPECT_TRUE(SM.rangeContains(R_ab, R_aa));
EXPECT_TRUE(SM.rangeContains(R_ac, R_aa));
EXPECT_TRUE(SM.rangeContains(R_ac, R_ab));
EXPECT_TRUE(SM.rangeContains(R_ad, R_ab));
EXPECT_TRUE(SM.rangeContains(R_ad, R_ac));
EXPECT_TRUE(SM.rangeContains(R_ac, R_bc));
EXPECT_TRUE(SM.rangeContains(R_ad, R_bc));
}