在Mongo生产数据库和测试数据库之间切换-Java Spring引导_Java_Spring_Mongodb

在Mongo生产数据库和测试数据库之间切换-Java Spring引导

java spring mongodb

在Mongo生产数据库和测试数据库之间切换-Java Spring引导,java,spring,mongodb,Java,Spring,Mongodb,我有以下型号： import org.springframework.data.annotation.Id; @Document public class Customer { @Id public String id; public String firstName; public String lastName; public Customer() {} public Customer(String firstName, Strin

我有以下型号：

import org.springframework.data.annotation.Id;

@Document
public class Customer {

    @Id
    public String id;

    public String firstName;
    public String lastName;


    public Customer() {}

    public Customer(String firstName, String lastName) {
        this.firstName = firstName;
        this.lastName = lastName;
    }

    @Override
    public String toString() {
        return String.format(
                "Customer[id=%s, firstName='%s', lastName='%s']",
                id, firstName, lastName);
    }

}

以及以下存储库：

import java.util.List;

import org.springframework.data.mongodb.repository.MongoRepository;

public interface CustomerRepository extends MongoRepository<Customer, String> {

    public Customer findByFirstName(String firstName);
    public List<Customer> findByLastName(String lastName);

}

当然我有一个

SpringMongoConfig.java

@Configuration
@PropertySource(value={"classpath:application.properties"})
public class SpringMongoConfig extends AbstractMongoConfiguration {

    @Value("${spring.data.mongodb.host}")
    private String host;

    @Value("${spring.data.mongodb.port}")
    private Integer port;

    @Value("${spring.data.mongodb.database}")
    private String database;

    public @Bean
    MongoDbFactory mongoDbFactory() throws Exception {
        return new SimpleMongoDbFactory(new MongoClient(this.host + ":" + this.port), database);
    }

    public @Bean
    MongoTemplate mongoTemplate() throws Exception {
        MappingMongoConverter converter = new MappingMongoConverter(new DefaultDbRefResolver(mongoDbFactory()),
                new MongoMappingContext());
        converter.setMapKeyDotReplacement("\\+");

        MongoTemplate mongoTemplate = new MongoTemplate(mongoDbFactory(), converter);

        return mongoTemplate;

    }

    @Override
    protected String getDatabaseName() {
        return this.database;
    }

    @Override
    public Mongo mongo() throws Exception {
        return new MongoClient(this.host, this.port);
    }
}

在执行单元测试时，如何在

生产

和

测试

数据库之间切换

最简单的方法就是在测试中覆盖所需的属性：

@RunWith(SpringRunner.class)  
@SpringBootTest(properties = {
    "spring.data.mongodb.database=test_db"
})
@ContextConfiguration(classes = {CustomerTest.class})
public class CustomerTest {

@RunWith(SpringRunner.class)  
@SpringBootTest(classes = TestConfig.class)
@ContextConfiguration(classes = {CustomTestConfig.class})
public class CustomerTest {

如果您需要重新定义bean，例如使用内存中的fongo实现，您可以创建

CustomTestConfig

并将其添加到测试中：

@RunWith(SpringRunner.class)  
@SpringBootTest(properties = {
    "spring.data.mongodb.database=test_db"
})
@ContextConfiguration(classes = {CustomerTest.class})
public class CustomerTest {

@RunWith(SpringRunner.class)  
@SpringBootTest(classes = TestConfig.class)
@ContextConfiguration(classes = {CustomTestConfig.class})
public class CustomerTest {

最简单的方法是在测试中覆盖所需的属性：

@RunWith(SpringRunner.class)  
@SpringBootTest(properties = {
    "spring.data.mongodb.database=test_db"
})
@ContextConfiguration(classes = {CustomerTest.class})
public class CustomerTest {

@RunWith(SpringRunner.class)  
@SpringBootTest(classes = TestConfig.class)
@ContextConfiguration(classes = {CustomTestConfig.class})
public class CustomerTest {

如果您需要重新定义bean，例如使用内存中的fongo实现，您可以创建

CustomTestConfig

并将其添加到测试中：

@RunWith(SpringRunner.class)  
@SpringBootTest(properties = {
    "spring.data.mongodb.database=test_db"
})
@ContextConfiguration(classes = {CustomerTest.class})
public class CustomerTest {

@RunWith(SpringRunner.class)  
@SpringBootTest(classes = TestConfig.class)
@ContextConfiguration(classes = {CustomTestConfig.class})
public class CustomerTest {

就个人而言，我更喜欢使用嵌入式数据库进行测试。你可以用

在测试班：

package com.backend.repository;

import com.backend.configuration.MongoConfiguration;
import org.junit.runner.RunWith;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes = MongoConfiguration.class)
public class CustomerRepositoryTest {

}

对于开发和生产等其他配置文件，您可以使用spring引导配置文件（请参阅），也可以使用maven配置文件：在

application.properties

中添加此属性

spring.data.mongodb.host=@mongo.host@

然后根据使用的配置文件在

pom.xml

中定义它：

    <profile>
        <id>dev</id>
        <activation>
            <activeByDefault>true</activeByDefault>
        </activation>
        <properties>
            <mongo.host>localhost</mongo.host>
        </properties>
    </profile>
    <profile>
        <id>prod</id>
        <properties>
            <mongo.host>mongo</mongo.host>
        </properties>
    </profile>


发展
真的
本地服务器
戳
蒙戈