diff --git a/src/main/java/org/springframework/data/mapping/context/PersistentEntities.java b/src/main/java/org/springframework/data/mapping/context/PersistentEntities.java index 15ce45c6a..ae5308ace 100644 --- a/src/main/java/org/springframework/data/mapping/context/PersistentEntities.java +++ b/src/main/java/org/springframework/data/mapping/context/PersistentEntities.java @@ -27,6 +27,7 @@ import java.util.function.BiFunction; import java.util.stream.Collectors; import java.util.stream.StreamSupport; +import org.springframework.data.mapping.MappingException; import org.springframework.data.mapping.PersistentEntity; import org.springframework.data.mapping.PersistentProperty; import org.springframework.data.util.Streamable; @@ -81,6 +82,8 @@ public class PersistentEntities implements Streamable>> getPersistentEntity(Class type) { @@ -100,9 +103,9 @@ public class PersistentEntities implements Streamable> getRequiredPersistentEntity(Class type) { @@ -113,7 +116,7 @@ public class PersistentEntities implements Streamable { - return new IllegalArgumentException(String.format( + return new MappingException(String.format( "Cannot get or create PersistentEntity for type %s! PersistentEntities knows about %s MappingContext instances and therefore cannot identify a single responsible one. Please configure the initialEntitySet through an entity scan using the base package in your configuration to pre initialize contexts.", type.getName(), contexts.size())); }); diff --git a/src/test/java/org/springframework/data/mapping/context/PersistentEntitiesUnitTests.java b/src/test/java/org/springframework/data/mapping/context/PersistentEntitiesUnitTests.java index 9b093965c..8acd4462e 100755 --- a/src/test/java/org/springframework/data/mapping/context/PersistentEntitiesUnitTests.java +++ b/src/test/java/org/springframework/data/mapping/context/PersistentEntitiesUnitTests.java @@ -27,6 +27,7 @@ import org.mockito.junit.jupiter.MockitoExtension; import org.springframework.data.annotation.Id; import org.springframework.data.annotation.Reference; +import org.springframework.data.mapping.MappingException; import org.springframework.data.mapping.PersistentEntity; import org.springframework.data.mapping.model.BasicPersistentEntity; import org.springframework.data.util.ClassTypeInformation; @@ -36,6 +37,7 @@ import org.springframework.data.util.ClassTypeInformation; * * @author Oliver Gierke * @author Christoph Strobl + * @author Mar Paluch */ @ExtendWith(MockitoExtension.class) class PersistentEntitiesUnitTests { @@ -157,7 +159,7 @@ class PersistentEntitiesUnitTests { @Test // DATACMNS-1780 void getRequiredPersistentEntityErrorsOnInFlightEntityCreationForUnknownTypesWhenHavingMultipleMappingContexts() { - assertThatExceptionOfType(IllegalArgumentException.class) + assertThatExceptionOfType(MappingException.class) .isThrownBy(() -> PersistentEntities.of(first, second).getRequiredPersistentEntity(Sample.class)); }