使用带有 NoSQL JUnit @Rule 的 Spring Data 随机(嵌入式)Mongo 端口

Leb*_*ski 3 java mongodb property-injection spring-data junit-rule

我目前正在尝试编写一个使用 Spring Data Mongo 存储库的集成测试类。我使用de.flapdoodle.embed.mongo依赖项提供的嵌入式 Mongo 实例。Spring Data 文档指定我们只需将此依赖项放在项目EmbedMongoAutoConfiguration中,其余部分由它来处理。

现在,没关系,将端口设置为0使自动配置过程找到一个空闲端口来启动 mongo 实例。

此功能对我来说是必要的,以避免与其他测试(这些测试与我公司的其他项目一起在 Jenkins CI 服务器上运行)发生冲突。

现在问题来了,我希望能够在我的每个测试方法运行之前从某个外部文件注入一些测试数据。我发现 NoSQL Unit 可以通过一个简单的方法注释和一个 JUnit 来做到这一点@Rule

下面是一个例子:

@Value("${local.mongo.port}")
private int mongoPort; // <- still 0 a the time the Rule below is created.

@Rule
public MongoDbRule managedMongoDb = new MongoDbRule(MongoDbConfigurationBuilder.mongoDb().databaseName("myAwesomeDb").port(mongoPort).build());

@Test
@UsingDataSet(locations = "testdata.json", loadStrategy = LoadStrategyEnum.CLEAN_INSERT)
public void testMyData() {
   // ...
}
Run Code Online (Sandbox Code Playgroud)

我的问题是,@Rule需要在其构建器中使用 Mongo 端口来实例化底层的 MongoClient,但是在实例化 @Rule 时,Spring 上下文尚未完全初始化并且EmbeddedMongoAutoConfiguration尚未发布端口。

所以我的问题是,有没有人曾经在 NoSQL 单元中使用过嵌入式 Mongo 功能,有没有办法,例如在 Spring 上下文初始化@Rule 之后创建?

我想知道自己(以静态方式)找到空闲端口,将其设置为@Rule,然后EmbeddedMongoAutoConfiguration通过覆盖IMongodConfigbean告诉使用它是个好主意吗?还是有“更简单”的方法?

注意:我刚刚看到 fladdoodle 库提供了一个类和一个静态方法来查找一个空闲的服务器端口,它被 Spring 使用,如下所示:

Network.getFreeServerPort(getHost()), Network.localhostIsIPv6()))
Run Code Online (Sandbox Code Playgroud)

提前谢谢大家!

编辑:我尝试了上面提到的解决方案,它似乎有效,但我仍然认为它有点“冗长”和肮脏。

private static final Logger log = LoggerFactory.getLogger(MyAwesomeIT.class);
private static int mongoPort;
static {
    try {
        mongoPort = Network.getFreeServerPort();
    } catch (IOException e) {
        log.error("Error while trying to find a free port for Mongo", e);
        mongoPort = -1; // test should then not work
    }
}

@Rule
public MongoDbRule managedMongoDb = new MongoDbRule(MongoDbConfigurationBuilder.mongoDb().databaseName("myAwesomeDb").port(mongoPort).build());
Run Code Online (Sandbox Code Playgroud)

然后在关联的配置类中:

@Configuration
@EnableAutoConfiguration
@EnableMongoRepositories
@EnableConfigurationProperties(MongoProperties.class)
static class ContextConfiguration {
    @Autowired
    private MongoProperties mongoProperties;

    @PostConstruct
    public void init() {
        // Here, I override the port property
        mongoProperties.setPort(mongoPort);
    }
}
Run Code Online (Sandbox Code Playgroud)

ric*_*din 6

细化@user6599111给出的解决方案,可以得到portFlapdoodle Embedded Mongo随机选择的,只需注入一个类型为 的对象即可IMongodConfig

春天开机自动地建立此对象为你,为说明这里

那么,配置类就会变成下面这样。

@Configuration
@EnableAutoConfiguration(exclude = { EmbeddedMongoAutoConfiguration.class })
public class MongoConfiguration {
    @Autowired
    private Environment environment;

    @Autowired
    private MongoProperties properties;

    @Autowired(required = false)
    private MongoClientOptions options;

    @Autowired
    private IMongodConfig mongoConfig;

    @Bean
    public MongoClient mongo() throws Exception {
        properties.setPort(mongoConfig.net().getPort());
        return properties.createMongoClient(this.options, this.environment);
    }
}
Run Code Online (Sandbox Code Playgroud)