Internal change

PiperOrigin-RevId: 696024387
google · Feb 12, 2025 · 35bef7e · 35bef7e
1 parent 3c2e1e7
commit 35bef7e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/src/java/com/google/devtools/mobileharness/shared/file/resolver/AbstractFileResolver.java b/src/java/com/google/devtools/mobileharness/shared/file/resolver/AbstractFileResolver.java
@@ -69,7 +69,7 @@ protected abstract ResolveResult actuallyResolve(ResolveSource resolveSource)
 
   /** Do the real pre process work. */
   protected Set<ResolveResult> actuallyPreBatchProcess(List<ResolveSource> resolveSources)
-      throws MobileHarnessException, InterruptedException {
+      throws MobileHarnessException, InterruptedException, ExecutionException {
     return ImmutableSet.of();
   }
 
@@ -156,7 +156,7 @@ public final ListenableFuture<List<Optional<ResolveResult>>> resolveAsync(
    * @param resolveSources all the files that one mobile test needs
    */
   protected Set<ResolveResult> preBatchProcess(List<ResolveSource> resolveSources)
-      throws MobileHarnessException, InterruptedException {
+      throws MobileHarnessException, InterruptedException, ExecutionException {
     Set<ResolveResult> resolveResults =
         actuallyPreBatchProcess(
             resolveSources.stream().filter(this::shouldActuallyResolve).collect(toImmutableList()));

diff --git a/src/java/com/google/devtools/mobileharness/shared/file/resolver/CacheFileResolver.java b/src/java/com/google/devtools/mobileharness/shared/file/resolver/CacheFileResolver.java
@@ -282,7 +282,7 @@ private void removeItemIfValueMatch(CachedResolveSource key, CachedResolveResult
 
   @Override
   protected Set<ResolveResult> preBatchProcess(List<ResolveSource> resolveSources)
-      throws MobileHarnessException, InterruptedException {
+      throws MobileHarnessException, InterruptedException, ExecutionException {
     Set<ResolveResult> resolveResults = super.preBatchProcess(resolveSources);
     for (ResolveResult resolveResult : resolveResults) {
       resolvedResultsCache.putIfAbsent(

diff --git a/src/javatests/com/google/devtools/mobileharness/shared/file/resolver/FileResolverTest.java b/src/javatests/com/google/devtools/mobileharness/shared/file/resolver/FileResolverTest.java
@@ -30,6 +30,7 @@
 import java.util.List;
 import java.util.Optional;
 import java.util.Set;
+import java.util.concurrent.ExecutionException;
 import java.util.concurrent.Executors;
 import java.util.function.Function;
 import org.junit.Rule;
@@ -133,7 +134,8 @@ public void batchResolve_parallelWithRealThreads()
   }
 
   @Test
-  public void preBatchProcess() throws MobileHarnessException, InterruptedException {
+  public void preBatchProcess()
+      throws MobileHarnessException, InterruptedException, ExecutionException {
     resolver =
         new FileResolverUnderTest(
             MoreExecutors.newDirectExecutorService(), source -> source.path().equals("/a/b"));