Make multiplatform source suite generator be able to process tests in… (#216)

Fixes #215 issue

Author: PavelPunegov

integration/src/test/kotlin/kotlinx/benchmark/integration/SuiteSourceGeneratorTest.kt

@@ -20,4 +20,36 @@ class SuiteSourceGeneratorTest : GradleTest() {
             }
         }
     }
+
+    @Test
+    fun generateAndCompileNativeBenchmarks() {
+        generateAndCompile("native")
+    }
+
+    @Test
+    fun generateAndCompileJSBenchmarks() {
+        generateAndCompile("jsIr")
+    }
+
+    @Test
+    fun generateAndCompileWasmBenchmarks() {
+        generateAndCompile("wasmJs")
+    }
+
+    private fun generateAndCompile(target: String) {
+        project("kotlin-multiplatform", true).let { runner ->
+            runner.run(":${target}BenchmarkGenerate")
+
+            runner.generatedDir(target, "RootBenchmark_Descriptor.kt") { descriptorFile ->
+                val text = descriptorFile.readText()
+                assertFalse(
+                    text.contains("<root>"),
+                    "Generated descriptor contains illegal characters '<root>' in $text"
+                )
+            }
+
+            val capitalizedTarget = target.replaceFirstChar { it.uppercaseChar() }
+            runner.run(":compile${capitalizedTarget}BenchmarkKotlin${capitalizedTarget}")
+        }
+    }
 }

integration/src/test/resources/templates/kotlin-multiplatform/src/jsIrMain/kotlin/RootBenchmark.kt

@@ -0,0 +1,13 @@
+import kotlinx.benchmark.*
+import kotlin.math.*
+
+@State(Scope.Benchmark)
+@Measurement(iterations = 3, time = 1, timeUnit = BenchmarkTimeUnit.SECONDS)
+@OutputTimeUnit(BenchmarkTimeUnit.MILLISECONDS)
+@BenchmarkMode(Mode.Throughput)
+open class RootBenchmark {
+    @Benchmark
+    open fun mathBenchmark(): Double {
+        return log(sqrt(3.0) * cos(3.0), 2.0)
+    }
+}

integration/src/test/resources/templates/kotlin-multiplatform/src/nativeMain/kotlin/RootBenchmark.kt

@@ -0,0 +1,13 @@
+import kotlinx.benchmark.*
+import kotlin.math.*
+
+@State(Scope.Benchmark)
+@Measurement(iterations = 3, time = 1, timeUnit = BenchmarkTimeUnit.SECONDS)
+@OutputTimeUnit(BenchmarkTimeUnit.MILLISECONDS)
+@BenchmarkMode(Mode.Throughput)
+open class RootBenchmark {
+    @Benchmark
+    open fun mathBenchmark(): Double {
+        return log(sqrt(3.0) * cos(3.0), 2.0)
+    }
+}

integration/src/test/resources/templates/kotlin-multiplatform/src/wasmJsMain/kotlin/RootBenchmark.kt

@@ -0,0 +1,13 @@
+import kotlinx.benchmark.*
+import kotlin.math.*
+
+@State(Scope.Benchmark)
+@Measurement(iterations = 3, time = 1, timeUnit = BenchmarkTimeUnit.SECONDS)
+@OutputTimeUnit(BenchmarkTimeUnit.MILLISECONDS)
+@BenchmarkMode(Mode.Throughput)
+open class RootBenchmark {
+    @Benchmark
+    open fun mathBenchmark(): Double {
+        return log(sqrt(3.0) * cos(3.0), 2.0)
+    }
+}

plugin/main/src/kotlinx/benchmark/gradle/SuiteSourceGenerator.kt

@@ -124,12 +124,15 @@ class SuiteSourceGenerator(val title: String, val module: ModuleDescriptor, val
     }
 
     private fun generateBenchmark(original: ClassDescriptor) {
-        val originalPackage = original.fqNameSafe.parent()
-        val originalName = original.fqNameSafe.shortName()
-        val originalClass = ClassName(originalPackage.toString(), originalName.toString())
+        val originalFqName = original.fqNameSafe
+        val originalPackage = originalFqName.parent().let {
+            if (it.isRoot) "" else it.asString()
+        }
+        val originalName = originalFqName.shortName().toString()
+        val originalClass = ClassName(originalPackage, originalName)
 
-        val benchmarkPackageName = "$mainBenchmarkPackage.$originalPackage"
-        val benchmarkName = originalName.toString() + "_Descriptor"
+        val benchmarkPackageName = mainBenchmarkPackage + if (originalPackage.isNotEmpty()) ".$originalPackage" else ""
+        val benchmarkName = "${originalName}_Descriptor"
         val benchmarkClass = ClassName(benchmarkPackageName, benchmarkName)
 
         val functions = DescriptorUtils.getAllDescriptors(original.unsubstitutedMemberScope)