tests

1 files changed, 149 insertions, 0 deletions

diff --git a/core/src/test/kotlin/transformerBuilders/PageTransformerBuilderTest.kt b/core/src/test/kotlin/transformerBuilders/PageTransformerBuilderTest.kt
new file mode 100644
index 00000000..13ea5580
--- /dev/null
+++ b/core/src/test/kotlin/transformerBuilders/PageTransformerBuilderTest.kt
@@ -0,0 +1,149 @@
+package transformerBuilders;
+
+import org.jetbrains.dokka.CoreExtensions
+import org.jetbrains.dokka.pages.*
+import org.jetbrains.dokka.plugability.DokkaPlugin
+import org.jetbrains.dokka.transformers.pages.PageNodeTransformer
+import org.jetbrains.dokka.transformers.pages.pageMapper
+import org.jetbrains.dokka.transformers.pages.pageScanner
+import org.jetbrains.dokka.transformers.pages.pageStructureTransformer
+import org.jetbrains.dokka.utilities.DokkaConsoleLogger
+import org.junit.Test
+import testApi.testRunner.AbstractCoreTest
+
+class PageTransformerBuilderTest : AbstractCoreTest() {
+
+    class ProxyPlugin(transformer: PageNodeTransformer) : DokkaPlugin() {
+        val pageTransformer by extending { CoreExtensions.pageTransformer with transformer }
+    }
+
+    @Test
+    fun scannerTest() {
+        val configuration = dokkaConfiguration {
+            passes {
+                pass {
+                    sourceRoots = listOf("src/main/kotlin/transformerBuilder/Test.kt")
+                }
+            }
+        }
+        val list = mutableListOf<String>()
+
+        var orig: PageNode? = null
+
+        testInline(
+            """
+            |/src/main/kotlin/transformerBuilder/Test.kt
+            |package transformerBuilder
+            |
+            |object Test {
+            |   fun test2(str: String): Unit {println(str)}
+            |}
+        """.trimMargin(),
+            configuration,
+            pluginOverrides = listOf(ProxyPlugin(pageScanner {
+                list += name
+            }))
+        ) {
+            pagesGenerationStage = {
+                orig = it
+            }
+            pagesTransformationStage = { root ->
+                list.assertCount(8, "Page list: ")
+                orig?.let { root.assertTransform(it) }
+            }
+        }
+    }
+
+    @Test
+    fun mapperTest() {
+        val configuration = dokkaConfiguration {
+            passes {
+                pass {
+                    sourceRoots = listOf("src/main/kotlin/transformerBuilder/Test.kt")
+                }
+            }
+        }
+
+        var orig: PageNode? = null
+
+        testInline(
+            """
+            |/src/main/kotlin/transformerBuilder/Test.kt
+            |package transformerBuilder
+            |
+            |object Test {
+            |   fun test2(str: String): Unit {println(str)}
+            |}
+        """.trimMargin(),
+            configuration,
+            pluginOverrides = listOf(ProxyPlugin(pageMapper {
+                modified(name = name + "2")
+            }))
+        ) {
+            pagesGenerationStage = {
+                orig = it
+            }
+            pagesTransformationStage = {
+                it.let { root ->
+                    root.name.assertEqual("root2", "Root name: ")
+                    orig?.let {
+                        root.assertTransform(it) { node -> node.modified(name = node.name + "2") }
+                    }
+                }
+            }
+        }
+    }
+
+    @Test
+    fun structureTransformerTest() {
+        val configuration = dokkaConfiguration {
+            passes {
+                pass {
+                    sourceRoots = listOf("src/main/kotlin/transformerBuilder/Test.kt")
+                }
+            }
+        }
+
+        testInline(
+            """
+            |/src/main/kotlin/transformerBuilder/Test.kt
+            |package transformerBuilder
+            |
+            |object Test {
+            |   fun test2(str: String): Unit {println(str)}
+            |}
+        """.trimMargin(),
+            configuration,
+            pluginOverrides = listOf(ProxyPlugin(pageStructureTransformer {
+                val ch = children.first()
+                modified(
+                    children = listOf(
+                        ch,
+                        RendererSpecificResourcePage("test", emptyList(), RenderingStrategy.DoNothing)
+                    )
+                )
+            }))
+        ) {
+            pagesTransformationStage = { root ->
+                root.children.assertCount(2, "Root children: ")
+                root.children.first().name.assertEqual("transformerBuilder")
+                root.children[1].name.assertEqual("test")
+            }
+        }
+    }
+
+    private fun <T> Collection<T>.assertCount(n: Int, prefix: String = "") =
+        assert(count() == n) { "${prefix}Expected $n, got ${count()}" }
+
+    private fun <T> T.assertEqual(expected: T, prefix: String = "") = assert(this == expected) {
+        "${prefix}Expected $expected, got $this"
+    }
+
+    private fun PageNode.assertTransform(expected: PageNode, block: (PageNode) -> PageNode = { it }): Unit = this.let {
+        it.name.assertEqual(block(expected).name)
+        it.children.zip(expected.children).forEach { (g, e) ->
+            g.name.assertEqual(block(e).name)
+            g.assertTransform(e, block)
+        }
+    }
+}