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

1 files changed, 58 insertions, 0 deletions

diff --git a/dottydoc/jvm/test/BaseTest.scala b/dottydoc/jvm/test/BaseTest.scala
new file mode 100644
index 000000000..2703f8169
--- /dev/null
+++ b/dottydoc/jvm/test/BaseTest.scala
@@ -0,0 +1,58 @@
+package dotty.tools
+package dottydoc
+
+import dotc.core.Contexts
+import Contexts.{ Context, ContextBase, FreshContext }
+import dotc.util.SourceFile
+import dotc.core.Phases.Phase
+import dotc.typer.FrontEnd
+import dottydoc.core.Phases.DocPhase
+
+trait DottyTest {
+  dotty.tools.dotc.parsing.Scanners // initialize keywords
+
+  implicit var ctx: FreshContext = {
+    val base = new ContextBase
+    import base.settings._
+    val ctx = base.initialCtx.fresh
+    ctx.setSetting(ctx.settings.language, List("Scala2"))
+    ctx.setSetting(ctx.settings.YkeepComments, true)
+    ctx.setSetting(ctx.settings.YDocNoWrite, true)
+    base.initialize()(ctx)
+    ctx
+  }
+
+  private def compilerWithChecker(assertion: DocPhase => Unit) = new DottyDocCompiler {
+    private[this] val docPhase = new DocPhase
+
+    override def phases =
+      List(new FrontEnd) ::
+      List(docPhase) ::
+      List(new Phase {
+        def phaseName = "assertionPhase"
+        override def run(implicit ctx: Context): Unit = assertion(docPhase)
+      }) ::
+      Nil
+  }
+
+  def checkSource(source: String)(assertion: DocPhase => Unit): Unit = {
+    val c = compilerWithChecker(assertion)
+    c.rootContext(ctx)
+    val run = c.newRun
+    run.compile(source)
+  }
+
+  def checkFiles(sources: List[String])(assertion: DocPhase => Unit): Unit = {
+    val c = compilerWithChecker(assertion)
+    c.rootContext(ctx)
+    val run = c.newRun
+    run.compile(sources)
+  }
+
+  def checkSources(sourceFiles: List[SourceFile])(assertion: DocPhase => Unit): Unit = {
+    val c = compilerWithChecker(assertion)
+    c.rootContext(ctx)
+    val run = c.newRun
+    run.compileSources(sourceFiles)
+  }
+}