blob: 1e87c3fdaeda6121f64b9ce0669fdad7b61cce4f [file] [log] [blame]
// Copyright 2015 The Bazel Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package com.google.devtools.build.lib.analysis;
import static com.google.common.truth.Truth.assertThat;
import static com.google.common.truth.Truth.assertWithMessage;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Iterables;
import com.google.devtools.build.lib.analysis.AspectCollection.AspectDeps;
import com.google.devtools.build.lib.analysis.util.AnalysisTestCase;
import com.google.devtools.build.lib.analysis.util.TestAspects;
import com.google.devtools.build.lib.causes.Cause;
import com.google.devtools.build.lib.cmdline.Label;
import com.google.devtools.build.lib.collect.nestedset.NestedSetBuilder;
import com.google.devtools.build.lib.packages.Aspect;
import com.google.devtools.build.lib.packages.AspectDescriptor;
import com.google.devtools.build.lib.packages.NativeAspectClass;
import com.google.devtools.build.lib.packages.NoSuchPackageException;
import com.google.devtools.build.lib.packages.NoSuchTargetException;
import com.google.devtools.build.lib.packages.Target;
import com.google.devtools.build.lib.util.OrderedSetMultimap;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
/**
* Tests for {@link DependencyResolver}.
*
* <p>These use custom rules so that all usual and unusual cases related to aspect processing can be
* tested.
*
* <p>It would be nicer is we didn't have a Skyframe executor, if we didn't have that, we'd need a
* way to create a configuration, a package manager and a whole lot of other things, so it's just
* easier this way.
*/
@RunWith(JUnit4.class)
public class DependencyResolverTest extends AnalysisTestCase {
private DependencyResolver dependencyResolver;
@Before
public final void createResolver() {
dependencyResolver =
new DependencyResolver() {
@Override
protected Map<Label, Target> getTargets(
OrderedSetMultimap<DependencyKind, Label> labelMap,
TargetAndConfiguration fromNode,
NestedSetBuilder<Cause> rootCauses) {
return labelMap.values().stream()
.distinct()
.collect(
Collectors.toMap(
Function.identity(),
label -> {
try {
return packageManager.getTarget(reporter, label);
} catch (NoSuchPackageException
| NoSuchTargetException
| InterruptedException e) {
throw new IllegalStateException(e);
}
}));
}
};
}
private void pkg(String name, String... contents) throws Exception {
scratch.file(name + "/BUILD", contents);
}
private OrderedSetMultimap<DependencyKind, DependencyKey> dependentNodeMap(
String targetName, NativeAspectClass aspect) throws Exception {
Target target =
packageManager.getTarget(reporter, Label.parseAbsolute(targetName, ImmutableMap.of()));
return dependencyResolver.dependentNodeMap(
new TargetAndConfiguration(target, getTargetConfiguration()),
getHostConfiguration(),
aspect != null ? Aspect.forNative(aspect) : null,
ImmutableMap.of(),
/*toolchainContexts=*/ null,
/*useToolchainTransition=*/ false,
/*trimmingTransitionFactory=*/ null);
}
private static void assertDep(
OrderedSetMultimap<DependencyKind, DependencyKey> dependentNodeMap,
String attrName,
String dep,
AspectDescriptor... aspects) {
DependencyKind kind = null;
for (DependencyKind candidate : dependentNodeMap.keySet()) {
if (candidate.getAttribute() != null && candidate.getAttribute().getName().equals(attrName)) {
kind = candidate;
break;
}
}
assertWithMessage("Attribute '" + attrName + "' not found").that(kind).isNotNull();
DependencyKey dependency = null;
for (DependencyKey depCandidate : dependentNodeMap.get(kind)) {
if (depCandidate.getLabel().toString().equals(dep)) {
dependency = depCandidate;
break;
}
}
assertWithMessage("Dependency '" + dep + "' on attribute '" + attrName + "' not found")
.that(dependency)
.isNotNull();
assertThat(Iterables.transform(dependency.getAspects().getUsedAspects(), AspectDeps::getAspect))
.containsExactlyElementsIn(aspects);
}
@Test
public void hasAspectsRequiredByRule() throws Exception {
setRulesAvailableInTests(TestAspects.ASPECT_REQUIRING_RULE, TestAspects.BASE_RULE);
pkg("a", "aspect(name='a', foo=[':b'])", "aspect(name='b', foo=[])");
OrderedSetMultimap<DependencyKind, DependencyKey> map = dependentNodeMap("//a:a", null);
assertDep(map, "foo", "//a:b", new AspectDescriptor(TestAspects.SIMPLE_ASPECT));
}
@Test
public void hasAspectsRequiredByAspect() throws Exception {
setRulesAvailableInTests(TestAspects.BASE_RULE, TestAspects.SIMPLE_RULE);
pkg("a", "simple(name='a', foo=[':b'])", "simple(name='b', foo=[])");
OrderedSetMultimap<DependencyKind, DependencyKey> map =
dependentNodeMap("//a:a", TestAspects.ATTRIBUTE_ASPECT);
assertDep(map, "foo", "//a:b", new AspectDescriptor(TestAspects.ATTRIBUTE_ASPECT));
}
@Test
public void hasAllAttributesAspect() throws Exception {
setRulesAvailableInTests(TestAspects.BASE_RULE, TestAspects.SIMPLE_RULE);
pkg("a", "simple(name='a', foo=[':b'])", "simple(name='b', foo=[])");
OrderedSetMultimap<DependencyKind, DependencyKey> map =
dependentNodeMap("//a:a", TestAspects.ALL_ATTRIBUTES_ASPECT);
assertDep(map, "foo", "//a:b", new AspectDescriptor(TestAspects.ALL_ATTRIBUTES_ASPECT));
}
@Test
public void hasAspectDependencies() throws Exception {
setRulesAvailableInTests(TestAspects.BASE_RULE);
pkg("a", "base(name='a')");
pkg("extra", "base(name='extra')");
OrderedSetMultimap<DependencyKind, DependencyKey> map =
dependentNodeMap("//a:a", TestAspects.EXTRA_ATTRIBUTE_ASPECT);
assertDep(map, "$dep", "//extra:extra");
}
}