From 2fd8e9096706545f8b77e1e66bcc876d7e29f82c Mon Sep 17 00:00:00 2001
From: Vadim Mishenev <vad-mishenev@yandex.ru>
Date: Tue, 8 Aug 2023 17:15:51 +0300
Subject: Fix and refactor Sample Transformer (#3102)

---
 .../src/test/kotlin/content/samples/ContentForSamplesTest.kt | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

(limited to 'plugins/base/src/test')

diff --git a/plugins/base/src/test/kotlin/content/samples/ContentForSamplesTest.kt b/plugins/base/src/test/kotlin/content/samples/ContentForSamplesTest.kt
index 37009e46..9b5ab7ad 100644
--- a/plugins/base/src/test/kotlin/content/samples/ContentForSamplesTest.kt
+++ b/plugins/base/src/test/kotlin/content/samples/ContentForSamplesTest.kt
@@ -2,12 +2,15 @@ package content.samples
 
 import matchers.content.*
 import org.jetbrains.dokka.base.testApi.testRunner.BaseAbstractTest
+import org.jetbrains.dokka.base.transformers.pages.KOTLIN_PLAYGROUND_SCRIPT
 import org.jetbrains.dokka.model.DisplaySourceSet
 import org.junit.jupiter.api.Test
+import utils.TestOutputWriterPlugin
 import utils.classSignature
 import utils.findTestType
 import java.nio.file.Paths
 import kotlin.test.assertEquals
+import kotlin.test.assertNotEquals
 
 class ContentForSamplesTest : BaseAbstractTest() {
     private val testDataDir = getTestDataDir("content/samples").toAbsolutePath()
@@ -61,6 +64,7 @@ class ContentForSamplesTest : BaseAbstractTest() {
 
     @Test
     fun `samples block is rendered in the description`() {
+        val writerPlugin = TestOutputWriterPlugin()
         testInline(
             """
             |/src/main/kotlin/test/source.kt
@@ -70,10 +74,12 @@ class ContentForSamplesTest : BaseAbstractTest() {
             | * @sample [test.sampleForClassDescription]
             | */
             |class Foo
-        """.trimIndent(), testConfiguration
+        """.trimIndent(), testConfiguration,
+            pluginOverrides = listOf(writerPlugin)
         ) {
             pagesTransformationStage = { module ->
                 val page = module.findTestType("test", "Foo")
+                assert(KOTLIN_PLAYGROUND_SCRIPT in page.embeddedResources)
                 page.content.assertNode {
                     group {
                         header(1) { +"Foo" }
@@ -101,6 +107,9 @@ class ContentForSamplesTest : BaseAbstractTest() {
                     skipAllNotMatching()
                 }
             }
+            renderingStage = { _, _ ->
+                assertNotEquals(-1, writerPlugin.writer.contents["root/test/-foo/index.html"]?.indexOf(KOTLIN_PLAYGROUND_SCRIPT))
+            }
         }
     }
 
@@ -134,6 +143,7 @@ class ContentForSamplesTest : BaseAbstractTest() {
         ) {
             pagesTransformationStage = { module ->
                 val page = module.findTestType("pageMerger", "Parent")
+                assert(KOTLIN_PLAYGROUND_SCRIPT in page.embeddedResources)
                 page.content.assertNode {
                     group {
                         header(1) { +"Parent" }
-- 
cgit