1 files changed, 19 insertions, 7 deletions

diff --git a/testApi/src/main/kotlin/testApi/context/MockContext.kt b/testApi/src/main/kotlin/testApi/context/MockContext.kt
index 4f481bce..758a4311 100644
--- a/testApi/src/main/kotlin/testApi/context/MockContext.kt
+++ b/testApi/src/main/kotlin/testApi/context/MockContext.kt
@@ -8,23 +8,30 @@ import org.jetbrains.dokka.plugability.DokkaPlugin
 import org.jetbrains.dokka.plugability.ExtensionPoint
 import org.jetbrains.dokka.utilities.DokkaConsoleLogger
 import kotlin.reflect.KClass
+import kotlin.reflect.KMutableProperty
+import kotlin.reflect.full.memberProperties
 
 @Suppress("UNCHECKED_CAST") // It is only usable from tests so we do not care about safety
 class MockContext(
-    vararg extensions: Pair<ExtensionPoint<*>, Any>,
+    vararg extensions: Pair<ExtensionPoint<*>, (DokkaContext) -> Any>,
     private val testConfiguration: DokkaConfiguration? = null,
     private val testPlatforms: Map<PlatformData, EnvironmentAndFacade>? = null
 ) : DokkaContext {
-    private val extensionMap: Map<ExtensionPoint<*>, List<Any>> =
-        extensions.groupBy(Pair<ExtensionPoint<*>, Any>::first, Pair<ExtensionPoint<*>, Any>::second)
-
-    override fun <T : DokkaPlugin> plugin(kclass: KClass<T>): T? = null.also {
-        logger.warn("Cannot access plugins from mock context")
+    private val extensionMap by lazy {
+        extensions.groupBy(Pair<ExtensionPoint<*>, (DokkaContext) -> Any>::first) {
+            it.second(this)
+        }
     }
 
+    private val plugins = mutableMapOf<KClass<out DokkaPlugin>, DokkaPlugin>()
+
+    override fun <T : DokkaPlugin> plugin(kclass: KClass<T>): T? = plugins.getOrPut(kclass) {
+        kclass.constructors.single { it.parameters.isEmpty() }.call().also { it.injectContext(this) }
+    } as T
+
     override fun <T : Any, E : ExtensionPoint<T>> get(point: E): List<T> = extensionMap[point] as List<T>
 
-    override fun <T : Any, E : ExtensionPoint<T>> single(point: E): T  = get(point).single()
+    override fun <T : Any, E : ExtensionPoint<T>> single(point: E): T = get(point).single()
 
     override val logger = DokkaConsoleLogger
 
@@ -33,4 +40,9 @@ class MockContext(
 
     override val platforms: Map<PlatformData, EnvironmentAndFacade>
         get() = testPlatforms ?: throw IllegalStateException("This mock context doesn't provide platforms data")
+}
+
+private fun DokkaPlugin.injectContext(context: DokkaContext) {
+    (DokkaPlugin::class.memberProperties.single { it.name == "context" } as KMutableProperty<*>)
+        .setter.call(this, context)
 }
\ No newline at end of file