java - 将 Spring Data 随机(嵌入式)Mongo 端口与 NoSQL JUnit @Rule 一起使用

Question

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

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

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

现在问题来了，我希望能够在我的每个测试方法运行之前从某个外部文件中注入(inject)一些测试数据。我发现 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() {
   // ...
}

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

所以我的问题是，有没有人曾经使用过 NoSQL Unit 的嵌入式 Mongo 功能，有没有办法，例如创建 @Rule after Spring 上下文已初始化？

我想知道是否自己找到空闲端口(以静态方式)，将其设置为 @Rule，然后告诉 EmbeddedMongoAutoConfiguration 通过覆盖IMongodConfig bean 是个好主意吗？还是有一种“更简单”的方法？

注意:我刚刚看到 flapdoodle 库提供了一个类和一个静态方法来查找空闲的服务器端口，Spring 像这样使用它:

Network.getFreeServerPort(getHost()), Network.localhostIsIPv6()))

先谢谢大家了!

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

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());

然后在关联的配置类中:

@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);
    }
}

Answer 1

细化@user6599111给出的解决方案，可以获取Flapdoodle Embedded Mongo随机选择的端口，只需注入(inject)IMongodConfig类型的对象即可。

Spring Boot 会自动为您构建此对象，如 here 所述.

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

@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);
    }
}