[Gradle] Track 'webTest' sources usage

^KT-79482 In Progress

Author: Tapchicoma

libraries/tools/kotlin-gradle-plugin/src/common/kotlin/org/jetbrains/kotlin/gradle/plugin/statistics/FusMetrics.kt

@@ -403,7 +403,17 @@ internal object KotlinSourceSetMetrics : FusMetrics {
     internal fun collectMetrics(project: Project) {
         project.launchInStage(KotlinPluginLifecycle.Stage.AfterFinaliseDsl) {
             project.reportGeneratedSourcesUsage()
-            project.reportWebMainSourceSetUsage()
+            project.reportSourceSetSourcesUsage(
+                mapOf(
+                    "webMain" to BooleanMetrics.KOTLIN_WEB_MAIN_SOURCES_USED,
+                    "webTest" to BooleanMetrics.KOTLIN_WEB_TEST_SOURCES_USED,
+                )
+            )
+            project.reportSourceSetDependenciesUsage(
+                mapOf(
+                    "webMain" to BooleanMetrics.KOTLIN_WEB_MAIN_DEPENDENCIES_PRESENT,
+                )
+            )
         }
     }
 
@@ -417,25 +427,39 @@ internal object KotlinSourceSetMetrics : FusMetrics {
         }
     }
 
-    private suspend fun Project.reportWebMainSourceSetUsage() {
-        project.kotlinExtension.awaitSourceSets().configureEach {
-            if (it.name == "webMain") {
-                if (!it.allKotlinSources.isEmpty) {
-                    project.addConfigurationMetrics {
-                        it.put(BooleanMetrics.KOTLIN_WEB_MAIN_SOURCES_USED, true)
+    private suspend fun Project.reportSourceSetSourcesUsage(
+        sourceSetNameToMetricMap: Map<String, BooleanMetrics>
+    ) {
+        project.kotlinExtension.awaitSourceSets().configureEach { sourceSet ->
+            if (sourceSetNameToMetricMap.keys.any { it == sourceSet.name }) {
+                if (!sourceSet.allKotlinSources.isEmpty) {
+                    project.addConfigurationMetrics { metricsContainer ->
+                        sourceSetNameToMetricMap[sourceSet.name]?.let { metric ->
+                            metricsContainer.put(metric, true)
+                        }
                     }
                 }
+            }
+        }
+    }
 
+    private suspend fun Project.reportSourceSetDependenciesUsage(
+        sourceSetNameToMetricMap: Map<String, BooleanMetrics>
+    ) {
+        project.kotlinExtension.awaitSourceSets().configureEach { sourceSet ->
+            if (sourceSetNameToMetricMap.keys.any { it == sourceSet.name }) {
                 listOf(
-                    it.apiConfigurationName,
-                    it.implementationConfigurationName,
-                    it.compileOnlyConfigurationName,
-                    it.runtimeOnlyConfigurationName,
+                    sourceSet.apiConfigurationName,
+                    sourceSet.implementationConfigurationName,
+                    sourceSet.compileOnlyConfigurationName,
+                    sourceSet.runtimeOnlyConfigurationName,
                 ).map { project.configurations.getByName(it) }
                     .forEach { configuration ->
                         if (configuration.dependencies.isNotEmpty()) {
-                            project.addConfigurationMetrics {
-                                it.put(BooleanMetrics.KOTLIN_WEB_MAIN_DEPENDENCIES_PRESENT, true)
+                            sourceSetNameToMetricMap[sourceSet.name]?.let { metric ->
+                                project.addConfigurationMetrics { metricsContainer ->
+                                    metricsContainer.put(metric, true)
+                                }
                             }
                         }
                     }

libraries/tools/kotlin-gradle-plugin/src/functionalTest/kotlin/org/jetbrains/kotlin/gradle/unitTests/fus/FUSWebMainSourceSetTest.kt

@@ -11,6 +11,7 @@ import org.jetbrains.kotlin.gradle.util.buildProjectWithMPP
 import org.jetbrains.kotlin.statistics.metrics.BooleanMetrics
 import kotlin.test.Test
 import kotlin.test.assertNotNull
+import kotlin.test.assertTrue
 
 @OptIn(ExperimentalWasmDsl::class)
 class FUSWebMainSourceSetTest {
@@ -26,7 +27,7 @@ class FUSWebMainSourceSetTest {
 
         project.evaluate()
 
-        assertNotNull(
+        assertTrue(
             project.collectedFusConfigurationTimeMetrics.booleanMetrics.keys.none {
                 it.name == BooleanMetrics.KOTLIN_WEB_MAIN_SOURCES_USED.name
             },

libraries/tools/kotlin-gradle-plugin/src/functionalTest/kotlin/org/jetbrains/kotlin/gradle/unitTests/fus/FUSWebTestSourceSetTest.kt

@@ -0,0 +1,59 @@
+/*
+ * Copyright 2010-2025 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.gradle.unitTests.fus
+
+import org.jetbrains.kotlin.gradle.ExperimentalWasmDsl
+import org.jetbrains.kotlin.gradle.dsl.multiplatformExtension
+import org.jetbrains.kotlin.gradle.util.buildProjectWithMPP
+import org.jetbrains.kotlin.statistics.metrics.BooleanMetrics
+import kotlin.test.Test
+import kotlin.test.assertNotNull
+import kotlin.test.assertTrue
+
+@OptIn(ExperimentalWasmDsl::class)
+class FUSWebTestSourceSetTest {
+    @Test
+    fun emptyWebTestSourcesAreNotReported() {
+        val project = buildProjectWithMPP(preApplyCode = enableFusOnCI) {
+            with(multiplatformExtension) {
+                js { nodejs() }
+                wasmJs()
+            }
+        }
+
+        project.evaluate()
+
+        assertTrue(
+            project.collectedFusConfigurationTimeMetrics.booleanMetrics.keys.none {
+                it.name == BooleanMetrics.KOTLIN_WEB_TEST_SOURCES_USED.name
+            },
+            "FUS event is present for webTest sources"
+        )
+    }
+
+    @Test
+    fun explicitlyAddedWebTestSourcesAreReported() {
+        val project = buildProjectWithMPP(preApplyCode = enableFusOnCI) {
+            with(multiplatformExtension) {
+                js { browser() }
+                wasmJs { browser() }
+            }
+        }
+
+        val sharedFile = project.layout.projectDirectory.file("src/webTest/kotlin/shared.kt").asFile
+        sharedFile.parentFile.mkdirs()
+        sharedFile.writeText("fun shared() = 1")
+
+        project.evaluate()
+
+        assertNotNull(
+            project.collectedFusConfigurationTimeMetrics.booleanMetrics.entries.singleOrNull {
+                it.key == BooleanMetrics.KOTLIN_WEB_TEST_SOURCES_USED && it.value
+            }
+        )
+    }
+
+}

libraries/tools/kotlin-gradle-statistics/src/main/kotlin/org/jetbrains/kotlin/statistics/metrics/BooleanMetrics.kt

@@ -50,6 +50,7 @@ enum class BooleanMetrics(val type: BooleanOverridePolicy, val anonymization: Bo
 
     KOTLIN_GENERATED_SOURCES_USED(OR, SAFE),
     KOTLIN_WEB_MAIN_SOURCES_USED(OR, SAFE),
+    KOTLIN_WEB_TEST_SOURCES_USED(OR, SAFE),
     KOTLIN_WEB_MAIN_DEPENDENCIES_PRESENT(OR, SAFE),
 
     // Disabled explicitly by the user
@@ -130,6 +131,6 @@ enum class BooleanMetrics(val type: BooleanOverridePolicy, val anonymization: Bo
     ;
 
     companion object {
-        const val VERSION = 22
+        const val VERSION = 23
     }
 }