#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"));
}
}