bugfix: Document highlight in self type

scalameta · Jan 10, 2024 · 33ae25a · 33ae25a
1 parent 1f3e9f2
commit 33ae25a
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/mtags/src/main/scala-2/scala/meta/internal/pc/PcCollector.scala b/mtags/src/main/scala-2/scala/meta/internal/pc/PcCollector.scala
@@ -431,7 +431,7 @@ abstract class PcCollector[T](
          * val opt: Option[<<String>>] =
          */
         case tpe: TypeTree if tpe.original != null =>
-          tpe.original.children.foldLeft(acc)(traverse(_, _))
+          traverse(acc, tpe.original)
         /**
          * Some type trees don't have symbols attached such as:
          * type A = List[_ <: <<Iterable>>[Int]]

diff --git a/tests/cross/src/test/scala/tests/highlight/DocumentHighlightSuite.scala b/tests/cross/src/test/scala/tests/highlight/DocumentHighlightSuite.scala
@@ -1138,4 +1138,22 @@ class DocumentHighlightSuite extends BaseDocumentHighlightSuite {
       |}""".stripMargin
   )
 
+  check(
+    "self-type",
+    """|trait Foo {
+       |  self: <<An@@y>> =>
+       |    def bar(): <<Any>>
+       |}
+       |""".stripMargin
+  )
+
+  check(
+    "self-type2",
+    """|trait Foo {
+       |  self: <<Any>> =>
+       |    def bar(): <<An@@y>>
+       |}
+       |""".stripMargin
+  )
+
 }