Set compile time classpath for Starlark java_binary for compilation_info

PiperOrigin-RevId: 464293636
Change-Id: I78361c64f13192dcd6dfcc7019d7fef34be25471
diff --git a/src/main/java/com/google/devtools/build/lib/rules/java/JavaStarlarkCommon.java b/src/main/java/com/google/devtools/build/lib/rules/java/JavaStarlarkCommon.java
index 9d94744..46cefed 100644
--- a/src/main/java/com/google/devtools/build/lib/rules/java/JavaStarlarkCommon.java
+++ b/src/main/java/com/google/devtools/build/lib/rules/java/JavaStarlarkCommon.java
@@ -371,11 +371,13 @@
     if (javaInfo.getProvider(JavaCompilationInfoProvider.class) != null) {
       builder.addProvider(JavaCompilationInfoProvider.class, javaInfo.getCompilationInfoProvider());
     } else if (javaInfo.getProvider(JavaCompilationArgsProvider.class) != null) {
+      JavaCompilationArgsProvider compilationArgsProvider =
+          javaInfo.getProvider(JavaCompilationArgsProvider.class);
       builder.addProvider(
           JavaCompilationInfoProvider.class,
           new JavaCompilationInfoProvider.Builder()
-              .setRuntimeClasspath(
-                  javaInfo.getProvider(JavaCompilationArgsProvider.class).getRuntimeJars())
+              .setCompilationClasspath(compilationArgsProvider.getTransitiveCompileTimeJars())
+              .setRuntimeClasspath(compilationArgsProvider.getRuntimeJars())
               .build());
     }
     if (javaInfo.getProvider(JavaGenJarsProvider.class) != null) {