forked from OSchip/llvm-project
38 lines
1.4 KiB
C++
38 lines
1.4 KiB
C++
|
//===- unittest/Tooling/RewriterTest.cpp ----------------------------------===//
|
||
|
|
//
|
||
|
|
// The LLVM Compiler Infrastructure
|
||
|
|
//
|
||
|
|
// This file is distributed under the University of Illinois Open Source
|
||
|
|
// License. See LICENSE.TXT for details.
|
||
|
|
//
|
||
|
|
//===----------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
#include "RewriterTestContext.h"
|
||
|
|
#include "gtest/gtest.h"
|
||
|
|
|
||
|
|
namespace clang {
|
||
|
|
|
||
|
|
TEST(Rewriter, OverwritesChangedFiles) {
|
||
|
|
RewriterTestContext Context;
|
||
|
|
FileID ID = Context.createOnDiskFile("t.cpp", "line1\nline2\nline3\nline4");
|
||
|
|
Context.Rewrite.ReplaceText(Context.getLocation(ID, 2, 1), 5, "replaced");
|
||
|
|
EXPECT_FALSE(Context.Rewrite.overwriteChangedFiles());
|
||
|
|
EXPECT_EQ("line1\nreplaced\nline3\nline4",
|
||
|
|
Context.getFileContentFromDisk("t.cpp"));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(Rewriter, ContinuesOverwritingFilesOnError) {
|
||
|
|
RewriterTestContext Context;
|
||
|
|
FileID FailingID = Context.createInMemoryFile("invalid/failing.cpp", "test");
|
||
|
|
Context.Rewrite.ReplaceText(Context.getLocation(FailingID, 1, 2), 1, "other");
|
||
|
|
FileID WorkingID = Context.createOnDiskFile(
|
||
|
|
"working.cpp", "line1\nline2\nline3\nline4");
|
||
|
|
Context.Rewrite.ReplaceText(Context.getLocation(WorkingID, 2, 1), 5,
|
||
|
|
"replaced");
|
||
|
|
EXPECT_TRUE(Context.Rewrite.overwriteChangedFiles());
|
||
|
|
EXPECT_EQ("line1\nreplaced\nline3\nline4",
|
||
|
|
Context.getFileContentFromDisk("working.cpp"));
|
||
|
|
}
|
||
|
|
|
||
|
|
} // end namespace clang
|