Skip to content

Commit

Permalink
Improve ArchCondition that checks for transitive dependencies #878
Browse files Browse the repository at this point in the history
This adds an ArchCondition to the fluent API to check whether there is any matching transitive dependency. This is a much more efficient variant of `#transitivelyDependOnClassesThat` that can be used to detect transitive dependencies in a huge codebase or to classes in large 3rd-party libraries like the Android SDK. It focuses on detecting all *direct* dependencies of the selected classes that are themselves matched or have any transitive dependencies on matched classes. Thus, it doesn't discover all possible dependency paths but stops at the first match to be fast and resource-friendly.

Sample usage:
```
noClasses().that.resideInAPackage(“de.foo.service..”)
    .should().transitivelyDependOnAnyClassesThat.resideInAPackage("android..")
```

Then this Architecture
![image](https://user-images.githubusercontent.com/17569373/172873445-17111662-30e2-4388-912e-840a105cd2bc.png)

would output the following violations:

```
java.lang.AssertionError: Architecture Violation [Priority: MEDIUM] - Rule 'no classes that reside in a package 'de.foo.service..' should transitively depend on any classes that reside in a package ['android..'] was violated (3 times):
Class <de.foo.service.B> transitively depends on <android.I> by [F->E->android.I] in (B.java:0)
Class <de.foo.service.B> transitively depends on <android.I> by [E->android.I] in (B.java:0)
Class <de.foo.service.C> transitively depends on <android.H> by [D->android.H] in (C.java:0)
```

Resolves: #780 
Resolves: #826
  • Loading branch information
codecholeric authored Jul 8, 2022
2 parents 65843a8 + 499020a commit 5e56c99
Show file tree
Hide file tree
Showing 5 changed files with 213 additions and 27 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,10 @@
import java.util.List;

import com.tngtech.archunit.PublicAPI;
import com.tngtech.archunit.base.HasDescription;
import com.tngtech.archunit.core.domain.JavaClass;
import com.tngtech.archunit.core.domain.JavaMethodCall;
import com.tngtech.archunit.core.domain.properties.HasSourceCodeLocation;

import static com.tngtech.archunit.PublicAPI.State.EXPERIMENTAL;
import static com.tngtech.archunit.PublicAPI.Usage.ACCESS;
Expand Down Expand Up @@ -66,6 +69,21 @@ public interface ConditionEvent {
@PublicAPI(usage = INHERITANCE, state = EXPERIMENTAL)
void handleWith(Handler handler);

/**
* Convenience method to create a standard ArchUnit {@link ConditionEvent} message. It will prepend the
* description of the object that caused the event (e.g. a {@link JavaClass}) and append the source code
* location of the respective object.
*
* @param object The object to describe, e.g. the {@link JavaClass} {@code com.example.SomeClass}
* @param message The message that should be filled into the template, e.g. "does not have simple name 'Correct'"
* @return The formatted message, e.g. {@code Class <com.example.SomeClass> does not have simple name 'Correct' in (SomeClass.java:0)}
* @param <T> The object described by the event.
*/
@PublicAPI(usage = ACCESS)
static <T extends HasDescription & HasSourceCodeLocation> String createMessage(T object, String message) {
return object.getDescription() + " " + message + " in " + object.getSourceCodeLocation();
}

/**
* Handles the data of a {@link ConditionEvent} that is the corresponding objects and the description
* (compare {@link #handleWith(Handler)}).<br>
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,6 @@
import static com.tngtech.archunit.core.domain.JavaClass.Functions.GET_FIELD_ACCESSES_FROM_SELF;
import static com.tngtech.archunit.core.domain.JavaClass.Functions.GET_METHOD_CALLS_FROM_SELF;
import static com.tngtech.archunit.core.domain.JavaClass.Functions.GET_PACKAGE_NAME;
import static com.tngtech.archunit.core.domain.JavaClass.Functions.GET_TRANSITIVE_DEPENDENCIES_FROM_SELF;
import static com.tngtech.archunit.core.domain.JavaClass.Predicates.assignableFrom;
import static com.tngtech.archunit.core.domain.JavaClass.Predicates.assignableTo;
import static com.tngtech.archunit.core.domain.JavaClass.Predicates.equivalentTo;
Expand Down Expand Up @@ -117,6 +116,7 @@
import static com.tngtech.archunit.core.domain.properties.HasReturnType.Predicates.rawReturnType;
import static com.tngtech.archunit.core.domain.properties.HasThrowsClause.Predicates.throwsClauseContainingType;
import static com.tngtech.archunit.core.domain.properties.HasType.Predicates.rawType;
import static com.tngtech.archunit.lang.ConditionEvent.createMessage;
import static com.tngtech.archunit.lang.conditions.ArchPredicates.have;
import static java.util.Arrays.asList;

Expand Down Expand Up @@ -304,10 +304,7 @@ public static ArchCondition<JavaClass> dependOnClassesThat(final DescribedPredic

@PublicAPI(usage = ACCESS)
public static ArchCondition<JavaClass> transitivelyDependOnClassesThat(final DescribedPredicate<? super JavaClass> predicate) {
return new AnyDependencyCondition(
"transitively depend on classes that " + predicate.getDescription(),
GET_TARGET_CLASS.is(predicate),
GET_TRANSITIVE_DEPENDENCIES_FROM_SELF);
return new TransitiveDependencyCondition(predicate);
}

@PublicAPI(usage = ACCESS)
Expand Down Expand Up @@ -1284,10 +1281,6 @@ public static ArchCondition<JavaCodeUnit> onlyBeCalledByConstructorsThat(Describ
origin.is(constructor().and(predicate)), GET_CALLS_OF_SELF);
}

private static <T extends HasDescription & HasSourceCodeLocation> String createMessage(T object, String message) {
return object.getDescription() + " " + message + " in " + object.getSourceCodeLocation();
}

private static final IsConditionByPredicate<JavaClass> BE_TOP_LEVEL_CLASSES =
new IsConditionByPredicate<>("a top level class", JavaClass.Predicates.TOP_LEVEL_CLASSES);
private static final IsConditionByPredicate<JavaClass> BE_NESTED_CLASSES =
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,125 @@
/*
* Copyright 2014-2022 TNG Technology Consulting GmbH
*
* 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.tngtech.archunit.lang.conditions;

import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import com.google.common.collect.ImmutableList;
import com.tngtech.archunit.base.DescribedPredicate;
import com.tngtech.archunit.core.domain.JavaClass;
import com.tngtech.archunit.lang.ArchCondition;
import com.tngtech.archunit.lang.ConditionEvent;
import com.tngtech.archunit.lang.ConditionEvents;
import com.tngtech.archunit.lang.SimpleConditionEvent;

import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.collect.Iterables.getLast;
import static com.tngtech.archunit.lang.ConditionEvent.createMessage;
import static java.util.stream.Collectors.joining;
import static java.util.stream.Collectors.toSet;

public class TransitiveDependencyCondition extends ArchCondition<JavaClass> {

private final DescribedPredicate<? super JavaClass> conditionPredicate;
private final TransitiveDependencyPath transitiveDependencyPath = new TransitiveDependencyPath();
private Collection<JavaClass> allClasses;

public TransitiveDependencyCondition(DescribedPredicate<? super JavaClass> conditionPredicate) {
super("transitively depend on classes that " + conditionPredicate.getDescription());

this.conditionPredicate = checkNotNull(conditionPredicate);
}

@Override
public void init(Collection<JavaClass> allObjectsToTest) {
this.allClasses = allObjectsToTest;
}

@Override
public void check(JavaClass javaClass, ConditionEvents events) {
boolean hasTransitiveDependency = false;
for (JavaClass target : getDirectDependencyTargetsOutsideOfAnalyzedClasses(javaClass)) {
List<JavaClass> dependencyPath = transitiveDependencyPath.findPathTo(target);
if (!dependencyPath.isEmpty()) {
events.add(newTransitiveDependencyPathFoundEvent(javaClass, dependencyPath));
hasTransitiveDependency = true;
}
}
if (!hasTransitiveDependency) {
events.add(newNoTransitiveDependencyPathFoundEvent(javaClass));
}
}

private static ConditionEvent newTransitiveDependencyPathFoundEvent(JavaClass javaClass, List<JavaClass> transitiveDependencyPath) {
String message = String.format("%sdepends on <%s>",
transitiveDependencyPath.size() > 1 ? "transitively " : "",
getLast(transitiveDependencyPath).getFullName());

if (transitiveDependencyPath.size() > 1) {
message += " by [" + transitiveDependencyPath.stream().map(JavaClass::getName).collect(joining("->")) + "]";
}

return SimpleConditionEvent.satisfied(javaClass, createMessage(javaClass, message));
}

private static ConditionEvent newNoTransitiveDependencyPathFoundEvent(JavaClass javaClass) {
return SimpleConditionEvent.violated(javaClass, createMessage(javaClass, "does not transitively depend on any matching class"));
}

private Set<JavaClass> getDirectDependencyTargetsOutsideOfAnalyzedClasses(JavaClass item) {
return item.getDirectDependenciesFromSelf().stream()
.map(dependency -> dependency.getTargetClass().getBaseComponentType())
.filter(targetClass -> !allClasses.contains(targetClass))
.collect(toSet());
}

private class TransitiveDependencyPath {
/**
* @return some outgoing transitive dependency path to the supplied class or empty if there is none
*/
List<JavaClass> findPathTo(JavaClass clazz) {
ImmutableList.Builder<JavaClass> transitivePath = ImmutableList.builder();
addDependenciesToPathFrom(clazz, transitivePath, new HashSet<>());
return transitivePath.build().reverse();
}

private boolean addDependenciesToPathFrom(
JavaClass clazz,
ImmutableList.Builder<JavaClass> dependencyPath,
Set<JavaClass> analyzedClasses
) {
if (conditionPredicate.test(clazz)) {
dependencyPath.add(clazz);
return true;
}

analyzedClasses.add(clazz);

for (JavaClass directDependency : getDirectDependencyTargetsOutsideOfAnalyzedClasses(clazz)) {
if (!analyzedClasses.contains(directDependency)
&& addDependenciesToPathFrom(directDependency, dependencyPath, analyzedClasses)) {
dependencyPath.add(clazz);
return true;
}
}

return false;
}
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -1016,7 +1016,10 @@ public interface ClassesShould {
ClassesShouldConjunction onlyDependOnClassesThat(DescribedPredicate<? super JavaClass> predicate);

/**
* Asserts that all classes selected by this rule transitively depend on certain classes.<br>
* Asserts that all classes selected by this rule transitively depend on any matching classes.<br>
* It focuses on detecting all <strong>direct</strong> dependencies of the selected classes that are themselves matched or have any
* transitive dependencies on matched classes. Thus, it doesn't discover all possible dependency paths but stops at the first match to be fast and
* resource-friendly.<br>
* NOTE: This usually makes more sense the negated way, e.g.
* <p>
* <pre><code>
Expand All @@ -1031,7 +1034,10 @@ public interface ClassesShould {
ClassesThat<ClassesShouldConjunction> transitivelyDependOnClassesThat();

/**
* Asserts that all classes selected by this rule transitively depend on certain classes.<br>
* Asserts that all classes selected by this rule transitively depend on any matching classes.<br>
* It focuses on detecting all <strong>direct</strong> dependencies of the selected classes that are themselves matched or have any
* transitive dependencies on matched classes. Thus, it doesn't discover all possible dependency paths but stops at the first match to be fast and
* resource-friendly.<br>
* NOTE: This usually makes more sense the negated way, e.g.
* <p>
* <pre><code>
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@

import com.tngtech.archunit.base.DescribedPredicate;
import com.tngtech.archunit.core.domain.JavaClass;
import com.tngtech.archunit.core.domain.JavaClass.Predicates;
import com.tngtech.archunit.core.domain.JavaClasses;
import com.tngtech.archunit.core.domain.properties.HasName;
import com.tngtech.archunit.core.domain.properties.HasType;
Expand All @@ -36,7 +37,6 @@
import static com.tngtech.archunit.core.domain.properties.HasName.Functions.GET_NAME;
import static com.tngtech.archunit.core.domain.properties.HasName.Predicates.name;
import static com.tngtech.archunit.core.domain.properties.HasType.Functions.GET_RAW_TYPE;
import static com.tngtech.archunit.lang.conditions.ArchConditions.fullyQualifiedName;
import static com.tngtech.archunit.lang.conditions.ArchPredicates.are;
import static com.tngtech.archunit.lang.conditions.ArchPredicates.have;
import static com.tngtech.archunit.lang.syntax.ArchRuleDefinition.classes;
Expand Down Expand Up @@ -1711,46 +1711,90 @@ private static class TransitivelyDependOnClassesThatTestCases {
static class TestClass {
DirectlyDependentClass1 directDependency1;
DirectlyDependentClass2 directDependency2;
DirectlyDependentClass3 directDependency3;
}

@SuppressWarnings("unused")
static class TestClassNotViolatingBecauseOnlyDependingOnOtherSelectedClass {
TestClass testClass;
}

@SuppressWarnings("unused")
static class DirectlyDependentClass1 {
TransitivelyDependentClass transitiveDependency1;
Level1TransitivelyDependentClass1 transitiveDependency1;
}

@SuppressWarnings("unused")
static class DirectlyDependentClass2{
static class DirectlyDependentClass2 {
DirectlyDependentClass1 otherDependency;
TransitivelyDependentClass transitiveDependency2;
Level2TransitivelyDependentClass2 transitiveDependency2;
}

static class TransitivelyDependentClass {
static class DirectlyDependentClass3 {
}

@SuppressWarnings("unused")
static class Level1TransitivelyDependentClass1 {
Level2TransitivelyDependentClass1 transitiveDependency1;
}

static class Level2TransitivelyDependentClass1 {
}

@SuppressWarnings("unused")
static class Level2TransitivelyDependentClass2 {
Level2TransitivelyDependentClass1 transitiveDependency1;
}
}

@Test
@DataProvider(value = {"true", "false"})
public void transitivelyDependOnClassesThat_reports_all_transitive_dependencies(boolean viaPredicate) {
Class<?> testClass = TransitivelyDependOnClassesThatTestCases.TestClass.class;
Class<?> testClass1 = TransitivelyDependOnClassesThatTestCases.TestClass.class;
Class<?> testClass2 = TransitivelyDependOnClassesThatTestCases.TestClassNotViolatingBecauseOnlyDependingOnOtherSelectedClass.class;
Class<?> directlyDependentClass1 = TransitivelyDependOnClassesThatTestCases.DirectlyDependentClass1.class;
Class<?> directlyDependentClass2 = TransitivelyDependOnClassesThatTestCases.DirectlyDependentClass2.class;
Class<?> transitivelyDependentClass = TransitivelyDependOnClassesThatTestCases.TransitivelyDependentClass.class;
Class<?> directlyDependentClass3 = TransitivelyDependOnClassesThatTestCases.DirectlyDependentClass3.class;
Class<?> level1TransitivelyDependentClass1 = TransitivelyDependOnClassesThatTestCases.Level1TransitivelyDependentClass1.class;
Class<?> level2TransitivelyDependentClass1 = TransitivelyDependOnClassesThatTestCases.Level2TransitivelyDependentClass1.class;
Class<?> level2TransitivelyDependentClass2 = TransitivelyDependOnClassesThatTestCases.Level2TransitivelyDependentClass2.class;
Class<?>[] matchingTransitivelyDependentClasses =
new Class<?>[]{level2TransitivelyDependentClass1, level2TransitivelyDependentClass2, directlyDependentClass3};

JavaClasses classes = new ClassFileImporter().importClasses(
testClass, directlyDependentClass1, directlyDependentClass2, transitivelyDependentClass
testClass1,
testClass2,
directlyDependentClass1,
directlyDependentClass2,
directlyDependentClass3,
level1TransitivelyDependentClass1,
level2TransitivelyDependentClass1,
level2TransitivelyDependentClass2
);

ClassesShould noClassesShould = noClasses().that().haveFullyQualifiedName(testClass.getName()).should();
ClassesShould noClassesShould = noClasses().that().haveSimpleNameStartingWith("TestClass").should();
ArchRule rule = viaPredicate
? noClassesShould.transitivelyDependOnClassesThat(have(fullyQualifiedName(transitivelyDependentClass.getName())))
: noClassesShould.transitivelyDependOnClassesThat().haveFullyQualifiedName(transitivelyDependentClass.getName());
? noClassesShould.transitivelyDependOnClassesThat(Predicates.belongToAnyOf(matchingTransitivelyDependentClasses))
: noClassesShould.transitivelyDependOnClassesThat().belongToAnyOf(matchingTransitivelyDependentClasses);

assertThatRule(rule).checking(classes)
.hasViolations(2)
.hasViolationMatching(String.format(".*%s\\.%s.* has type .*%s.*",
quote(directlyDependentClass1.getName()), "transitiveDependency1", quote(transitivelyDependentClass.getName())
.hasViolations(3)
.hasViolationMatching(String.format(".*<%s> transitively depends on <(?:%s|%s)> by \\[%s->.*\\] in .*",
quote(testClass1.getName()),
quote(level2TransitivelyDependentClass1.getName()),
quote(level2TransitivelyDependentClass2.getName()),
quote(directlyDependentClass2.getName())
))
.hasViolationMatching(String.format(".*<%s> transitively depends on <%s> by \\[%s->%s->%s\\] in .*",
quote(testClass1.getName()),
quote(level2TransitivelyDependentClass1.getName()),
quote(directlyDependentClass1.getName()),
quote(level1TransitivelyDependentClass1.getName()),
quote(level2TransitivelyDependentClass1.getName())
))
.hasViolationMatching(String.format(".*%s\\.%s.* has type .*%s.*",
quote(directlyDependentClass2.getName()), "transitiveDependency2", quote(transitivelyDependentClass.getName())
.hasViolationMatching(String.format(".*<%s> depends on <%s> in .*",
quote(testClass1.getName()),
quote(directlyDependentClass3.getName())
));
}

Expand Down

0 comments on commit 5e56c99

Please sign in to comment.