From 9f21cc2f28dd47f5b82c46e8c26c8da10c63bfef Mon Sep 17 00:00:00 2001
From: Thomas Eizinger <thomas.eizinger@gmail.com>
Date: Sun, 18 Jan 2015 22:41:10 +0100
Subject: [PATCH] DATACMNS-634 - Repositories now als returns repositories for
 super types of a domain class.

In case the repository lookup for a given domain type fails we traverse the given types super-types and try to detect a repository for those.

Original pull request: #110.
---
 .../data/repository/support/Repositories.java     | 15 ++++++++++++---
 .../repository/support/RepositoriesUnitTests.java | 12 ++++++++++++
 2 files changed, 24 insertions(+), 3 deletions(-)

diff --git a/src/main/java/org/springframework/data/repository/support/Repositories.java b/src/main/java/org/springframework/data/repository/support/Repositories.java
index 5542eb939..70d5f5d68 100644
--- a/src/main/java/org/springframework/data/repository/support/Repositories.java
+++ b/src/main/java/org/springframework/data/repository/support/Repositories.java
@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2014 the original author or authors.
+ * Copyright 2012-2015 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.
@@ -39,6 +39,7 @@ import org.springframework.util.ClassUtils;
  * 
  * @author Oliver Gierke
  * @author Thomas Darimont
+ * @author Thomas Eizinger
  */
 public class Repositories implements Iterable<Class<?>> {
 
@@ -133,11 +134,19 @@ public class Repositories implements Iterable<Class<?>> {
 
 		Assert.notNull(domainClass, DOMAIN_TYPE_MUST_NOT_BE_NULL);
 
-		RepositoryFactoryInformation<Object, Serializable> repositoryInfo = repositoryFactoryInfos.get(ClassUtils
-				.getUserClass(domainClass));
+		Class<?> classToInspect = domainClass;
+		RepositoryFactoryInformation<Object, Serializable> repositoryInfo = repositoryFactoryInfos
+				.get(ClassUtils.getUserClass(classToInspect));
+
+		while (repositoryInfo == null && !classToInspect.equals(Object.class)) {
+			classToInspect = classToInspect.getSuperclass();
+			repositoryInfo = repositoryFactoryInfos.get(ClassUtils.getUserClass(classToInspect));
+		}
+
 		return repositoryInfo == null ? EMPTY_REPOSITORY_FACTORY_INFO : repositoryInfo;
 	}
 
+
 	/**
 	 * Returns the {@link EntityInformation} for the given domain class.
 	 * 
diff --git a/src/test/java/org/springframework/data/repository/support/RepositoriesUnitTests.java b/src/test/java/org/springframework/data/repository/support/RepositoriesUnitTests.java
index c08a8b026..09e47d423 100644
--- a/src/test/java/org/springframework/data/repository/support/RepositoriesUnitTests.java
+++ b/src/test/java/org/springframework/data/repository/support/RepositoriesUnitTests.java
@@ -108,10 +108,22 @@ public class RepositoriesUnitTests {
 		assertThat(repositories.getPersistentEntity(Address.class), is(notNullValue()));
 	}
 
+	/**
+	 * @see DATACMNS-634
+	 */
+	@Test
+	public void findsRepositoryForSubTypes() {
+
+		Repositories repositories = new Repositories(context);
+		assertThat(repositories.getPersistentEntity(AdvancedAddress.class), is(notNullValue()));
+	}
+
 	class Person {}
 
 	class Address {}
 
+	class AdvancedAddress extends Address {}
+
 	interface PersonRepository extends CrudRepository<Person, Long> {}
 
 	interface AddressRepository extends Repository<Address, Long> {}