java - 如何在反应式 mongo 驱动程序中模拟 FindPublisher

Question

我正在使用 Java 中的 mongo 反应驱动程序和反应流库编写应用程序。

我有以下 DAO 代码：

@Override
public Flux<ContentVersion> findBySearch(String appKey, ContentVersionSearchRequest request, Pager pager) {
    final var searchResultsPublisher = mongoClient.getDatabase(appKey)
            .getCollection(COLLECTION_CONTENT_VERSION, ContentVersion.class)
            .find(prepareSearchFilter(request))
            .sort(orderBy(ascending(FIELD_VERSION_STATUS_ORDER), descending(FIELD_UPDATE_DATE)))
            .skip(pager.getSkip())
            .limit(pager.getMax());
    return Flux.from(searchResultsPublisher);
}

在junit测试中，我模拟了MongoClient、MongoDatabase、MongoCollection，但最后MongoCollection返回了一个FindPublisher，我不知道如何正确模拟它。

我已经通过模拟订阅方法成功编写了一个单元测试，如下所示。然而，这对我来说似乎不正确。

@Mock
private MongoClient mongoClient;

@Mock
private MongoDatabase database;

@Mock
private MongoCollection<ContentVersion> collection;

@Mock
private FindPublisher<ContentVersion> findPublisher;

@Mock
private UpdateResult updateResult;

@InjectMocks
private ContentVersionDaoImpl contentVersionDao;

@BeforeEach
void initCommonMocks() {
    when(mongoClient.getDatabase("ddpApp")).thenReturn(database);
    when(database.getCollection(MongoConstants.COLLECTION_CONTENT_VERSION, ContentVersion.class)).thenReturn(collection);
    when(collection.find(any(Bson.class))).thenReturn(findPublisher);
    when(collection.find(any(Document.class))).thenReturn(findPublisher);
    when(findPublisher.limit(anyInt())).thenReturn(findPublisher);
    when(findPublisher.skip(anyInt())).thenReturn(findPublisher);
    when(findPublisher.sort(any())).thenReturn(findPublisher);
}

@Test
void shouldFindBySearch() {
    final var contentVersion1 = new ContentVersion();
    final var contentVersion2 = new ContentVersion();

    final var testPublisher = TestPublisher.<ContentVersion>createCold()
            .emit(contentVersion1, contentVersion2);

    doAnswer(invocation -> {
        testPublisher.subscribe(invocation.getArgument(0, Subscriber.class));
        return null;
    }).when(findPublisher).subscribe(any());

    final var searchFlux = contentVersionDao
            .findBySearch("ddpApp", new ContentVersionSearchRequest(null, null, null), new Pager(1, 10));

    StepVerifier.create(searchFlux)
            .expectNext(contentVersion1)
            .expectNext(contentVersion2)
            .expectComplete()
            .verify();
}

有人知道编写 junit 测试以测试从 mongodb 获取多个文档的优雅方式吗？

标签： javamongodbunit-testingreactive-streams