issue_122 use generated asseertion package or default

assertj · Sep 17, 2024 · 3254b16 · 3254b16
1 parent e2e4d1b
commit 3254b16
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/org/assertj/assertions/generator/BaseAssertionGenerator.java b/src/main/java/org/assertj/assertions/generator/BaseAssertionGenerator.java
@@ -497,11 +497,11 @@ private String generateAssertionEntryPointMethodsFor(final Set<ClassDescription>
       // resolve class assert (ex: PlayerAssert)
       // in case of inner classes like Movie.PublicCategory, class assert will be MoviePublicCategoryAssert
       assertionEntryPointMethodContent = replace(assertionEntryPointMethodContent, CUSTOM_ASSERTION_CLASS,
-                                                 classDescription.getFullyQualifiedAssertClassName());
+                                                 getAssertClassName(classDescription));
       // resolve class (ex: Player)
       // in case of inner classes like Movie.PublicCategory use class name with outer class i.e. Movie.PublicCategory.
       assertionEntryPointMethodContent = replace(assertionEntryPointMethodContent, CLASS_TO_ASSERT,
-              getAssertClassName(classDescription));
+              classDescription.getFullyQualifiedClassName());
 
       allAssertThatsContentBuilder.append(lineSeparator).append(assertionEntryPointMethodContent);
     }