Skip to content

Commit 1eae0bb

Browse files
committed
Add additional assert proving type is in use
1 parent bd5fafa commit 1eae0bb

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

components/sbm-core/src/test/java/org/springframework/sbm/java/migration/conditions/HasMemberAnnotationTest.java

+11-2
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616
package org.springframework.sbm.java.migration.conditions;
1717

1818
import org.openrewrite.java.marker.JavaSourceSet;
19+
import org.openrewrite.java.tree.J;
20+
import org.openrewrite.java.tree.JavaType;
1921
import org.springframework.sbm.engine.context.ProjectContext;
2022
import org.springframework.sbm.project.resource.TestProjectContext;
2123
import org.junit.jupiter.api.Test;
@@ -42,11 +44,18 @@ class AnnotatedClass {
4244
.withBuildFileHavingDependencies("javax.validation:validation-api:2.0.1.Final")
4345
.build();
4446

45-
boolean isTypeResolved = context.getProjectJavaSources().list().get(0).getResource().getSourceFile().getMarkers().findFirst(JavaSourceSet.class).get().getClasspath().stream().map(fq -> fq.getFullyQualifiedName()).anyMatch(fq -> fq.startsWith("javax.validation"));
47+
J.CompilationUnit sourceFile = context.getProjectJavaSources().list().get(0).getResource().getSourceFile();
48+
boolean isTypeResolved = sourceFile.getMarkers().findFirst(JavaSourceSet.class).get().getClasspath().stream().map(fq -> fq.getFullyQualifiedName()).anyMatch(fq -> fq.startsWith("javax.validation"));
4649
assertThat(isTypeResolved).isTrue();
4750

51+
52+
String annotation = "javax.validation.constraints.Min";
4853
HasMemberAnnotation sut = new HasMemberAnnotation();
49-
sut.setAnnotation("javax.validation.constraints.Min");
54+
sut.setAnnotation(annotation);
55+
56+
boolean isTypeInUse = sourceFile.getTypesInUse().getTypesInUse().stream()
57+
.anyMatch(t -> ((JavaType.FullyQualified)t).getFullyQualifiedName().equals(annotation));
58+
assertThat(isTypeInUse).isTrue();
5059

5160
assertThat(sut.evaluate(context)).isTrue();
5261
}

0 commit comments

Comments
 (0)