diff --git a/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/repository/support/MongoRepositoryFactory.java b/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/repository/support/MongoRepositoryFactory.java index 9a78014b2..e03a7eed1 100644 --- a/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/repository/support/MongoRepositoryFactory.java +++ b/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/repository/support/MongoRepositoryFactory.java @@ -81,7 +81,7 @@ public class MongoRepositoryFactory extends RepositoryFactorySupport { protected Object getTargetRepository(RepositoryMetadata metadata) { Class repositoryInterface = metadata.getRepositoryInterface(); - MongoEntityInformation entityInformation = getEntityInformation(metadata.getDomainClass()); + MongoEntityInformation entityInformation = getEntityInformation(metadata.getDomainType()); if (isQueryDslRepository(repositoryInterface)) { return new QueryDslMongoRepository(entityInformation, mongoOperations); diff --git a/spring-data-mongodb/src/test/java/org/springframework/data/mongodb/repository/support/MongoRepositoryFactoryBeanUnitTests.java b/spring-data-mongodb/src/test/java/org/springframework/data/mongodb/repository/support/MongoRepositoryFactoryBeanUnitTests.java index d6bbf5b96..9a1cfd41c 100644 --- a/spring-data-mongodb/src/test/java/org/springframework/data/mongodb/repository/support/MongoRepositoryFactoryBeanUnitTests.java +++ b/spring-data-mongodb/src/test/java/org/springframework/data/mongodb/repository/support/MongoRepositoryFactoryBeanUnitTests.java @@ -1,5 +1,5 @@ /* - * Copyright 2011 the original author or authors. + * Copyright 2011-2012 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. @@ -15,7 +15,7 @@ */ package org.springframework.data.mongodb.repository.support; -import static org.hamcrest.CoreMatchers.*; +import static org.hamcrest.Matchers.*; import static org.junit.Assert.*; import static org.mockito.Mockito.*; @@ -28,7 +28,6 @@ import org.mockito.runners.MockitoJUnitRunner; import org.springframework.data.mapping.context.MappingContext; import org.springframework.data.mongodb.core.MongoOperations; import org.springframework.data.mongodb.core.convert.MongoConverter; -import org.springframework.data.repository.core.support.QueryCreationListener; import org.springframework.data.repository.core.support.RepositoryFactorySupport; import org.springframework.test.util.ReflectionTestUtils; @@ -57,21 +56,21 @@ public class MongoRepositoryFactoryBeanUnitTests { MongoRepositoryFactoryBean factory = new MongoRepositoryFactoryBean(); factory.setCreateIndexesForQueryMethods(true); - List> listeners = getListenersFromFactory(factory); - assertThat(listeners.size(), is(1)); - assertThat(listeners.get(0), is(instanceOf(IndexEnsuringQueryCreationListener.class))); + List listeners = getListenersFromFactory(factory); + assertThat(listeners.isEmpty(), is(false)); + assertThat(listeners, hasItem(instanceOf(IndexEnsuringQueryCreationListener.class))); } @Test @SuppressWarnings("rawtypes") public void doesNotAddIndexEnsuringQueryCreationListenerByDefault() { - List> listeners = getListenersFromFactory(new MongoRepositoryFactoryBean()); - assertThat(listeners.isEmpty(), is(true)); + List listeners = getListenersFromFactory(new MongoRepositoryFactoryBean()); + assertThat(listeners.size(), is(1)); } @SuppressWarnings({ "unchecked", "rawtypes" }) - private List> getListenersFromFactory(MongoRepositoryFactoryBean factoryBean) { + private List getListenersFromFactory(MongoRepositoryFactoryBean factoryBean) { when(operations.getConverter()).thenReturn(converter); when(converter.getMappingContext()).thenReturn(context); @@ -80,6 +79,6 @@ public class MongoRepositoryFactoryBeanUnitTests { factoryBean.afterPropertiesSet(); RepositoryFactorySupport factory = factoryBean.createRepositoryFactory(); - return (List>) ReflectionTestUtils.getField(factory, "queryPostProcessors"); + return (List) ReflectionTestUtils.getField(factory, "queryPostProcessors"); } }