1 //===- unittest/Tooling/RefactoringCallbacksTest.cpp ----------------------===// 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 "clang/Tooling/RefactoringCallbacks.h" 11 #include "RewriterTestContext.h" 12 #include "clang/ASTMatchers/ASTMatchFinder.h" 13 #include "clang/ASTMatchers/ASTMatchers.h" 14 #include "gtest/gtest.h" 15 16 namespace clang { 17 namespace tooling { 18 19 using namespace ast_matchers; 20 21 template <typename T> 22 void expectRewritten(const std::string &Code, 23 const std::string &Expected, 24 const T &AMatcher, 25 RefactoringCallback &Callback) { 26 MatchFinder Finder; 27 Finder.addMatcher(AMatcher, &Callback); 28 OwningPtr<tooling::FrontendActionFactory> Factory( 29 tooling::newFrontendActionFactory(&Finder)); 30 ASSERT_TRUE(tooling::runToolOnCode(Factory->create(), Code)) 31 << "Parsing error in \"" << Code << "\""; 32 RewriterTestContext Context; 33 FileID ID = Context.createInMemoryFile("input.cc", Code); 34 EXPECT_TRUE(tooling::applyAllReplacements(Callback.getReplacements(), 35 Context.Rewrite)); 36 EXPECT_EQ(Expected, Context.getRewrittenText(ID)); 37 } 38 39 TEST(RefactoringCallbacksTest, ReplacesStmtsWithString) { 40 std::string Code = "void f() { int i = 1; }"; 41 std::string Expected = "void f() { ; }"; 42 ReplaceStmtWithText Callback("id", ";"); 43 expectRewritten(Code, Expected, id("id", declStmt()), Callback); 44 } 45 46 TEST(RefactoringCallbacksTest, ReplacesStmtsInCalledMacros) { 47 std::string Code = "#define A void f() { int i = 1; }\nA"; 48 std::string Expected = "#define A void f() { ; }\nA"; 49 ReplaceStmtWithText Callback("id", ";"); 50 expectRewritten(Code, Expected, id("id", declStmt()), Callback); 51 } 52 53 TEST(RefactoringCallbacksTest, IgnoresStmtsInUncalledMacros) { 54 std::string Code = "#define A void f() { int i = 1; }"; 55 std::string Expected = "#define A void f() { int i = 1; }"; 56 ReplaceStmtWithText Callback("id", ";"); 57 expectRewritten(Code, Expected, id("id", declStmt()), Callback); 58 } 59 60 TEST(RefactoringCallbacksTest, ReplacesInteger) { 61 std::string Code = "void f() { int i = 1; }"; 62 std::string Expected = "void f() { int i = 2; }"; 63 ReplaceStmtWithText Callback("id", "2"); 64 expectRewritten(Code, Expected, id("id", expr(integerLiteral())), 65 Callback); 66 } 67 68 TEST(RefactoringCallbacksTest, ReplacesStmtWithStmt) { 69 std::string Code = "void f() { int i = false ? 1 : i * 2; }"; 70 std::string Expected = "void f() { int i = i * 2; }"; 71 ReplaceStmtWithStmt Callback("always-false", "should-be"); 72 expectRewritten(Code, Expected, 73 id("always-false", conditionalOperator( 74 hasCondition(boolLiteral(equals(false))), 75 hasFalseExpression(id("should-be", expr())))), 76 Callback); 77 } 78 79 TEST(RefactoringCallbacksTest, ReplacesIfStmt) { 80 std::string Code = "bool a; void f() { if (a) f(); else a = true; }"; 81 std::string Expected = "bool a; void f() { f(); }"; 82 ReplaceIfStmtWithItsBody Callback("id", true); 83 expectRewritten(Code, Expected, 84 id("id", ifStmt( 85 hasCondition(implicitCastExpr(hasSourceExpression( 86 declRefExpr(to(varDecl(hasName("a"))))))))), 87 Callback); 88 } 89 90 TEST(RefactoringCallbacksTest, RemovesEntireIfOnEmptyElse) { 91 std::string Code = "void f() { if (false) int i = 0; }"; 92 std::string Expected = "void f() { }"; 93 ReplaceIfStmtWithItsBody Callback("id", false); 94 expectRewritten(Code, Expected, 95 id("id", ifStmt(hasCondition(boolLiteral(equals(false))))), 96 Callback); 97 } 98 99 } // end namespace ast_matchers 100 } // end namespace clang 101
