Spring Data MongoDB:使用存储库进行单元测试

时间:2013-04-22 17:37:20

标签: junit4 spring-data spring-data-mongodb

如何在Spring Data MongoDB中使用存储库方法构建一些测试?我想为我的测试设置测试数据库,因为我不想为此目的使用生产数据库。应该可能,但我不知道。这是我的应用程序上下文:

<?xml version="1.0" encoding="UTF-8"?>
    <beans xmlns="http://www.springframework.org/schema/beans"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xmlns:context="http://www.springframework.org/schema/context"
         xmlns:mongo="http://www.springframework.org/schema/data/mongo"
         xmlns:neo4j="http://www.springframework.org/schema/data/neo4j"
         xsi:schemaLocation=
             "http://www.springframework.org/schema/context
              http://www.springframework.org/schema/context/spring-context-3.0.xsd
              http://www.springframework.org/schema/data/mongo
              http://www.springframework.org/schema/data/mongo/spring-mongo-1.0.xsd
              http://www.springframework.org/schema/beans
              http://www.springframework.org/schema/beans/spring-beans-3.0.xsd
              http://www.springframework.org/schema/data/neo4j
              http://www.springframework.org/schema/data/neo4j/spring-neo4j.xsd">

        <!-- Default bean name is 'mongo' -->
        <mongo:mongo host="${mongo.host}" port="${mongo.port}">
        <mongo:options connections-per-host="8"
            threads-allowed-to-block-for-connection-multiplier="4"
            connect-timeout="${mongo.connect-timeout}"
            max-wait-time="${mongo.max-wait-time}"
            auto-connect-retry="true"
            socket-keep-alive="true"
            socket-timeout="${mongo.socket-timeout}"
            slave-ok="true"
            write-number="1"
            write-timeout="0"
            write-fsync="true"/>
         </mongo:mongo>

         <bean id="mongoTemplate" class="org.springframework.data.mongodb.core.MongoTemplate">
            <constructor-arg ref="mongo" />
            <constructor-arg name="databaseName" value="${mongo.db}" />
         </bean>

        <context:component-scan base-package="domain.company.group.project.data.repositories"/>

        <!-- MongoDB repositories -->
        <mongo:repositories base-package="domain.company.group.project.data.repositories.mongodb"/>

        <!-- some other stuff -->

    </beans>

让我说我有一个简单的存储库如下:

public interface LocationRepository extends MongoRepository<Location, String>, LocationRepositoryCustom {

}

其中LocationRepositoryImpl是实现某个Location(域对象)类的所有自定义方法的类。我的测试类看起来像:

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations={"/test-context.xml"})
public class LocationRepositoryTest {

    @Autowired
    private LocationRepository locationRepository;

    /* Some tests... */
}

我试图在我的运行测试中嵌入一个MongoDB实例(如here所解释的那样)但它不起作用:建立了与测试数据库的连接但是mongo模板似乎无法覆盖所有保存方法继续将数据插入“生产”数据库。

我使用的是Spring 3.2.0和Spring Data Mongo 1.1.0.RELEASE。我正在使用Junit进行测试。

有什么建议吗?

提前谢谢。

1 个答案:

答案 0 :(得分:6)

Jaranda,

上周我遇到了同样的问题,巧合的是我听说过Fongo,“mongo的内存中java实现。”

所以我决定用它来测试我的自定义存储库并且对我很有效。下面是如何配置Spring以在JUnit测试中使用Fongo的示例。请注意,我没有使用xml配置。

希望这会有用!

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration
public class LocationRepositoryTest {

   private static final String PLAYER_ID = ObjectId.get().toString();

   @Autowired private LocationRepositoryCustom playerRepository;
   @Autowired private MongoTemplate mongoTemplate;

   /* Some tests... */

   @Configuration
   static class LocationRepositoryTestConfiguration {

        @Bean
        public Mongo mongo() {
            // Configure a Fongo instance
            return new Fongo("mongo-test").getMongo();
        }

        @Bean
        public MongoTemplate mongoTemplate() {
            return new MongoTemplate(mongo(), "collection-name");
        }

        @Bean
        public LocationRepositoryCustom playerRepository() {
            // This is necessary if MongoTemplate is an argument of custom implementation constructor
            return new LocationRepositoryCustomImpl(mongoTemplate());
        }
    }
}