Fix recursive expansion of templates

2 files changed, 164 insertions, 0 deletions

diff --git a/doc-tool/test/dotty/tools/dottydoc/staticsite/PageTests.scala b/doc-tool/test/dotty/tools/dottydoc/staticsite/PageTests.scala
new file mode 100644
index 000000000..87358d738
--- /dev/null
+++ b/doc-tool/test/dotty/tools/dottydoc/staticsite/PageTests.scala
@@ -0,0 +1,105 @@
+package dotty.tools
+package dottydoc
+package staticsite
+
+import org.junit.Test
+import org.junit.Assert._
+
+class PageTests extends DottyDocTest {
+  import scala.collection.JavaConverters._
+
+  @Test def mdHas1Key = {
+    val page = new MarkdownPage(
+      """|---
+         |key:
+         |---
+         |
+         |great""".stripMargin,
+      Map.empty
+    )
+
+    assert(
+      page.yaml == Map("key" -> ""),
+      s"""incorrect yaml, expected "key:" without key in: ${page.yaml}"""
+    )
+
+    assertEquals("<p>great</p>\n", page.html)
+  }
+
+  @Test def yamlPreservesLiquidTags = {
+    val page1 = new MarkdownPage(
+      """|---
+         |key:
+         |---
+         |
+         |{{ content }}""".stripMargin,
+       Map("content" -> "Hello, world!")
+    )
+
+    assert(
+      page1.yaml == Map("key" -> ""),
+      s"""incorrect yaml, expected "key:" without key in: ${page1.yaml}"""
+    )
+
+    assertEquals("<p>Hello, world!</p>\n", page1.html)
+
+    val page2 = new MarkdownPage(
+      """|{{ content }}""".stripMargin,
+      Map("content" -> "hello")
+    )
+    assert(
+      page2.yaml == Map(),
+      s"""incorrect yaml, expected "key:" without key in: ${page2.yaml}"""
+    )
+    assertEquals("<p>hello</p>\n", page2.html)
+
+    val page3 = new MarkdownPage(
+      """|{% if product.title == "Awesome Shoes" %}
+         |These shoes are awesome!
+         |{% endif %}""".stripMargin,
+      Map("product" -> Map("title" -> "Awesome Shoes").asJava)
+    )
+
+    assertEquals(
+      "<p>These shoes are awesome!</p>\n",
+      page3.html
+    )
+  }
+
+  @Test def simpleHtmlPage = {
+    val p1 = new HtmlPage("""<h1>{{ "hello, world!" }}</h1>""", Map.empty)
+    assert(p1.yaml == Map(), "non-empty yaml found")
+    assertEquals("<h1>hello, world!</h1>", p1.html)
+  }
+
+  @Test def htmlPageHasNoYaml = {
+    val page = new HtmlPage(
+      """|---
+         |layout: main
+         |---
+         |
+         |Hello, world!""".stripMargin,
+      Map.empty
+    )
+
+    assert(!page.html.contains("---\nlayout: main\n---"),
+           s"page still contains yaml:\n${page.html}")
+  }
+
+  @Test def illegalYamlFrontMatter = try {
+    val page = new HtmlPage(
+      """|---
+         |layout: main
+         |
+         |
+         |Hello, world!""".stripMargin,
+      Map.empty
+    )
+
+    page.html
+    fail("illegal front matter didn't throw exception")
+  } catch {
+    case IllegalFrontMatter(x) => // success!
+    case t: Throwable => throw t
+  }
+}
diff --git a/doc-tool/test/dotty/tools/dottydoc/staticsite/SiteTests.scala b/doc-tool/test/dotty/tools/dottydoc/staticsite/SiteTests.scala
new file mode 100644
index 000000000..b73d7555b
--- /dev/null
+++ b/doc-tool/test/dotty/tools/dottydoc/staticsite/SiteTests.scala
@@ -0,0 +1,59 @@
+package dotty.tools
+package dottydoc
+package staticsite
+
+import org.junit.Test
+import org.junit.Assert._
+
+class SiteTests extends DottyDocTest {
+  @Test def hasCorrectLayoutFiles = {
+    val site = new Site(new java.io.File("../doc-tool/resources/"))
+
+    assert(site.root.exists && site.root.isDirectory,
+           s"'${site.root.getName}' is not a directory")
+
+    val expectedLayouts = Set("main", "index")
+    assert(site.layouts.keys == expectedLayouts,
+           s"Incorrect layouts in: ${site.layouts.keys}, expected: $expectedLayouts")
+  }
+
+  @Test def renderHelloInMainLayout = {
+    val site = new Site(new java.io.File("../doc-tool/resources/"))
+
+    val renderedPage = site.render(new HtmlPage(
+      """|---
+         |layout: main
+         |---
+         |
+         |Hello, world!""".stripMargin,
+      Map.empty
+    ), Map.empty)
+
+    assert(
+      renderedPage.contains("Hello, world!") &&
+      !renderedPage.contains("---\nlayout: main\n---\n") &&
+      renderedPage.contains("<!DOCTYPE html>"),
+      "html page did not render properly"
+    )
+  }
+
+  @Test def renderMultipleTemplates = {
+    val site = new Site(new java.io.File("../doc-tool/resources/"))
+
+    val renderedPage = site.render(new HtmlPage(
+      """|---
+         |layout: index
+         |---
+         |Hello, world!""".stripMargin,
+      Map.empty
+    ), Map.empty)
+
+    assert(
+      renderedPage.contains("<h1>Hello, world!</h1>") &&
+      !renderedPage.contains("---\nlayout: main\n---\n") &&
+      !renderedPage.contains("---\nlayout: index\n---\n") &&
+      renderedPage.contains("<!DOCTYPE html>"),
+      "html page did not render properly"
+    )
+  }
+}