diff --git a/spring-data-jpa/src/test/java/org/springframework/data/jpa/mapping/JpaMetamodelMappingContextIntegrationTests.java b/spring-data-jpa/src/test/java/org/springframework/data/jpa/mapping/JpaMetamodelMappingContextIntegrationTests.java index 127abcedd..5febb4b46 100644 --- a/spring-data-jpa/src/test/java/org/springframework/data/jpa/mapping/JpaMetamodelMappingContextIntegrationTests.java +++ b/spring-data-jpa/src/test/java/org/springframework/data/jpa/mapping/JpaMetamodelMappingContextIntegrationTests.java @@ -17,16 +17,15 @@ package org.springframework.data.jpa.mapping; import static org.assertj.core.api.Assertions.*; -import java.util.Collections; - import jakarta.persistence.EntityManager; +import java.util.Collections; + import org.hibernate.proxy.HibernateProxy; import org.hibernate.proxy.LazyInitializer; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; import org.junit.jupiter.api.extension.ExtendWith; - import org.springframework.beans.factory.annotation.Autowired; import org.springframework.context.annotation.ComponentScan.Filter; import org.springframework.context.annotation.Configuration; @@ -145,8 +144,10 @@ class JpaMetamodelMappingContextIntegrationTests { IdentifierAccessor accessor = entity.getIdentifierAccessor(loadedProduct); assertThat(accessor.getIdentifier()).isEqualTo(category.getProduct().getId()); - assertThat(loadedProduct).isInstanceOf(HibernateProxy.class); - assertThat(((HibernateProxy) loadedProduct).getHibernateLazyInitializer().isUninitialized()).isTrue(); + + if (loadedProduct instanceof HibernateProxy proxy) { + assertThat(proxy.getHibernateLazyInitializer().isUninitialized()).isTrue(); + } status.setRollbackOnly(); diff --git a/spring-data-jpa/src/test/java/org/springframework/data/jpa/provider/PersistenceProviderIntegrationTests.java b/spring-data-jpa/src/test/java/org/springframework/data/jpa/provider/PersistenceProviderIntegrationTests.java index 392b5b3ad..c459d4e0e 100644 --- a/spring-data-jpa/src/test/java/org/springframework/data/jpa/provider/PersistenceProviderIntegrationTests.java +++ b/spring-data-jpa/src/test/java/org/springframework/data/jpa/provider/PersistenceProviderIntegrationTests.java @@ -19,10 +19,10 @@ import static org.assertj.core.api.Assertions.*; import jakarta.persistence.EntityManager; +import org.hibernate.proxy.HibernateProxy; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; import org.junit.jupiter.api.extension.ExtendWith; - import org.springframework.beans.factory.annotation.Autowired; import org.springframework.context.annotation.ComponentScan.Filter; import org.springframework.context.annotation.Configuration; @@ -74,8 +74,9 @@ public class PersistenceProviderIntegrationTests { Product product = categories.findById(category.getId()).get().getProduct(); ProxyIdAccessor accessor = PersistenceProvider.fromEntityManager(em); - assertThat(accessor.shouldUseAccessorFor(product)).isTrue(); - assertThat(accessor.getIdentifierFrom(product).toString()).isEqualTo((Object) product.getId().toString()); + if (product instanceof HibernateProxy proxy) { + assertThat(proxy.getHibernateLazyInitializer().isUninitialized()).isTrue(); + } return null; }