From 951768d7203baa33568e9b713053f30bc9153c4e Mon Sep 17 00:00:00 2001
From: Szymon Świstun <sswistun@virtuslab.com>
Date: Mon, 10 Feb 2020 13:07:24 +0100
Subject: added test point after render

---
 testApi/src/main/kotlin/testApi/testRunner/DokkaTestGenerator.kt | 1 +
 testApi/src/main/kotlin/testApi/testRunner/TestRunner.kt         | 7 +++++--
 2 files changed, 6 insertions(+), 2 deletions(-)

(limited to 'testApi/src/main/kotlin')

diff --git a/testApi/src/main/kotlin/testApi/testRunner/DokkaTestGenerator.kt b/testApi/src/main/kotlin/testApi/testRunner/DokkaTestGenerator.kt
index 21bbe72d..33f0aff8 100644
--- a/testApi/src/main/kotlin/testApi/testRunner/DokkaTestGenerator.kt
+++ b/testApi/src/main/kotlin/testApi/testRunner/DokkaTestGenerator.kt
@@ -39,5 +39,6 @@ internal class DokkaTestGenerator(
         pagesTransformationStage(transformedPages)
 
         dokkaGenerator.render(transformedPages, context)
+        renderingStage(transformedPages, context)
     }
 }
\ No newline at end of file
diff --git a/testApi/src/main/kotlin/testApi/testRunner/TestRunner.kt b/testApi/src/main/kotlin/testApi/testRunner/TestRunner.kt
index edeedf10..6a5bd303 100644
--- a/testApi/src/main/kotlin/testApi/testRunner/TestRunner.kt
+++ b/testApi/src/main/kotlin/testApi/testRunner/TestRunner.kt
@@ -101,6 +101,7 @@ abstract class AbstractCoreTest {
         var documentablesTransformationStage: (Module) -> Unit = {}
         var pagesGenerationStage: (ModulePageNode) -> Unit = {}
         var pagesTransformationStage: (RootPageNode) -> Unit = {}
+        var renderingStage: (RootPageNode, DokkaContext) -> Unit = { a, b -> }
 
         fun build() = TestMethods(
             analysisSetupStage,
@@ -109,7 +110,8 @@ abstract class AbstractCoreTest {
             documentablesMergingStage,
             documentablesTransformationStage,
             pagesGenerationStage,
-            pagesTransformationStage
+            pagesTransformationStage,
+            renderingStage
         )
     }
 
@@ -210,5 +212,6 @@ data class TestMethods(
     val documentablesMergingStage: (Module) -> Unit,
     val documentablesTransformationStage: (Module) -> Unit,
     val pagesGenerationStage: (ModulePageNode) -> Unit,
-    val pagesTransformationStage: (RootPageNode) -> Unit
+    val pagesTransformationStage: (RootPageNode) -> Unit,
+    val renderingStage: (RootPageNode, DokkaContext) -> Unit
 )
\ No newline at end of file
-- 
cgit