1*67e74705SXin Li //===- unittest/Tooling/RewriterTest.cpp ----------------------------------===//
2*67e74705SXin Li //
3*67e74705SXin Li // The LLVM Compiler Infrastructure
4*67e74705SXin Li //
5*67e74705SXin Li // This file is distributed under the University of Illinois Open Source
6*67e74705SXin Li // License. See LICENSE.TXT for details.
7*67e74705SXin Li //
8*67e74705SXin Li //===----------------------------------------------------------------------===//
9*67e74705SXin Li
10*67e74705SXin Li #include "RewriterTestContext.h"
11*67e74705SXin Li #include "clang/Tooling/Core/Replacement.h"
12*67e74705SXin Li #include "gtest/gtest.h"
13*67e74705SXin Li
14*67e74705SXin Li namespace clang {
15*67e74705SXin Li namespace tooling {
16*67e74705SXin Li namespace {
17*67e74705SXin Li
TEST(Rewriter,OverwritesChangedFiles)18*67e74705SXin Li TEST(Rewriter, OverwritesChangedFiles) {
19*67e74705SXin Li RewriterTestContext Context;
20*67e74705SXin Li FileID ID = Context.createOnDiskFile("t.cpp", "line1\nline2\nline3\nline4");
21*67e74705SXin Li Context.Rewrite.ReplaceText(Context.getLocation(ID, 2, 1), 5, "replaced");
22*67e74705SXin Li EXPECT_FALSE(Context.Rewrite.overwriteChangedFiles());
23*67e74705SXin Li EXPECT_EQ("line1\nreplaced\nline3\nline4",
24*67e74705SXin Li Context.getFileContentFromDisk("t.cpp"));
25*67e74705SXin Li }
26*67e74705SXin Li
TEST(Rewriter,ContinuesOverwritingFilesOnError)27*67e74705SXin Li TEST(Rewriter, ContinuesOverwritingFilesOnError) {
28*67e74705SXin Li RewriterTestContext Context;
29*67e74705SXin Li FileID FailingID = Context.createInMemoryFile("invalid/failing.cpp", "test");
30*67e74705SXin Li Context.Rewrite.ReplaceText(Context.getLocation(FailingID, 1, 2), 1, "other");
31*67e74705SXin Li FileID WorkingID = Context.createOnDiskFile(
32*67e74705SXin Li "working.cpp", "line1\nline2\nline3\nline4");
33*67e74705SXin Li Context.Rewrite.ReplaceText(Context.getLocation(WorkingID, 2, 1), 5,
34*67e74705SXin Li "replaced");
35*67e74705SXin Li EXPECT_TRUE(Context.Rewrite.overwriteChangedFiles());
36*67e74705SXin Li EXPECT_EQ("line1\nreplaced\nline3\nline4",
37*67e74705SXin Li Context.getFileContentFromDisk("working.cpp"));
38*67e74705SXin Li }
39*67e74705SXin Li
TEST(Rewriter,AdjacentInsertAndDelete)40*67e74705SXin Li TEST(Rewriter, AdjacentInsertAndDelete) {
41*67e74705SXin Li Replacements Replaces;
42*67e74705SXin Li Replaces.insert(Replacement("<file>", 6, 6, ""));
43*67e74705SXin Li Replaces.insert(Replacement("<file>", 6, 0, "replaced\n"));
44*67e74705SXin Li auto Rewritten = applyAllReplacements("line1\nline2\nline3\nline4", Replaces);
45*67e74705SXin Li EXPECT_TRUE(static_cast<bool>(Rewritten));
46*67e74705SXin Li EXPECT_EQ("line1\nreplaced\nline3\nline4", *Rewritten);
47*67e74705SXin Li }
48*67e74705SXin Li
49*67e74705SXin Li } // end namespace
50*67e74705SXin Li } // end namespace tooling
51*67e74705SXin Li } // end namespace clang
52