diff --git a/spring-core/src/main/java/org/springframework/aot/generate/GeneratedTypeReference.java b/spring-core/src/main/java/org/springframework/aot/generate/GeneratedTypeReference.java new file mode 100644 index 0000000000..9e2ec90886 --- /dev/null +++ b/spring-core/src/main/java/org/springframework/aot/generate/GeneratedTypeReference.java @@ -0,0 +1,60 @@ +/* + * Copyright 2002-2022 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.springframework.aot.generate; + +import org.springframework.aot.hint.AbstractTypeReference; +import org.springframework.aot.hint.TypeReference; +import org.springframework.javapoet.ClassName; +import org.springframework.lang.Nullable; +import org.springframework.util.Assert; + +/** + * A {@link TypeReference} for a generated {@linkplain ClassName type}. + * + * @author Stephane Nicoll + * @since 6.0 + */ +public final class GeneratedTypeReference extends AbstractTypeReference { + + private final ClassName className; + + private GeneratedTypeReference(ClassName className) { + super(className.packageName(), className.simpleName(), safeCreate(className.enclosingClassName())); + this.className = className; + } + + @Nullable + private static GeneratedTypeReference safeCreate(@Nullable ClassName className) { + return (className != null ? new GeneratedTypeReference(className) : null); + } + + public static GeneratedTypeReference of(ClassName className) { + Assert.notNull(className, "ClassName must not be null"); + return new GeneratedTypeReference(className); + } + + @Override + public String getCanonicalName() { + return this.className.canonicalName(); + } + + @Override + protected boolean isPrimitive() { + return this.className.isPrimitive(); + } + +} diff --git a/spring-core/src/test/java/org/springframework/aot/generate/GeneratedTypeReferenceTests.java b/spring-core/src/test/java/org/springframework/aot/generate/GeneratedTypeReferenceTests.java new file mode 100644 index 0000000000..7976304eda --- /dev/null +++ b/spring-core/src/test/java/org/springframework/aot/generate/GeneratedTypeReferenceTests.java @@ -0,0 +1,82 @@ +/* + * Copyright 2002-2022 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * https://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.springframework.aot.generate; + +import java.util.stream.Stream; + +import org.junit.jupiter.api.Test; +import org.junit.jupiter.params.ParameterizedTest; +import org.junit.jupiter.params.provider.Arguments; +import org.junit.jupiter.params.provider.MethodSource; + +import org.springframework.aot.hint.TypeReference; +import org.springframework.javapoet.ClassName; + +import static org.assertj.core.api.Assertions.assertThat; + +/** + * Tests for {@link GeneratedTypeReference}. + * + * @author Stephane Nicoll + */ +class GeneratedTypeReferenceTests { + + + @ParameterizedTest + @MethodSource("reflectionTargetNames") + void hasSuitableReflectionTargetName(TypeReference typeReference, String binaryName) { + assertThat(typeReference.getName()).isEqualTo(binaryName); + } + + static Stream reflectionTargetNames() { + return Stream.of( + Arguments.of(GeneratedTypeReference.of(ClassName.get("com.example", "Test")), "com.example.Test"), + Arguments.of(GeneratedTypeReference.of(ClassName.get("com.example", "Test", "Inner")), "com.example.Test$Inner")); + } + + @Test + void createWithClassName() { + GeneratedTypeReference typeReference = GeneratedTypeReference.of( + ClassName.get("com.example", "Test")); + assertThat(typeReference.getPackageName()).isEqualTo("com.example"); + assertThat(typeReference.getSimpleName()).isEqualTo("Test"); + assertThat(typeReference.getCanonicalName()).isEqualTo("com.example.Test"); + assertThat(typeReference.getEnclosingType()).isNull(); + } + + @Test + void createWithClassNameAndParent() { + GeneratedTypeReference typeReference = GeneratedTypeReference.of( + ClassName.get("com.example", "Test").nestedClass("Nested")); + assertThat(typeReference.getPackageName()).isEqualTo("com.example"); + assertThat(typeReference.getSimpleName()).isEqualTo("Nested"); + assertThat(typeReference.getCanonicalName()).isEqualTo("com.example.Test.Nested"); + assertThat(typeReference.getEnclosingType()).satisfies(parentTypeReference -> { + assertThat(parentTypeReference.getPackageName()).isEqualTo("com.example"); + assertThat(parentTypeReference.getSimpleName()).isEqualTo("Test"); + assertThat(parentTypeReference.getCanonicalName()).isEqualTo("com.example.Test"); + assertThat(parentTypeReference.getEnclosingType()).isNull(); + }); + } + + @Test + void equalsWithIdenticalCanonicalNameIsTrue() { + assertThat(GeneratedTypeReference.of(ClassName.get("java.lang", "String"))) + .isEqualTo(TypeReference.of(String.class)); + } + +}