Output writer for renderer tests

1 files changed, 27 insertions, 0 deletions

diff --git a/plugins/base/src/test/kotlin/utils/TestOutputWriter.kt b/plugins/base/src/test/kotlin/utils/TestOutputWriter.kt
new file mode 100644
index 00000000..43161929
--- /dev/null
+++ b/plugins/base/src/test/kotlin/utils/TestOutputWriter.kt
@@ -0,0 +1,27 @@
+package utils
+
+import org.jetbrains.dokka.base.DokkaBase
+import org.jetbrains.dokka.base.renderers.OutputWriter
+import org.jetbrains.dokka.plugability.DokkaPlugin
+import java.io.File
+
+class TestOutputWriterPlugin(failOnOverwrite: Boolean): DokkaPlugin() {
+    private val writer = TestOutputWriter(failOnOverwrite)
+
+    val testWriter by extending { plugin<DokkaBase>().outputWriter with writer }
+}
+
+class TestOutputWriter(private val failOnOverwrite: Boolean): OutputWriter {
+    val contents: Map<String, String> get() = _contents
+
+    private val _contents = mutableMapOf<String, String>()
+
+    override fun write(path: String, text: String, ext: String) {
+        val fullPath = listOf(path, ext).joinToString(separator = ".")
+        _contents.putIfAbsent(fullPath, text)?.also {
+            if (failOnOverwrite) throw AssertionError("File $fullPath is being overwritten.")
+        }
+    }
+
+    override fun writeResources(pathFrom: String, pathTo: String) = write(pathFrom, File(pathTo).readText(), "")
+}