(test) apply fix for github.com/TNG/ArchUnit/issues/413

jonathanlermitage · Oct 26, 2020 · 4230ebf · 4230ebf
1 parent 76f1573
commit 4230ebf
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/test/java/manon/util/ArchTest.java b/src/test/java/manon/util/ArchTest.java
@@ -66,13 +66,12 @@ public void check(@NotNull JavaClass item, ConditionEvents events) {
             public void check(@NotNull JavaClass item, ConditionEvents events) {
                 item.getMethods()
                     .forEach(method -> {
-                        // TODO can't detect collections of forbidden objects, see https://github.com/TNG/ArchUnit/issues/413
-                        JavaClass rawReturnType = method.getRawReturnType();
-                        if (rawReturnType.getName().contains("manon.document.")) {
+                        String typeName = method.reflect().getGenericReturnType().getTypeName();
+                        if (typeName.contains("manon.document.")) {
                             String message = String.format("Class %s's method %s returns an entity of type %s",
                                 item.getSimpleName(),
                                 method.getName(),
-                                rawReturnType.getName());
+                                typeName);
                             events.add(SimpleConditionEvent.violated(method, message));
                         }
                     });