GH-2138 - Add additional test.

Author: meistermeier

src/test/java/org/springframework/data/neo4j/integration/imperative/InheritanceMappingIT.java

@@ -33,6 +33,7 @@
 
 import java.util.Collection;
 import java.util.Collections;
+import java.util.List;
 
 import static org.assertj.core.api.Assertions.assertThat;
 
@@ -103,6 +104,31 @@ void collectionsShouldHaveCorrectTypes(@Autowired TerritoryRepository repository
 		);
 	}
 
+	@Test // GH-2138
+	void resultCollectionShouldHaveCorrectTypes(@Autowired TerritoryRepository repository) {
+
+		try (Session session = driver.session()) {
+			session.run("CREATE (c:Country:BaseTerritory:BaseEntity{nameEn:'country', countryProperty:'baseCountry'}) " +
+					"CREATE (c)-[:LINK]->(:Country:BaseTerritory:BaseEntity{nameEn:'anotherCountry', countryProperty:'large'}) " +
+					"CREATE (c)-[:LINK]->(:Continent:BaseTerritory:BaseEntity{nameEn:'continent', continentProperty:'small'}) " +
+					"CREATE (c)-[:LINK]->(:GenericTerritory:BaseTerritory:BaseEntity{nameEn:'generic'})").consume();
+		}
+
+		List<Inheritance.BaseTerritory> territories = repository.findAll();
+
+		Inheritance.Country country1 = new Inheritance.Country("country", "baseCountry");
+		Inheritance.Country country2 = new Inheritance.Country("anotherCountry", "large");
+		Inheritance.Continent continent = new Inheritance.Continent("continent", "small");
+		Inheritance.GenericTerritory genericTerritory = new Inheritance.GenericTerritory("generic");
+
+		assertThat(territories).containsExactlyInAnyOrder(
+				country1,
+				country2,
+				continent,
+				genericTerritory
+		);
+	}
+
 	@Repository
 	interface BuildingRepository extends Neo4jRepository<Inheritance.Building, Long> {}