diff --git a/clang-tools-extra/clangd/XRefs.cpp b/clang-tools-extra/clangd/XRefs.cpp index 089f8158c9aa5..5bbc681cf04e2 100644 --- a/clang-tools-extra/clangd/XRefs.cpp +++ b/clang-tools-extra/clangd/XRefs.cpp @@ -2287,7 +2287,8 @@ prepareCallHierarchy(ParsedAST &AST, Position Pos, PathRef TUPath) { Decl->getKind() != Decl::Kind::FunctionTemplate && !(Decl->getKind() == Decl::Kind::Var && !cast(Decl)->isLocalVarDecl()) && - Decl->getKind() != Decl::Kind::Field) + Decl->getKind() != Decl::Kind::Field && + Decl->getKind() != Decl::Kind::EnumConstant) continue; if (auto CHI = declToCallHierarchyItem(*Decl, AST.tuPath())) Result.emplace_back(std::move(*CHI)); diff --git a/clang-tools-extra/clangd/unittests/CallHierarchyTests.cpp b/clang-tools-extra/clangd/unittests/CallHierarchyTests.cpp index eb852ef5ee00b..08cc80ff8981e 100644 --- a/clang-tools-extra/clangd/unittests/CallHierarchyTests.cpp +++ b/clang-tools-extra/clangd/unittests/CallHierarchyTests.cpp @@ -633,6 +633,35 @@ TEST(CallHierarchy, HierarchyOnVar) { iFromRanges(Source.range("Callee"))))); } +TEST(CallHierarchy, HierarchyOnEnumConstant) { + // Tests that the call hierarchy works on enum constants. + Annotations Source(R"cpp( + enum class Coin { heads$Heads^ , tai$Tails^ls }; + void caller() { + Coin::$CallerH[[heads]]; + Coin::$CallerT[[tails]]; + } + )cpp"); + TestTU TU = TestTU::withCode(Source.code()); + auto AST = TU.build(); + auto Index = TU.index(); + + std::vector Items = + prepareCallHierarchy(AST, Source.point("Heads"), testPath(TU.Filename)); + ASSERT_THAT(Items, ElementsAre(withName("heads"))); + auto IncomingLevel1 = incomingCalls(Items[0], Index.get()); + ASSERT_THAT(IncomingLevel1, + ElementsAre(AllOf(from(withName("caller")), + iFromRanges(Source.range("CallerH"))))); + Items = + prepareCallHierarchy(AST, Source.point("Tails"), testPath(TU.Filename)); + ASSERT_THAT(Items, ElementsAre(withName("tails"))); + IncomingLevel1 = incomingCalls(Items[0], Index.get()); + ASSERT_THAT(IncomingLevel1, + ElementsAre(AllOf(from(withName("caller")), + iFromRanges(Source.range("CallerT"))))); +} + TEST(CallHierarchy, CallInDifferentFileThanCaller) { Annotations Header(R"cpp( #define WALDO void caller() {