Build: Fix embedded configuration import in jps build

Should affect only test configurations

gradle/jps.gradle.kts

@@ -37,10 +37,24 @@ fun JUnit.configureForKotlin() {
 if (kotlinBuildProperties.isInJpsBuildIdeaSync) {
     allprojects {
         apply(mapOf("plugin" to "idea"))
+        // Make Idea import embedded configuration as transitive dependency for some configurations
         afterEvaluate {
-            // Make Idea import embedded configuration as transitive dependency
+            listOf(
-            configurations.findByName("embedded")?.let { embedded ->
+                "testCompile",
-                configurations.findByName("runtime")?.extendsFrom(embedded)
+                "testCompileOnly",
+                "testRuntime",
+                "testRuntimeOnly"
+            ).forEach { configurationName ->
+                val dependencyProjects = configurations
+                    .findByName(configurationName)
+                    ?.dependencies
+                    ?.mapNotNull { (it as? ProjectDependency)?.dependencyProject }
+
+                dependencies {
+                    dependencyProjects?.forEach {dependencyProject ->
+                        add(configurationName, project(dependencyProject.path, configuration = "embedded"))
+                    }
+                }
             }
         }
     }