add toClassDependencies()

Signed-off-by: Dimitry Polivaev <[email protected]>
TNG · Feb 15, 2024 · f9305c6 · f9305c6
1 parent a95c7c3
commit f9305c6
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/archunit/src/main/java/com/tngtech/archunit/core/domain/JavaAccess.java b/archunit/src/main/java/com/tngtech/archunit/core/domain/JavaAccess.java
@@ -107,6 +107,14 @@ String additionalToStringFields() {
         return "";
     }
 
+    /**
+     * @return all matching class dependencies.
+     */
+    @PublicAPI(usage = ACCESS)
+    public Set<Dependency> toClassDependencies() {
+        return Dependency.tryCreateFromAccess(this);
+    }
+
     @Override
     @PublicAPI(usage = ACCESS)
     public String getDescription() {

diff --git a/archunit/src/main/java/com/tngtech/archunit/library/dependencies/SliceDependency.java b/archunit/src/main/java/com/tngtech/archunit/library/dependencies/SliceDependency.java
@@ -18,6 +18,7 @@
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Objects;
+import java.util.Set;
 import java.util.SortedSet;
 import java.util.TreeSet;
 
@@ -62,6 +63,11 @@ public Slice getTarget() {
         return target;
     }
 
+    @PublicAPI(usage = ACCESS)
+    public Set<Dependency> toClassDependencies() {
+        return relevantDependencies;
+    }
+
     @Override
     @PublicAPI(usage = ACCESS)
     public String getDescription() {

diff --git a/archunit/src/test/java/com/tngtech/archunit/core/domain/JavaAccessTest.java b/archunit/src/test/java/com/tngtech/archunit/core/domain/JavaAccessTest.java
@@ -5,15 +5,18 @@
 import com.tngtech.archunit.core.domain.JavaAccess.Functions.Get;
 import com.tngtech.archunit.core.importer.testexamples.SomeClass;
 import com.tngtech.archunit.core.importer.testexamples.SomeEnum;
+import org.assertj.core.api.Assertions;
 import org.junit.Test;
 
+import static com.google.common.collect.Iterables.getOnlyElement;
 import static com.tngtech.archunit.base.DescribedPredicate.alwaysFalse;
 import static com.tngtech.archunit.core.domain.TestUtils.importClassWithContext;
 import static com.tngtech.archunit.core.domain.TestUtils.importClassesWithContext;
 import static com.tngtech.archunit.core.domain.TestUtils.newMethodCallBuilder;
 import static com.tngtech.archunit.core.domain.TestUtils.resolvedTargetFrom;
 import static com.tngtech.archunit.core.domain.TestUtils.simulateCall;
 import static com.tngtech.archunit.testutil.Assertions.assertThat;
+import static com.tngtech.archunit.testutil.Assertions.assertThatType;
 
 public class JavaAccessTest {
     @Test
@@ -23,6 +26,9 @@ public void when_the_origin_is_an_inner_class_the_toplevel_class_is_displayed_as
                 .inLineNumber(7);
 
         assertThat(access.getDescription()).contains("(SomeClass.java:7)");
+        Dependency dependency = getOnlyElement(access.toClassDependencies());
+        assertThatType(dependency.getTargetClass()).as("target class").isEqualTo(access.getTargetOwner());
+        Assertions.assertThat(dependency.getDescription()).as("description").isEqualTo(access.getDescription());
     }
 
     @Test