head 1.1; branch 1.1.1; access; symbols netbsd-11-0-RC4:1.1.1.2 netbsd-11-0-RC3:1.1.1.2 netbsd-11-0-RC2:1.1.1.2 netbsd-11-0-RC1:1.1.1.2 perseant-exfatfs-base-20250801:1.1.1.2 netbsd-11:1.1.1.2.0.10 netbsd-11-base:1.1.1.2 netbsd-10-1-RELEASE:1.1.1.2 perseant-exfatfs-base-20240630:1.1.1.2 perseant-exfatfs:1.1.1.2.0.8 perseant-exfatfs-base:1.1.1.2 netbsd-9-4-RELEASE:1.1.1.1 netbsd-10-0-RELEASE:1.1.1.2 netbsd-10-0-RC6:1.1.1.2 netbsd-10-0-RC5:1.1.1.2 netbsd-10-0-RC4:1.1.1.2 netbsd-10-0-RC3:1.1.1.2 netbsd-10-0-RC2:1.1.1.2 netbsd-10-0-RC1:1.1.1.2 netbsd-10:1.1.1.2.0.6 netbsd-10-base:1.1.1.2 netbsd-9-3-RELEASE:1.1.1.1 cjep_sun2x:1.1.1.2.0.4 cjep_sun2x-base:1.1.1.2 cjep_staticlib_x-base1:1.1.1.2 netbsd-9-2-RELEASE:1.1.1.1 cjep_staticlib_x:1.1.1.2.0.2 cjep_staticlib_x-base:1.1.1.2 netbsd-9-1-RELEASE:1.1.1.1 phil-wifi-20200421:1.1.1.2 phil-wifi-20200411:1.1.1.2 phil-wifi-20200406:1.1.1.2 netbsd-9-0-RELEASE:1.1.1.1 netbsd-9-0-RC2:1.1.1.1 netbsd-9-0-RC1:1.1.1.1 netbsd-9:1.1.1.1.0.6 netbsd-9-base:1.1.1.1 phil-wifi:1.1.1.1.0.4 phil-wifi-20190609:1.1.1.1 pgoyette-compat-merge-20190127:1.1.1.1.2.2 pgoyette-compat-20190127:1.1.1.1 pgoyette-compat-20190118:1.1.1.1 pgoyette-compat-1226:1.1.1.1 pgoyette-compat-1126:1.1.1.1 pgoyette-compat-1020:1.1.1.1 pgoyette-compat-0930:1.1.1.1 pgoyette-compat-0906:1.1.1.1 pgoyette-compat:1.1.1.1.0.2 pgoyette-compat-0728:1.1.1.1 clang-337282:1.1.1.1 LLVM:1.1.1; locks; strict; comment @// @; 1.1 date 2018.07.17.18.32.14; author joerg; state Exp; branches 1.1.1.1; next ; commitid wDzL46ALjrCZgwKA; 1.1.1.1 date 2018.07.17.18.32.14; author joerg; state Exp; branches 1.1.1.1.2.1 1.1.1.1.4.1; next 1.1.1.2; commitid wDzL46ALjrCZgwKA; 1.1.1.2 date 2019.11.13.22.23.11; author joerg; state dead; branches; next ; commitid QD8YATxuNG34YJKB; 1.1.1.1.2.1 date 2018.07.17.18.32.14; author pgoyette; state dead; branches; next 1.1.1.1.2.2; commitid 1UP1xAIUxv1ZgRLA; 1.1.1.1.2.2 date 2018.07.28.04.34.19; author pgoyette; state Exp; branches; next ; commitid 1UP1xAIUxv1ZgRLA; 1.1.1.1.4.1 date 2018.07.17.18.32.14; author christos; state dead; branches; next 1.1.1.1.4.2; commitid jtc8rnCzWiEEHGqB; 1.1.1.1.4.2 date 2019.06.10.21.46.47; author christos; state Exp; branches; next 1.1.1.1.4.3; commitid jtc8rnCzWiEEHGqB; 1.1.1.1.4.3 date 2020.04.13.07.50.39; author martin; state dead; branches; next ; commitid X01YhRUPVUDaec4C; desc @@ 1.1 log @Initial revision @ text @//===- unittest/Tooling/CleanupTest.cpp - Include insertion/deletion tests ===// // // The LLVM Compiler Infrastructure // // This file is distributed under the University of Illinois Open Source // License. See LICENSE.TXT for details. // //===----------------------------------------------------------------------===// #include "clang/Tooling/Inclusions/HeaderIncludes.h" #include "../Tooling/ReplacementTest.h" #include "../Tooling/RewriterTestContext.h" #include "clang/Format/Format.h" #include "clang/Tooling/Core/Replacement.h" #include "gtest/gtest.h" using clang::tooling::ReplacementTest; using clang::tooling::toReplacements; namespace clang { namespace tooling { namespace { class HeaderIncludesTest : public ::testing::Test { protected: std::string insert(llvm::StringRef Code, llvm::StringRef Header) { HeaderIncludes Includes(FileName, Code, Style); assert(Header.startswith("\"") || Header.startswith("<")); auto R = Includes.insert(Header.trim("\"<>"), Header.startswith("<")); if (!R) return Code; auto Result = applyAllReplacements(Code, Replacements(*R)); EXPECT_TRUE(static_cast(Result)); return *Result; } std::string remove(llvm::StringRef Code, llvm::StringRef Header) { HeaderIncludes Includes(FileName, Code, Style); assert(Header.startswith("\"") || Header.startswith("<")); auto Replaces = Includes.remove(Header.trim("\"<>"), Header.startswith("<")); auto Result = applyAllReplacements(Code, Replaces); EXPECT_TRUE(static_cast(Result)); return *Result; } const std::string FileName = "fix.cpp"; IncludeStyle Style = format::getLLVMStyle().IncludeStyle; }; TEST_F(HeaderIncludesTest, NoExistingIncludeWithoutDefine) { std::string Code = "int main() {}"; std::string Expected = "#include \"a.h\"\n" "int main() {}"; EXPECT_EQ(Expected, insert(Code, "\"a.h\"")); } TEST_F(HeaderIncludesTest, NoExistingIncludeWithDefine) { std::string Code = "#ifndef A_H\n" "#define A_H\n" "class A {};\n" "#define MMM 123\n" "#endif"; std::string Expected = "#ifndef A_H\n" "#define A_H\n" "#include \"b.h\"\n" "class A {};\n" "#define MMM 123\n" "#endif"; EXPECT_EQ(Expected, insert(Code, "\"b.h\"")); } TEST_F(HeaderIncludesTest, InsertBeforeCategoryWithLowerPriority) { std::string Code = "#ifndef A_H\n" "#define A_H\n" "\n" "\n" "\n" "#include \n" "class A {};\n" "#define MMM 123\n" "#endif"; std::string Expected = "#ifndef A_H\n" "#define A_H\n" "\n" "\n" "\n" "#include \"a.h\"\n" "#include \n" "class A {};\n" "#define MMM 123\n" "#endif"; EXPECT_EQ(Expected, insert(Code, "\"a.h\"")); } TEST_F(HeaderIncludesTest, InsertAfterMainHeader) { std::string Code = "#include \"fix.h\"\n" "\n" "int main() {}"; std::string Expected = "#include \"fix.h\"\n" "#include \n" "\n" "int main() {}"; Style = format::getGoogleStyle(format::FormatStyle::LanguageKind::LK_Cpp) .IncludeStyle; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, InsertBeforeSystemHeaderLLVM) { std::string Code = "#include \n" "\n" "int main() {}"; std::string Expected = "#include \"z.h\"\n" "#include \n" "\n" "int main() {}"; EXPECT_EQ(Expected, insert(Code, "\"z.h\"")); } TEST_F(HeaderIncludesTest, InsertAfterSystemHeaderGoogle) { std::string Code = "#include \n" "\n" "int main() {}"; std::string Expected = "#include \n" "#include \"z.h\"\n" "\n" "int main() {}"; Style = format::getGoogleStyle(format::FormatStyle::LanguageKind::LK_Cpp) .IncludeStyle; EXPECT_EQ(Expected, insert(Code, "\"z.h\"")); } TEST_F(HeaderIncludesTest, InsertOneIncludeLLVMStyle) { std::string Code = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"b.h\"\n" "#include \"clang/Format/Format.h\"\n" "#include \n"; std::string Expected = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"b.h\"\n" "#include \"clang/Format/Format.h\"\n" "#include \"llvm/x/y.h\"\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "\"llvm/x/y.h\"")); } TEST_F(HeaderIncludesTest, InsertIntoBlockSorted) { std::string Code = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"c.h\"\n" "#include \n"; std::string Expected = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"b.h\"\n" "#include \"c.h\"\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "\"b.h\"")); } TEST_F(HeaderIncludesTest, InsertIntoFirstBlockOfSameKind) { std::string Code = "#include \"x/fix.h\"\n" "#include \"c.h\"\n" "#include \"e.h\"\n" "#include \"f.h\"\n" "#include \n" "#include \n" "#include \"m.h\"\n" "#include \"n.h\"\n"; std::string Expected = "#include \"x/fix.h\"\n" "#include \"c.h\"\n" "#include \"d.h\"\n" "#include \"e.h\"\n" "#include \"f.h\"\n" "#include \n" "#include \n" "#include \"m.h\"\n" "#include \"n.h\"\n"; EXPECT_EQ(Expected, insert(Code, "\"d.h\"")); } TEST_F(HeaderIncludesTest, InsertIntoSystemBlockSorted) { std::string Code = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"c.h\"\n" "#include \n" "#include \n"; std::string Expected = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"c.h\"\n" "#include \n" "#include \n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, InsertNewSystemIncludeGoogleStyle) { std::string Code = "#include \"x/fix.h\"\n" "\n" "#include \"y/a.h\"\n" "#include \"z/b.h\"\n"; // FIXME: inserting after the empty line following the main header might be // preferred. std::string Expected = "#include \"x/fix.h\"\n" "#include \n" "\n" "#include \"y/a.h\"\n" "#include \"z/b.h\"\n"; Style = format::getGoogleStyle(format::FormatStyle::LanguageKind::LK_Cpp) .IncludeStyle; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, NotConfusedByDefine) { std::string Code = "void f() {}\n" "#define A \\\n" " int i;"; std::string Expected = "#include \n" "void f() {}\n" "#define A \\\n" " int i;"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, SkippedTopComment) { std::string Code = "// comment\n" "\n" " // comment\n"; std::string Expected = "// comment\n" "\n" " // comment\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, SkippedMixedComments) { std::string Code = "// comment\n" "// comment \\\n" " comment continued\n" "/*\n" "* comment\n" "*/\n"; std::string Expected = "// comment\n" "// comment \\\n" " comment continued\n" "/*\n" "* comment\n" "*/\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, MultipleBlockCommentsInOneLine) { std::string Code = "/*\n" "* comment\n" "*/ /* comment\n" "*/\n" "\n\n" "/* c1 */ /*c2 */\n"; std::string Expected = "/*\n" "* comment\n" "*/ /* comment\n" "*/\n" "\n\n" "/* c1 */ /*c2 */\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, CodeAfterComments) { std::string Code = "/*\n" "* comment\n" "*/ /* comment\n" "*/\n" "\n\n" "/* c1 */ /*c2 */\n" "\n" "int x;\n"; std::string Expected = "/*\n" "* comment\n" "*/ /* comment\n" "*/\n" "\n\n" "/* c1 */ /*c2 */\n" "\n" "#include \n" "int x;\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, FakeHeaderGuardIfDef) { std::string Code = "// comment \n" "#ifdef X\n" "#define X\n"; std::string Expected = "// comment \n" "#include \n" "#ifdef X\n" "#define X\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, RealHeaderGuardAfterComments) { std::string Code = "// comment \n" "#ifndef X\n" "#define X\n" "int x;\n" "#define Y 1\n"; std::string Expected = "// comment \n" "#ifndef X\n" "#define X\n" "#include \n" "int x;\n" "#define Y 1\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, IfNDefWithNoDefine) { std::string Code = "// comment \n" "#ifndef X\n" "int x;\n" "#define Y 1\n"; std::string Expected = "// comment \n" "#include \n" "#ifndef X\n" "int x;\n" "#define Y 1\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, FakeHeaderGuard) { std::string Code = "// comment \n" "#ifndef X\n" "#define 1\n"; std::string Expected = "// comment \n" "#include \n" "#ifndef X\n" "#define 1\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, HeaderGuardWithComment) { std::string Code = "// comment \n" "#ifndef X // comment\n" "// comment\n" "/* comment\n" "*/\n" "/* comment */ #define X\n" "int x;\n" "#define Y 1\n"; std::string Expected = "// comment \n" "#ifndef X // comment\n" "// comment\n" "/* comment\n" "*/\n" "/* comment */ #define X\n" "#include \n" "int x;\n" "#define Y 1\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, EmptyCode) { std::string Code = ""; std::string Expected = "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, NoNewLineAtTheEndOfCode) { std::string Code = "#include "; std::string Expected = "#include \n#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, SkipExistingHeaders) { std::string Code = "#include \"a.h\"\n" "#include \n"; std::string Expected = "#include \"a.h\"\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); EXPECT_EQ(Expected, insert(Code, "\"a.h\"")); } TEST_F(HeaderIncludesTest, AddIncludesWithDifferentForms) { std::string Code = "#include \n"; // FIXME: this might not be the best behavior. std::string Expected = "#include \"vector\"\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "\"vector\"")); } TEST_F(HeaderIncludesTest, NoInsertionAfterCode) { std::string Code = "#include \"a.h\"\n" "void f() {}\n" "#include \"b.h\"\n"; std::string Expected = "#include \"a.h\"\n" "#include \"c.h\"\n" "void f() {}\n" "#include \"b.h\"\n"; EXPECT_EQ(Expected, insert(Code, "\"c.h\"")); } TEST_F(HeaderIncludesTest, NoInsertionInStringLiteral) { std::string Code = "#include \"a.h\"\n" "const char[] = R\"(\n" "#include \"b.h\"\n" ")\";\n"; std::string Expected = "#include \"a.h\"\n" "#include \"c.h\"\n" "const char[] = R\"(\n" "#include \"b.h\"\n" ")\";\n"; EXPECT_EQ(Expected, insert(Code, "\"c.h\"")); } TEST_F(HeaderIncludesTest, NoInsertionAfterOtherDirective) { std::string Code = "#include \"a.h\"\n" "#ifdef X\n" "#include \"b.h\"\n" "#endif\n"; std::string Expected = "#include \"a.h\"\n" "#include \"c.h\"\n" "#ifdef X\n" "#include \"b.h\"\n" "#endif\n"; EXPECT_EQ(Expected, insert(Code, "\"c.h\"")); } TEST_F(HeaderIncludesTest, CanInsertAfterLongSystemInclude) { std::string Code = "#include \"a.h\"\n" "// comment\n\n" "#include \n"; std::string Expected = "#include \"a.h\"\n" "// comment\n\n" "#include \n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, CanInsertAfterComment) { std::string Code = "#include \"a.h\"\n" "// Comment\n" "\n" "/* Comment */\n" "// Comment\n" "\n" "#include \"b.h\"\n"; std::string Expected = "#include \"a.h\"\n" "// Comment\n" "\n" "/* Comment */\n" "// Comment\n" "\n" "#include \"b.h\"\n" "#include \"c.h\"\n"; EXPECT_EQ(Expected, insert(Code, "\"c.h\"")); } TEST_F(HeaderIncludesTest, LongCommentsInTheBeginningOfFile) { std::string Code = "// Loooooooooooooooooooooooooong comment\n" "// Loooooooooooooooooooooooooong comment\n" "// Loooooooooooooooooooooooooong comment\n" "#include \n" "#include \n" "\n" "#include \"a.h\"\n" "#include \"b.h\"\n"; std::string Expected = "// Loooooooooooooooooooooooooong comment\n" "// Loooooooooooooooooooooooooong comment\n" "// Loooooooooooooooooooooooooong comment\n" "#include \n" "#include \n" "\n" "#include \"a.h\"\n" "#include \"b.h\"\n" "#include \"third.h\"\n"; Style = format::getGoogleStyle(format::FormatStyle::LanguageKind::LK_Cpp) .IncludeStyle; EXPECT_EQ(Expected, insert(Code, "\"third.h\"")); } TEST_F(HeaderIncludesTest, SimpleDeleteInclude) { std::string Code = "#include \"abc.h\"\n" "#include \"xyz.h\" // comment\n" "int x;\n"; std::string Expected = "#include \"abc.h\"\n" "int x;\n"; EXPECT_EQ(Expected, remove(Code, "\"xyz.h\"")); } TEST_F(HeaderIncludesTest, DeleteQuotedOnly) { std::string Code = "#include \"abc.h\"\n" "#include \n" "int x;\n"; std::string Expected = "#include \n" "int x;\n"; EXPECT_EQ(Expected, remove(Code, "\"abc.h\"")); } TEST_F(HeaderIncludesTest, DeleteAllCode) { std::string Code = "#include \"xyz.h\"\n"; std::string Expected = ""; EXPECT_EQ(Expected, remove(Code, "\"xyz.h\"")); } TEST_F(HeaderIncludesTest, DeleteOnlyIncludesWithSameQuote) { std::string Code = "#include \"xyz.h\"\n" "#include \"xyz\"\n" "#include \n"; std::string Expected = "#include \"xyz.h\"\n" "#include \"xyz\"\n"; EXPECT_EQ(Expected, remove(Code, "")); } TEST_F(HeaderIncludesTest, CanDeleteAfterCode) { std::string Code = "#include \"a.h\"\n" "void f() {}\n" "#include \"b.h\"\n"; std::string Expected = "#include \"a.h\"\n" "void f() {}\n"; EXPECT_EQ(Expected, remove(Code, "\"b.h\"")); } } // namespace } // namespace tooling } // namespace clang @ 1.1.1.1 log @Import clang r337282 from trunk @ text @@ 1.1.1.2 log @Mark old LLVM instance as dead. @ text @@ 1.1.1.1.4.1 log @file HeaderIncludesTest.cpp was added on branch phil-wifi on 2019-06-10 21:46:47 +0000 @ text @d1 527 @ 1.1.1.1.4.2 log @Sync with HEAD @ text @a0 527 //===- unittest/Tooling/CleanupTest.cpp - Include insertion/deletion tests ===// // // The LLVM Compiler Infrastructure // // This file is distributed under the University of Illinois Open Source // License. See LICENSE.TXT for details. // //===----------------------------------------------------------------------===// #include "clang/Tooling/Inclusions/HeaderIncludes.h" #include "../Tooling/ReplacementTest.h" #include "../Tooling/RewriterTestContext.h" #include "clang/Format/Format.h" #include "clang/Tooling/Core/Replacement.h" #include "gtest/gtest.h" using clang::tooling::ReplacementTest; using clang::tooling::toReplacements; namespace clang { namespace tooling { namespace { class HeaderIncludesTest : public ::testing::Test { protected: std::string insert(llvm::StringRef Code, llvm::StringRef Header) { HeaderIncludes Includes(FileName, Code, Style); assert(Header.startswith("\"") || Header.startswith("<")); auto R = Includes.insert(Header.trim("\"<>"), Header.startswith("<")); if (!R) return Code; auto Result = applyAllReplacements(Code, Replacements(*R)); EXPECT_TRUE(static_cast(Result)); return *Result; } std::string remove(llvm::StringRef Code, llvm::StringRef Header) { HeaderIncludes Includes(FileName, Code, Style); assert(Header.startswith("\"") || Header.startswith("<")); auto Replaces = Includes.remove(Header.trim("\"<>"), Header.startswith("<")); auto Result = applyAllReplacements(Code, Replaces); EXPECT_TRUE(static_cast(Result)); return *Result; } const std::string FileName = "fix.cpp"; IncludeStyle Style = format::getLLVMStyle().IncludeStyle; }; TEST_F(HeaderIncludesTest, NoExistingIncludeWithoutDefine) { std::string Code = "int main() {}"; std::string Expected = "#include \"a.h\"\n" "int main() {}"; EXPECT_EQ(Expected, insert(Code, "\"a.h\"")); } TEST_F(HeaderIncludesTest, NoExistingIncludeWithDefine) { std::string Code = "#ifndef A_H\n" "#define A_H\n" "class A {};\n" "#define MMM 123\n" "#endif"; std::string Expected = "#ifndef A_H\n" "#define A_H\n" "#include \"b.h\"\n" "class A {};\n" "#define MMM 123\n" "#endif"; EXPECT_EQ(Expected, insert(Code, "\"b.h\"")); } TEST_F(HeaderIncludesTest, InsertBeforeCategoryWithLowerPriority) { std::string Code = "#ifndef A_H\n" "#define A_H\n" "\n" "\n" "\n" "#include \n" "class A {};\n" "#define MMM 123\n" "#endif"; std::string Expected = "#ifndef A_H\n" "#define A_H\n" "\n" "\n" "\n" "#include \"a.h\"\n" "#include \n" "class A {};\n" "#define MMM 123\n" "#endif"; EXPECT_EQ(Expected, insert(Code, "\"a.h\"")); } TEST_F(HeaderIncludesTest, InsertAfterMainHeader) { std::string Code = "#include \"fix.h\"\n" "\n" "int main() {}"; std::string Expected = "#include \"fix.h\"\n" "#include \n" "\n" "int main() {}"; Style = format::getGoogleStyle(format::FormatStyle::LanguageKind::LK_Cpp) .IncludeStyle; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, InsertBeforeSystemHeaderLLVM) { std::string Code = "#include \n" "\n" "int main() {}"; std::string Expected = "#include \"z.h\"\n" "#include \n" "\n" "int main() {}"; EXPECT_EQ(Expected, insert(Code, "\"z.h\"")); } TEST_F(HeaderIncludesTest, InsertAfterSystemHeaderGoogle) { std::string Code = "#include \n" "\n" "int main() {}"; std::string Expected = "#include \n" "#include \"z.h\"\n" "\n" "int main() {}"; Style = format::getGoogleStyle(format::FormatStyle::LanguageKind::LK_Cpp) .IncludeStyle; EXPECT_EQ(Expected, insert(Code, "\"z.h\"")); } TEST_F(HeaderIncludesTest, InsertOneIncludeLLVMStyle) { std::string Code = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"b.h\"\n" "#include \"clang/Format/Format.h\"\n" "#include \n"; std::string Expected = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"b.h\"\n" "#include \"clang/Format/Format.h\"\n" "#include \"llvm/x/y.h\"\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "\"llvm/x/y.h\"")); } TEST_F(HeaderIncludesTest, InsertIntoBlockSorted) { std::string Code = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"c.h\"\n" "#include \n"; std::string Expected = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"b.h\"\n" "#include \"c.h\"\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "\"b.h\"")); } TEST_F(HeaderIncludesTest, InsertIntoFirstBlockOfSameKind) { std::string Code = "#include \"x/fix.h\"\n" "#include \"c.h\"\n" "#include \"e.h\"\n" "#include \"f.h\"\n" "#include \n" "#include \n" "#include \"m.h\"\n" "#include \"n.h\"\n"; std::string Expected = "#include \"x/fix.h\"\n" "#include \"c.h\"\n" "#include \"d.h\"\n" "#include \"e.h\"\n" "#include \"f.h\"\n" "#include \n" "#include \n" "#include \"m.h\"\n" "#include \"n.h\"\n"; EXPECT_EQ(Expected, insert(Code, "\"d.h\"")); } TEST_F(HeaderIncludesTest, InsertIntoSystemBlockSorted) { std::string Code = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"c.h\"\n" "#include \n" "#include \n"; std::string Expected = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"c.h\"\n" "#include \n" "#include \n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, InsertNewSystemIncludeGoogleStyle) { std::string Code = "#include \"x/fix.h\"\n" "\n" "#include \"y/a.h\"\n" "#include \"z/b.h\"\n"; // FIXME: inserting after the empty line following the main header might be // preferred. std::string Expected = "#include \"x/fix.h\"\n" "#include \n" "\n" "#include \"y/a.h\"\n" "#include \"z/b.h\"\n"; Style = format::getGoogleStyle(format::FormatStyle::LanguageKind::LK_Cpp) .IncludeStyle; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, NotConfusedByDefine) { std::string Code = "void f() {}\n" "#define A \\\n" " int i;"; std::string Expected = "#include \n" "void f() {}\n" "#define A \\\n" " int i;"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, SkippedTopComment) { std::string Code = "// comment\n" "\n" " // comment\n"; std::string Expected = "// comment\n" "\n" " // comment\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, SkippedMixedComments) { std::string Code = "// comment\n" "// comment \\\n" " comment continued\n" "/*\n" "* comment\n" "*/\n"; std::string Expected = "// comment\n" "// comment \\\n" " comment continued\n" "/*\n" "* comment\n" "*/\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, MultipleBlockCommentsInOneLine) { std::string Code = "/*\n" "* comment\n" "*/ /* comment\n" "*/\n" "\n\n" "/* c1 */ /*c2 */\n"; std::string Expected = "/*\n" "* comment\n" "*/ /* comment\n" "*/\n" "\n\n" "/* c1 */ /*c2 */\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, CodeAfterComments) { std::string Code = "/*\n" "* comment\n" "*/ /* comment\n" "*/\n" "\n\n" "/* c1 */ /*c2 */\n" "\n" "int x;\n"; std::string Expected = "/*\n" "* comment\n" "*/ /* comment\n" "*/\n" "\n\n" "/* c1 */ /*c2 */\n" "\n" "#include \n" "int x;\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, FakeHeaderGuardIfDef) { std::string Code = "// comment \n" "#ifdef X\n" "#define X\n"; std::string Expected = "// comment \n" "#include \n" "#ifdef X\n" "#define X\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, RealHeaderGuardAfterComments) { std::string Code = "// comment \n" "#ifndef X\n" "#define X\n" "int x;\n" "#define Y 1\n"; std::string Expected = "// comment \n" "#ifndef X\n" "#define X\n" "#include \n" "int x;\n" "#define Y 1\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, IfNDefWithNoDefine) { std::string Code = "// comment \n" "#ifndef X\n" "int x;\n" "#define Y 1\n"; std::string Expected = "// comment \n" "#include \n" "#ifndef X\n" "int x;\n" "#define Y 1\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, FakeHeaderGuard) { std::string Code = "// comment \n" "#ifndef X\n" "#define 1\n"; std::string Expected = "// comment \n" "#include \n" "#ifndef X\n" "#define 1\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, HeaderGuardWithComment) { std::string Code = "// comment \n" "#ifndef X // comment\n" "// comment\n" "/* comment\n" "*/\n" "/* comment */ #define X\n" "int x;\n" "#define Y 1\n"; std::string Expected = "// comment \n" "#ifndef X // comment\n" "// comment\n" "/* comment\n" "*/\n" "/* comment */ #define X\n" "#include \n" "int x;\n" "#define Y 1\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, EmptyCode) { std::string Code = ""; std::string Expected = "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, NoNewLineAtTheEndOfCode) { std::string Code = "#include "; std::string Expected = "#include \n#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, SkipExistingHeaders) { std::string Code = "#include \"a.h\"\n" "#include \n"; std::string Expected = "#include \"a.h\"\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); EXPECT_EQ(Expected, insert(Code, "\"a.h\"")); } TEST_F(HeaderIncludesTest, AddIncludesWithDifferentForms) { std::string Code = "#include \n"; // FIXME: this might not be the best behavior. std::string Expected = "#include \"vector\"\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "\"vector\"")); } TEST_F(HeaderIncludesTest, NoInsertionAfterCode) { std::string Code = "#include \"a.h\"\n" "void f() {}\n" "#include \"b.h\"\n"; std::string Expected = "#include \"a.h\"\n" "#include \"c.h\"\n" "void f() {}\n" "#include \"b.h\"\n"; EXPECT_EQ(Expected, insert(Code, "\"c.h\"")); } TEST_F(HeaderIncludesTest, NoInsertionInStringLiteral) { std::string Code = "#include \"a.h\"\n" "const char[] = R\"(\n" "#include \"b.h\"\n" ")\";\n"; std::string Expected = "#include \"a.h\"\n" "#include \"c.h\"\n" "const char[] = R\"(\n" "#include \"b.h\"\n" ")\";\n"; EXPECT_EQ(Expected, insert(Code, "\"c.h\"")); } TEST_F(HeaderIncludesTest, NoInsertionAfterOtherDirective) { std::string Code = "#include \"a.h\"\n" "#ifdef X\n" "#include \"b.h\"\n" "#endif\n"; std::string Expected = "#include \"a.h\"\n" "#include \"c.h\"\n" "#ifdef X\n" "#include \"b.h\"\n" "#endif\n"; EXPECT_EQ(Expected, insert(Code, "\"c.h\"")); } TEST_F(HeaderIncludesTest, CanInsertAfterLongSystemInclude) { std::string Code = "#include \"a.h\"\n" "// comment\n\n" "#include \n"; std::string Expected = "#include \"a.h\"\n" "// comment\n\n" "#include \n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, CanInsertAfterComment) { std::string Code = "#include \"a.h\"\n" "// Comment\n" "\n" "/* Comment */\n" "// Comment\n" "\n" "#include \"b.h\"\n"; std::string Expected = "#include \"a.h\"\n" "// Comment\n" "\n" "/* Comment */\n" "// Comment\n" "\n" "#include \"b.h\"\n" "#include \"c.h\"\n"; EXPECT_EQ(Expected, insert(Code, "\"c.h\"")); } TEST_F(HeaderIncludesTest, LongCommentsInTheBeginningOfFile) { std::string Code = "// Loooooooooooooooooooooooooong comment\n" "// Loooooooooooooooooooooooooong comment\n" "// Loooooooooooooooooooooooooong comment\n" "#include \n" "#include \n" "\n" "#include \"a.h\"\n" "#include \"b.h\"\n"; std::string Expected = "// Loooooooooooooooooooooooooong comment\n" "// Loooooooooooooooooooooooooong comment\n" "// Loooooooooooooooooooooooooong comment\n" "#include \n" "#include \n" "\n" "#include \"a.h\"\n" "#include \"b.h\"\n" "#include \"third.h\"\n"; Style = format::getGoogleStyle(format::FormatStyle::LanguageKind::LK_Cpp) .IncludeStyle; EXPECT_EQ(Expected, insert(Code, "\"third.h\"")); } TEST_F(HeaderIncludesTest, SimpleDeleteInclude) { std::string Code = "#include \"abc.h\"\n" "#include \"xyz.h\" // comment\n" "int x;\n"; std::string Expected = "#include \"abc.h\"\n" "int x;\n"; EXPECT_EQ(Expected, remove(Code, "\"xyz.h\"")); } TEST_F(HeaderIncludesTest, DeleteQuotedOnly) { std::string Code = "#include \"abc.h\"\n" "#include \n" "int x;\n"; std::string Expected = "#include \n" "int x;\n"; EXPECT_EQ(Expected, remove(Code, "\"abc.h\"")); } TEST_F(HeaderIncludesTest, DeleteAllCode) { std::string Code = "#include \"xyz.h\"\n"; std::string Expected = ""; EXPECT_EQ(Expected, remove(Code, "\"xyz.h\"")); } TEST_F(HeaderIncludesTest, DeleteOnlyIncludesWithSameQuote) { std::string Code = "#include \"xyz.h\"\n" "#include \"xyz\"\n" "#include \n"; std::string Expected = "#include \"xyz.h\"\n" "#include \"xyz\"\n"; EXPECT_EQ(Expected, remove(Code, "")); } TEST_F(HeaderIncludesTest, CanDeleteAfterCode) { std::string Code = "#include \"a.h\"\n" "void f() {}\n" "#include \"b.h\"\n"; std::string Expected = "#include \"a.h\"\n" "void f() {}\n"; EXPECT_EQ(Expected, remove(Code, "\"b.h\"")); } } // namespace } // namespace tooling } // namespace clang @ 1.1.1.1.4.3 log @Mostly merge changes from HEAD upto 20200411 @ text @@ 1.1.1.1.2.1 log @file HeaderIncludesTest.cpp was added on branch pgoyette-compat on 2018-07-28 04:34:19 +0000 @ text @d1 527 @ 1.1.1.1.2.2 log @Sync with HEAD @ text @a0 527 //===- unittest/Tooling/CleanupTest.cpp - Include insertion/deletion tests ===// // // The LLVM Compiler Infrastructure // // This file is distributed under the University of Illinois Open Source // License. See LICENSE.TXT for details. // //===----------------------------------------------------------------------===// #include "clang/Tooling/Inclusions/HeaderIncludes.h" #include "../Tooling/ReplacementTest.h" #include "../Tooling/RewriterTestContext.h" #include "clang/Format/Format.h" #include "clang/Tooling/Core/Replacement.h" #include "gtest/gtest.h" using clang::tooling::ReplacementTest; using clang::tooling::toReplacements; namespace clang { namespace tooling { namespace { class HeaderIncludesTest : public ::testing::Test { protected: std::string insert(llvm::StringRef Code, llvm::StringRef Header) { HeaderIncludes Includes(FileName, Code, Style); assert(Header.startswith("\"") || Header.startswith("<")); auto R = Includes.insert(Header.trim("\"<>"), Header.startswith("<")); if (!R) return Code; auto Result = applyAllReplacements(Code, Replacements(*R)); EXPECT_TRUE(static_cast(Result)); return *Result; } std::string remove(llvm::StringRef Code, llvm::StringRef Header) { HeaderIncludes Includes(FileName, Code, Style); assert(Header.startswith("\"") || Header.startswith("<")); auto Replaces = Includes.remove(Header.trim("\"<>"), Header.startswith("<")); auto Result = applyAllReplacements(Code, Replaces); EXPECT_TRUE(static_cast(Result)); return *Result; } const std::string FileName = "fix.cpp"; IncludeStyle Style = format::getLLVMStyle().IncludeStyle; }; TEST_F(HeaderIncludesTest, NoExistingIncludeWithoutDefine) { std::string Code = "int main() {}"; std::string Expected = "#include \"a.h\"\n" "int main() {}"; EXPECT_EQ(Expected, insert(Code, "\"a.h\"")); } TEST_F(HeaderIncludesTest, NoExistingIncludeWithDefine) { std::string Code = "#ifndef A_H\n" "#define A_H\n" "class A {};\n" "#define MMM 123\n" "#endif"; std::string Expected = "#ifndef A_H\n" "#define A_H\n" "#include \"b.h\"\n" "class A {};\n" "#define MMM 123\n" "#endif"; EXPECT_EQ(Expected, insert(Code, "\"b.h\"")); } TEST_F(HeaderIncludesTest, InsertBeforeCategoryWithLowerPriority) { std::string Code = "#ifndef A_H\n" "#define A_H\n" "\n" "\n" "\n" "#include \n" "class A {};\n" "#define MMM 123\n" "#endif"; std::string Expected = "#ifndef A_H\n" "#define A_H\n" "\n" "\n" "\n" "#include \"a.h\"\n" "#include \n" "class A {};\n" "#define MMM 123\n" "#endif"; EXPECT_EQ(Expected, insert(Code, "\"a.h\"")); } TEST_F(HeaderIncludesTest, InsertAfterMainHeader) { std::string Code = "#include \"fix.h\"\n" "\n" "int main() {}"; std::string Expected = "#include \"fix.h\"\n" "#include \n" "\n" "int main() {}"; Style = format::getGoogleStyle(format::FormatStyle::LanguageKind::LK_Cpp) .IncludeStyle; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, InsertBeforeSystemHeaderLLVM) { std::string Code = "#include \n" "\n" "int main() {}"; std::string Expected = "#include \"z.h\"\n" "#include \n" "\n" "int main() {}"; EXPECT_EQ(Expected, insert(Code, "\"z.h\"")); } TEST_F(HeaderIncludesTest, InsertAfterSystemHeaderGoogle) { std::string Code = "#include \n" "\n" "int main() {}"; std::string Expected = "#include \n" "#include \"z.h\"\n" "\n" "int main() {}"; Style = format::getGoogleStyle(format::FormatStyle::LanguageKind::LK_Cpp) .IncludeStyle; EXPECT_EQ(Expected, insert(Code, "\"z.h\"")); } TEST_F(HeaderIncludesTest, InsertOneIncludeLLVMStyle) { std::string Code = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"b.h\"\n" "#include \"clang/Format/Format.h\"\n" "#include \n"; std::string Expected = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"b.h\"\n" "#include \"clang/Format/Format.h\"\n" "#include \"llvm/x/y.h\"\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "\"llvm/x/y.h\"")); } TEST_F(HeaderIncludesTest, InsertIntoBlockSorted) { std::string Code = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"c.h\"\n" "#include \n"; std::string Expected = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"b.h\"\n" "#include \"c.h\"\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "\"b.h\"")); } TEST_F(HeaderIncludesTest, InsertIntoFirstBlockOfSameKind) { std::string Code = "#include \"x/fix.h\"\n" "#include \"c.h\"\n" "#include \"e.h\"\n" "#include \"f.h\"\n" "#include \n" "#include \n" "#include \"m.h\"\n" "#include \"n.h\"\n"; std::string Expected = "#include \"x/fix.h\"\n" "#include \"c.h\"\n" "#include \"d.h\"\n" "#include \"e.h\"\n" "#include \"f.h\"\n" "#include \n" "#include \n" "#include \"m.h\"\n" "#include \"n.h\"\n"; EXPECT_EQ(Expected, insert(Code, "\"d.h\"")); } TEST_F(HeaderIncludesTest, InsertIntoSystemBlockSorted) { std::string Code = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"c.h\"\n" "#include \n" "#include \n"; std::string Expected = "#include \"x/fix.h\"\n" "#include \"a.h\"\n" "#include \"c.h\"\n" "#include \n" "#include \n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, InsertNewSystemIncludeGoogleStyle) { std::string Code = "#include \"x/fix.h\"\n" "\n" "#include \"y/a.h\"\n" "#include \"z/b.h\"\n"; // FIXME: inserting after the empty line following the main header might be // preferred. std::string Expected = "#include \"x/fix.h\"\n" "#include \n" "\n" "#include \"y/a.h\"\n" "#include \"z/b.h\"\n"; Style = format::getGoogleStyle(format::FormatStyle::LanguageKind::LK_Cpp) .IncludeStyle; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, NotConfusedByDefine) { std::string Code = "void f() {}\n" "#define A \\\n" " int i;"; std::string Expected = "#include \n" "void f() {}\n" "#define A \\\n" " int i;"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, SkippedTopComment) { std::string Code = "// comment\n" "\n" " // comment\n"; std::string Expected = "// comment\n" "\n" " // comment\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, SkippedMixedComments) { std::string Code = "// comment\n" "// comment \\\n" " comment continued\n" "/*\n" "* comment\n" "*/\n"; std::string Expected = "// comment\n" "// comment \\\n" " comment continued\n" "/*\n" "* comment\n" "*/\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, MultipleBlockCommentsInOneLine) { std::string Code = "/*\n" "* comment\n" "*/ /* comment\n" "*/\n" "\n\n" "/* c1 */ /*c2 */\n"; std::string Expected = "/*\n" "* comment\n" "*/ /* comment\n" "*/\n" "\n\n" "/* c1 */ /*c2 */\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, CodeAfterComments) { std::string Code = "/*\n" "* comment\n" "*/ /* comment\n" "*/\n" "\n\n" "/* c1 */ /*c2 */\n" "\n" "int x;\n"; std::string Expected = "/*\n" "* comment\n" "*/ /* comment\n" "*/\n" "\n\n" "/* c1 */ /*c2 */\n" "\n" "#include \n" "int x;\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, FakeHeaderGuardIfDef) { std::string Code = "// comment \n" "#ifdef X\n" "#define X\n"; std::string Expected = "// comment \n" "#include \n" "#ifdef X\n" "#define X\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, RealHeaderGuardAfterComments) { std::string Code = "// comment \n" "#ifndef X\n" "#define X\n" "int x;\n" "#define Y 1\n"; std::string Expected = "// comment \n" "#ifndef X\n" "#define X\n" "#include \n" "int x;\n" "#define Y 1\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, IfNDefWithNoDefine) { std::string Code = "// comment \n" "#ifndef X\n" "int x;\n" "#define Y 1\n"; std::string Expected = "// comment \n" "#include \n" "#ifndef X\n" "int x;\n" "#define Y 1\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, FakeHeaderGuard) { std::string Code = "// comment \n" "#ifndef X\n" "#define 1\n"; std::string Expected = "// comment \n" "#include \n" "#ifndef X\n" "#define 1\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, HeaderGuardWithComment) { std::string Code = "// comment \n" "#ifndef X // comment\n" "// comment\n" "/* comment\n" "*/\n" "/* comment */ #define X\n" "int x;\n" "#define Y 1\n"; std::string Expected = "// comment \n" "#ifndef X // comment\n" "// comment\n" "/* comment\n" "*/\n" "/* comment */ #define X\n" "#include \n" "int x;\n" "#define Y 1\n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, EmptyCode) { std::string Code = ""; std::string Expected = "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, NoNewLineAtTheEndOfCode) { std::string Code = "#include "; std::string Expected = "#include \n#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, SkipExistingHeaders) { std::string Code = "#include \"a.h\"\n" "#include \n"; std::string Expected = "#include \"a.h\"\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); EXPECT_EQ(Expected, insert(Code, "\"a.h\"")); } TEST_F(HeaderIncludesTest, AddIncludesWithDifferentForms) { std::string Code = "#include \n"; // FIXME: this might not be the best behavior. std::string Expected = "#include \"vector\"\n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "\"vector\"")); } TEST_F(HeaderIncludesTest, NoInsertionAfterCode) { std::string Code = "#include \"a.h\"\n" "void f() {}\n" "#include \"b.h\"\n"; std::string Expected = "#include \"a.h\"\n" "#include \"c.h\"\n" "void f() {}\n" "#include \"b.h\"\n"; EXPECT_EQ(Expected, insert(Code, "\"c.h\"")); } TEST_F(HeaderIncludesTest, NoInsertionInStringLiteral) { std::string Code = "#include \"a.h\"\n" "const char[] = R\"(\n" "#include \"b.h\"\n" ")\";\n"; std::string Expected = "#include \"a.h\"\n" "#include \"c.h\"\n" "const char[] = R\"(\n" "#include \"b.h\"\n" ")\";\n"; EXPECT_EQ(Expected, insert(Code, "\"c.h\"")); } TEST_F(HeaderIncludesTest, NoInsertionAfterOtherDirective) { std::string Code = "#include \"a.h\"\n" "#ifdef X\n" "#include \"b.h\"\n" "#endif\n"; std::string Expected = "#include \"a.h\"\n" "#include \"c.h\"\n" "#ifdef X\n" "#include \"b.h\"\n" "#endif\n"; EXPECT_EQ(Expected, insert(Code, "\"c.h\"")); } TEST_F(HeaderIncludesTest, CanInsertAfterLongSystemInclude) { std::string Code = "#include \"a.h\"\n" "// comment\n\n" "#include \n"; std::string Expected = "#include \"a.h\"\n" "// comment\n\n" "#include \n" "#include \n"; EXPECT_EQ(Expected, insert(Code, "")); } TEST_F(HeaderIncludesTest, CanInsertAfterComment) { std::string Code = "#include \"a.h\"\n" "// Comment\n" "\n" "/* Comment */\n" "// Comment\n" "\n" "#include \"b.h\"\n"; std::string Expected = "#include \"a.h\"\n" "// Comment\n" "\n" "/* Comment */\n" "// Comment\n" "\n" "#include \"b.h\"\n" "#include \"c.h\"\n"; EXPECT_EQ(Expected, insert(Code, "\"c.h\"")); } TEST_F(HeaderIncludesTest, LongCommentsInTheBeginningOfFile) { std::string Code = "// Loooooooooooooooooooooooooong comment\n" "// Loooooooooooooooooooooooooong comment\n" "// Loooooooooooooooooooooooooong comment\n" "#include \n" "#include \n" "\n" "#include \"a.h\"\n" "#include \"b.h\"\n"; std::string Expected = "// Loooooooooooooooooooooooooong comment\n" "// Loooooooooooooooooooooooooong comment\n" "// Loooooooooooooooooooooooooong comment\n" "#include \n" "#include \n" "\n" "#include \"a.h\"\n" "#include \"b.h\"\n" "#include \"third.h\"\n"; Style = format::getGoogleStyle(format::FormatStyle::LanguageKind::LK_Cpp) .IncludeStyle; EXPECT_EQ(Expected, insert(Code, "\"third.h\"")); } TEST_F(HeaderIncludesTest, SimpleDeleteInclude) { std::string Code = "#include \"abc.h\"\n" "#include \"xyz.h\" // comment\n" "int x;\n"; std::string Expected = "#include \"abc.h\"\n" "int x;\n"; EXPECT_EQ(Expected, remove(Code, "\"xyz.h\"")); } TEST_F(HeaderIncludesTest, DeleteQuotedOnly) { std::string Code = "#include \"abc.h\"\n" "#include \n" "int x;\n"; std::string Expected = "#include \n" "int x;\n"; EXPECT_EQ(Expected, remove(Code, "\"abc.h\"")); } TEST_F(HeaderIncludesTest, DeleteAllCode) { std::string Code = "#include \"xyz.h\"\n"; std::string Expected = ""; EXPECT_EQ(Expected, remove(Code, "\"xyz.h\"")); } TEST_F(HeaderIncludesTest, DeleteOnlyIncludesWithSameQuote) { std::string Code = "#include \"xyz.h\"\n" "#include \"xyz\"\n" "#include \n"; std::string Expected = "#include \"xyz.h\"\n" "#include \"xyz\"\n"; EXPECT_EQ(Expected, remove(Code, "")); } TEST_F(HeaderIncludesTest, CanDeleteAfterCode) { std::string Code = "#include \"a.h\"\n" "void f() {}\n" "#include \"b.h\"\n"; std::string Expected = "#include \"a.h\"\n" "void f() {}\n"; EXPECT_EQ(Expected, remove(Code, "\"b.h\"")); } } // namespace } // namespace tooling } // namespace clang @