Fix repeated traversal of packages when generating docs (5x speedup)

1 files changed, 29 insertions, 0 deletions

diff --git a/dottydoc/jvm/test/SimpleComments.scala b/dottydoc/jvm/test/SimpleComments.scala
new file mode 100644
index 000000000..ad9b7b1a6
--- /dev/null
+++ b/dottydoc/jvm/test/SimpleComments.scala
@@ -0,0 +1,29 @@
+package dotty.tools
+package dottydoc
+
+import org.junit.Test
+import org.junit.Assert._
+
+class TestSimpleComments extends DottyTest {
+
+  @Test def simpleComment = {
+    val source =
+      """
+      |package scala
+      |
+      |/** Hello, world! */
+      |trait HelloWorld
+      """.stripMargin
+
+    checkSource(source) { doc =>
+      val traitCmt = doc
+        .packages("scala")
+        .children.find(_.path.mkString(".") == "scala.HelloWorld")
+        .flatMap(_.comment.map(_.body))
+        .get
+
+      assertEquals(traitCmt, "<p>Hello, world!</p>")
+    }
+  }
+
+}