DATACMNS-376 - Repositories now inspects user class for factory lookups.

When looking up the RepositoryFactoryInformation for a domain type we now make sure we look up the user type to not accidentally try to do so for a persistence provider generated proxy class for the original type. This prevents us from not finding PersistentEntity instances, Repository- or EntityInformation accidentally.

src/main/java/org/springframework/data/repository/support/Repositories.java

@@ -37,6 +37,7 @@ import org.springframework.data.repository.core.RepositoryInformation;
 import org.springframework.data.repository.core.support.RepositoryFactoryInformation;
 import org.springframework.data.repository.query.QueryMethod;
 import org.springframework.util.Assert;
+import org.springframework.util.ClassUtils;
 
 /**
  * Wrapper class to access repository instances obtained from a {@link ListableBeanFactory}.
@@ -175,14 +176,15 @@ public class Repositories implements Iterable<Class<?>> {
 
 		// Create defensive copy of the keys to allow threads to potentially add values while iterating over them
 		Set<RepositoryFactoryInformation<Object, Serializable>> keys = Collections.unmodifiableSet(repositories.keySet());
+		Class<?> type = ClassUtils.getUserClass(domainClass);
 
 		for (RepositoryFactoryInformation<Object, Serializable> information : keys) {
-			if (domainClass.equals(information.getEntityInformation().getJavaType())) {
+			if (type.equals(information.getEntityInformation().getJavaType())) {
 				return information;
 			}
 		}
 
-		return lookupRepositoryFactoryInformationFor(domainClass);
+		return lookupRepositoryFactoryInformationFor(type);
 	}
 
 	/* 

src/test/java/org/springframework/data/repository/support/RepositoriesIntegrationTests.java

@@ -17,6 +17,7 @@ package org.springframework.data.repository.support;
 
 import static org.hamcrest.CoreMatchers.*;
 import static org.junit.Assert.*;
+import static org.mockito.Mockito.*;
 
 import org.junit.Test;
 import org.junit.runner.RunWith;
@@ -87,6 +88,16 @@ public class RepositoriesIntegrationTests {
 		assertThat(repositories.getCrudInvoker(Product.class), is(instanceOf(ReflectionRepositoryInvoker.class)));
 	}
 
+	/**
+	 * @see DATACMNS-376
+	 */
+	@Test
+	public void returnsPersistentEntityForProxiedClass() {
+
+		User user = mock(User.class);
+		assertThat(repositories.getPersistentEntity(user.getClass()), is(notNullValue()));
+	}
+
 	static class User {
 
 	}