1 files changed, 32 insertions, 0 deletions
diff --git a/doc-tool/test/SourceFileOps.scala b/doc-tool/test/SourceFileOps.scala
index 7b0c2e807..37520921d 100644
--- a/doc-tool/test/SourceFileOps.scala
+++ b/doc-tool/test/SourceFileOps.scala
@@ -7,7 +7,12 @@ import java.io.{ BufferedWriter, OutputStreamWriter }
 import io.VirtualFile
 import scala.io.Codec
 
+import model.Package
+
 trait SourceFileOps {
+  import scala.collection.JavaConverters._
+  val site = new Site(new java.io.File("../doc-tool/resources/"), "test-site", Map.empty)
+
   def stringToSource(path: String, sourceCode: String): SourceFile = {
     val virtualFile = new VirtualFile(path, path)
     val writer = new BufferedWriter(new OutputStreamWriter(virtualFile.output, "UTF-8"))
@@ -16,4 +21,31 @@ trait SourceFileOps {
 
     new SourceFile(virtualFile, Codec.UTF8)
   }
+
+  def markdownPage(
+    sourceCode: String,
+    path: String = "test-page",
+    params: Map[String, AnyRef] = Map.empty,
+    includes: Map[String, Include] = Map.empty,
+    docs: Map[String, Package] = Map.empty
+  ) = new MarkdownPage(
+    path,
+    stringToSource(path, sourceCode),
+    params,
+    includes,
+    docs
+  )
+
+  def htmlPage(
+    sourceCode: String,
+    path: String = "test-page",
+    params: Map[String, AnyRef] = Map.empty,
+    includes: Map[String, Include] = Map.empty,
+    docs: Map[String, Package] = Map.empty
+  ) = new HtmlPage(
+    path,
+    stringToSource(path, sourceCode),
+    params,
+    includes
+  )
 }