2 files changed, 6 insertions, 2 deletions
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