Add test deps to the runtime

neoforged · May 25, 2024 · 8a0e733 · 8a0e733
1 parent f083594
commit 8a0e733
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...src/main/java/net/neoforged/gradle/userdev/runtime/extension/UserDevRuntimeExtension.java b/...src/main/java/net/neoforged/gradle/userdev/runtime/extension/UserDevRuntimeExtension.java
@@ -7,6 +7,7 @@
 import net.neoforged.gradle.common.util.run.TypesUtil;
 import net.neoforged.gradle.dsl.common.extensions.Mappings;
 import net.neoforged.gradle.dsl.common.extensions.Minecraft;
+import net.neoforged.gradle.dsl.common.runs.run.Run;
 import net.neoforged.gradle.dsl.common.runs.type.RunType;
 import net.neoforged.gradle.dsl.common.runtime.tasks.tree.TaskTreeAdapter;
 import net.neoforged.gradle.dsl.common.tasks.WithOutput;
@@ -90,11 +91,16 @@ public UserDevRuntimeExtension(Project project) {
 
         spec.setMinecraftVersion(mcpRuntimeDefinition.getSpecification().getMinecraftVersion());
 
+        final NamedDomainObjectContainer<Run> runs = (NamedDomainObjectContainer<Run>) getProject().getExtensions().getByName(RunsConstants.Extensions.RUNS);
+        spec.getProject().afterEvaluate(project -> runs.stream().filter(run -> run.getIsJUnit().get())
+                .forEach(run -> spec.getProfile().getAdditionalTestDependencyArtifactCoordinates()
+                        .get().forEach(run.getDependencies().get().getRuntime()::add)));
+
         final NamedDomainObjectContainer<RunType> runTypes = (NamedDomainObjectContainer<RunType>) getProject().getExtensions().getByName(RunsConstants.Extensions.RUN_TYPES);
         userdevProfile.getRunTypes().forEach((type) -> {
             TypesUtil.registerWithPotentialPrefix(runTypes, spec.getIdentifier(), type.getName(), type::copyTo);
         });
-        
+
         return new UserDevRuntimeDefinition(
                 spec,
                 mcpRuntimeDefinition,