3434import static com .tngtech .archunit .core .domain .Formatters .formatNamesOf ;
3535import static com .tngtech .archunit .core .domain .JavaConstructor .CONSTRUCTOR_NAME ;
3636import static com .tngtech .archunit .core .domain .properties .HasName .Utils .namesOf ;
37+ import static com .tngtech .archunit .core .importer .ClassFileImporterTestUtils .getMethodCallsFromClassWithoutAutomaticNullCheck ;
3738import static com .tngtech .archunit .core .importer .JavaClassDescriptorImporterTestUtils .isLambdaMethodName ;
3839import static com .tngtech .archunit .testutil .Assertions .assertThat ;
3940import static com .tngtech .archunit .testutil .Assertions .assertThatAccess ;
@@ -68,7 +69,7 @@ Runnable call() {
6869 }
6970
7071 JavaClasses classes = new ClassFileImporter ().importClasses (Target .class , Caller .class );
71- JavaMethodCall call = getOnlyElement (classes .get (Caller .class ). getMethodCallsFromSelf ( ));
72+ JavaMethodCall call = getOnlyElement (getMethodCallsFromClassWithoutAutomaticNullCheck ( classes .get (Caller .class )));
7273
7374 assertThatCall (call ).isFrom ("call" ).isTo (Target .class , "target" );
7475 }
@@ -172,7 +173,7 @@ Consumer<String> call() {
172173 }
173174
174175 JavaClasses classes = new ClassFileImporter ().importClasses (Target .class , Caller .class );
175- JavaMethodCall call = getOnlyElement (classes .get (Caller .class ). getMethodCallsFromSelf ( ));
176+ JavaMethodCall call = getOnlyElement (getMethodCallsFromClassWithoutAutomaticNullCheck ( classes .get (Caller .class )));
176177
177178 assertThatCall (call ).isFrom ("call" ).isTo (Target .class , "target" , String .class );
178179 }
@@ -196,7 +197,7 @@ Supplier<String> call() {
196197 }
197198
198199 JavaClasses classes = new ClassFileImporter ().importClasses (Target .class , Caller .class );
199- JavaMethodCall call = getOnlyElement (classes .get (Caller .class ). getMethodCallsFromSelf ( ));
200+ JavaMethodCall call = getOnlyElement (getMethodCallsFromClassWithoutAutomaticNullCheck ( classes .get (Caller .class )));
200201
201202 assertThatCall (call ).isFrom ("call" ).isTo (Target .class , "target" );
202203 }
@@ -252,7 +253,7 @@ Runnable call() {
252253 }
253254
254255 JavaClasses classes = new ClassFileImporter ().importClasses (Target .class , Caller .class );
255- Set <JavaMethodCall > calls = classes .get (Caller .class ). getMethodCallsFromSelf ( );
256+ Set <JavaMethodCall > calls = getMethodCallsFromClassWithoutAutomaticNullCheck ( classes .get (Caller .class ));
256257
257258 assertThat (calls ).hasSize (2 );
258259 calls .forEach (call -> assertThatCall (call ).isFrom ("call" ).isTo (Target .class , "target" ));
@@ -285,7 +286,7 @@ Function<String, Target> call2() {
285286 }
286287
287288 JavaClasses classes = new ClassFileImporter ().importClasses (Target .class , Caller .class );
288- Set <JavaMethodCall > calls = classes .get (Caller .class ). getMethodCallsFromSelf ( );
289+ Set <JavaMethodCall > calls = getMethodCallsFromClassWithoutAutomaticNullCheck ( classes .get (Caller .class ));
289290 assertThat (calls ).hasSize (5 );
290291
291292 assertThat (filterOriginByName (calls , "call1" ))
@@ -361,10 +362,10 @@ Supplier<String> call() {
361362
362363 JavaClasses classes = new ClassFileImporter ().importClasses (Target .class , Caller1 .class , Caller2 .class );
363364
364- JavaMethodCall call1 = getOnlyElement (classes .get (Caller1 .class ). getMethodCallsFromSelf ( ));
365+ JavaMethodCall call1 = getOnlyElement (getMethodCallsFromClassWithoutAutomaticNullCheck ( classes .get (Caller1 .class )));
365366 assertThatCall (call1 ).isFrom ("call" ).isTo (Target .class , "target" );
366367
367- JavaMethodCall call2 = getOnlyElement (classes .get (Caller2 .class ). getMethodCallsFromSelf ( ));
368+ JavaMethodCall call2 = getOnlyElement (getMethodCallsFromClassWithoutAutomaticNullCheck ( classes .get (Caller2 .class )));
368369 assertThatCall (call2 ).isFrom ("call" ).isTo (Target .class , "target" );
369370 }
370371
0 commit comments