1 files changed, 59 insertions, 0 deletions
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"
+    )
+  }
+}