如何为Spring MongoDB文本查询函数搭建单元或集成测试？

刚好有过类似的测试经验，来给你拆解下怎么给这个MongoDB文本搜索函数写单元和集成测试，分两种场景给你具体方案：

单元测试的核心是验证你的业务逻辑（比如查询条件的构建、结果的封装），不需要真的连接MongoDB，用 Mock 框架（比如 Mockito）模拟 MongoTemplate 和相关对象就行。

步骤和示例代码 #

假设你的 search 函数在一个叫 YourMongoService 的类里，依赖 MongoTemplate 获取集合，测试用 JUnit 5 + Mockito 来写：

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.data.mongodb.core.MongoTemplate;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoCursor;
import org.bson.Document;
import java.util.ArrayList;
import java.util.List;
import static org.mockito.Mockito.*;
import static org.assertj.core.api.Assertions.assertThat;

@ExtendWith(MockitoExtension.class)
public class YourMongoServiceTest {

    @Mock
    private MongoTemplate mockMongoTemplate;

    @Mock
    private MongoCollection<Document> mockRandomCollection;

    @Mock
    private MongoCursor<Document> mockCursor;

    private YourMongoService service;

    @Test
    void search_shouldBuildCorrectTextQueryAndReturnResults() {
        // 1. 初始化测试数据
        String testSearchText = "spring mongodb";
        service = new YourMongoService(mockMongoTemplate);
        service.setText(testSearchText); // 假设你有set方法给text变量赋值

        // 2. 准备预期返回的文档
        Document expectedDoc1 = new Document("id", 1).append("content", "Spring Data MongoDB guide");
        Document expectedDoc2 = new Document("id", 2).append("content", "MongoDB with Spring Boot");
        List<Document> expectedResults = List.of(expectedDoc1, expectedDoc2);

        // 3. 模拟MongoTemplate和集合的行为
        when(mockMongoTemplate.getCollection("randomCollection")).thenReturn(mockRandomCollection);
        when(mockRandomCollection.find(any(Document.class))).thenReturn(mockCursor);
        // 模拟游标返回两个结果后结束
        when(mockCursor.hasNext()).thenReturn(true, true, false);
        when(mockCursor.next()).thenReturn(expectedDoc1, expectedDoc2);

        // 4. 执行测试方法
        ArrayList<Document> actualResults = service.search();

        // 5. 验证结果和查询条件
        assertThat(actualResults).containsExactlyElementsOf(expectedResults);
        // 确认构建的$text查询是否正确
        Document expectedQuery = new Document("$text", new Document("$search", testSearchText));
        verify(mockRandomCollection).find(eq(expectedQuery));
    }
}

关键要点 #

• 重点验证查询条件是否正确构建（也就是你拼的 $text 查询文档），以及函数是否正确把查询结果转成 ArrayList<Document>
• 完全脱离真实数据库，测试速度快，适合做单元级别的逻辑验证

如果要验证真实的MongoDB查询行为（比如文本索引是否生效、$text 查询的匹配逻辑是否符合预期），就得用集成测试，推荐用嵌入式MongoDB（不用依赖外部服务，适合CI/CD）。

步骤和示例代码 #

以Spring Boot项目为例，先引入嵌入式Mongo的依赖（Maven）：

<dependency>
    <groupId>de.flapdoodle.embed</groupId>
    <artifactId>de.flapdoodle.embed.mongo</artifactId>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-test</artifactId>
    <scope>test</scope>
</dependency>

然后写集成测试：

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.bson.Document;
import java.util.ArrayList;
import static org.assertj.core.api.Assertions.assertThat;

@SpringBootTest
public class YourMongoServiceIntegrationTest {

    @Autowired
    private MongoTemplate mongoTemplate;

    @Autowired
    private YourMongoService service;

    @BeforeEach
    void setUpTestData() {
        // 每次测试前清空集合，避免数据污染
        mongoTemplate.dropCollection("randomCollection");
        
        // 必须创建文本索引，否则$text查询会报错
        mongoTemplate.getCollection("randomCollection")
                .createIndex(new Document("content", "text").append("title", "text"));

        // 插入测试用的文档
        Document doc1 = new Document("title", "MongoDB 101").append("content", "Learn MongoDB text search");
        Document doc2 = new Document("title", "Spring Data Guide").append("content", "Spring Data MongoDB integration");
        Document doc3 = new Document("title", "Java Basics").append("content", "Core Java programming");
        mongoTemplate.insert(List.of(doc1, doc2, doc3), "randomCollection");
    }

    @Test
    void search_shouldReturnTextMatchingDocuments() {
        // 设置搜索关键词
        service.setText("mongodb");

        // 执行搜索
        ArrayList<Document> results = service.search();

        // 验证结果：应该匹配doc1和doc2
        assertThat(results).hasSize(2);
        assertThat(results).extracting(doc -> doc.getString("title"))
                .containsExactlyInAnyOrder("MongoDB 101", "Spring Data Guide");
    }

    @Test
    void search_shouldReturnEmptyListWhenNoMatches() {
        service.setText("python");

        ArrayList<Document> results = service.search();

        assertThat(results).isEmpty();
    }
}

关键要点 #

• 必须给集合创建文本索引，否则$text查询会抛出异常，这是MongoDB的要求
• 每次测试前清空集合，保证测试的独立性
• 可以测试各种边界场景：比如无匹配结果、多个匹配结果、部分匹配等

额外提示 #

• 如果是非Spring项目，需要手动启动嵌入式Mongo实例，创建MongoClient连接，再初始化MongoTemplate
• 集成测试速度比单元测试慢，但能验证真实的数据库交互逻辑，建议和单元测试配合使用

内容的提问来源于stack exchange，提问作者user1796624