diff options
Diffstat (limited to 'src/test/java/com/google/devtools/build/lib/analysis/DependencyResolverTest.java')
-rw-r--r-- | src/test/java/com/google/devtools/build/lib/analysis/DependencyResolverTest.java | 30 |
1 files changed, 23 insertions, 7 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/analysis/DependencyResolverTest.java b/src/test/java/com/google/devtools/build/lib/analysis/DependencyResolverTest.java index 26005cfc3d..4ee313c221 100644 --- a/src/test/java/com/google/devtools/build/lib/analysis/DependencyResolverTest.java +++ b/src/test/java/com/google/devtools/build/lib/analysis/DependencyResolverTest.java @@ -18,6 +18,7 @@ import static com.google.common.truth.Truth.assertWithMessage; import com.google.common.collect.ImmutableMap; import com.google.common.collect.ImmutableSet; +import com.google.common.collect.Streams; import com.google.devtools.build.lib.analysis.config.BuildConfiguration; import com.google.devtools.build.lib.analysis.config.BuildOptions; import com.google.devtools.build.lib.analysis.config.ConfigMatchingProvider; @@ -40,6 +41,9 @@ import com.google.devtools.build.lib.testutil.Suite; import com.google.devtools.build.lib.testutil.TestSpec; import com.google.devtools.build.lib.util.OrderedSetMultimap; import java.util.List; +import java.util.Map; +import java.util.function.Function; +import java.util.stream.Collectors; import javax.annotation.Nullable; import org.junit.Before; import org.junit.Test; @@ -80,14 +84,26 @@ public class DependencyResolverTest extends AnalysisTestCase { throw new IllegalStateException(e); } - @Nullable @Override - protected Target getTarget(Target from, Label label, NestedSetBuilder<Cause> rootCauses) { - try { - return packageManager.getTarget(reporter, label); - } catch (NoSuchPackageException | NoSuchTargetException | InterruptedException e) { - throw new IllegalStateException(e); - } + protected Map<Label, Target> getTargets( + Iterable<Label> labels, + Target fromTarget, + NestedSetBuilder<Cause> rootCauses, + int labelsSizeHint) { + return Streams.stream(labels) + .distinct() + .collect( + Collectors.toMap( + Function.identity(), + label -> { + try { + return packageManager.getTarget(reporter, label); + } catch (NoSuchPackageException + | NoSuchTargetException + | InterruptedException e) { + throw new IllegalStateException(e); + } + })); } @Nullable |