feat: Use main class for fast tests if there is any

bazelbuild · Nov 16, 2023 · 6468dfb · 6468dfb
1 parent 1e5adf2
commit 6468dfb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/java/src/com/google/idea/blaze/java/run/fastbuild/FastBuildTestEnvironmentCreator.java b/java/src/com/google/idea/blaze/java/run/fastbuild/FastBuildTestEnvironmentCreator.java
@@ -103,7 +103,11 @@ GeneralCommandLine createCommandLine(
         getTestClassProperty(),
         FastBuildTestClassFinder.getInstance(project).getTestClass(target, targetJavaInfo));
 
-    commandBuilder.setMainClass(getTestRunner());
+    if (targetJavaInfo.mainClass().isPresent()) {
+        commandBuilder.setMainClass(targetJavaInfo.mainClass().get());
+    } else {
+        commandBuilder.setMainClass(getTestRunner());
+    }
 
     commandBuilder
         .addEnvironmentVariable(TEST_BINARY_VARIABLE, getTestBinary(target))