适配Azure Cosmos DB的内存数据库推荐及Spring Boot集成测试问题

针对你的两个问题，我来逐一解答：

如果你需要内存中运行的、和Cosmos DB兼容的数据库/模拟工具，主要有这几个实用选项：

• Azure Cosmos DB Emulator：这是微软官方提供的本地模拟工具，完全兼容Cosmos DB SQL API（以及其他API），运行在本地内存中，支持所有Cosmos DB的核心特性（比如一致性级别、分区、索引等）。它的行为和云端Cosmos DB几乎一致，是开发和集成测试的首选方案。
• LocalStack：一个开源的云服务模拟工具，支持模拟包括Azure Cosmos DB在内的多种云服务，运行在内存中。适合需要同时模拟多个云服务的集成测试场景，能覆盖Cosmos DB的大部分核心功能。
• 第三方Mock库：比如cosmosdb-mock这类专门的Mock库，通过内存中模拟Cosmos DB的客户端行为来实现，适合快速编写单元测试，不需要启动任何服务进程。

首先得明确：H2是关系型内存数据库，而Cosmos DB是NoSQL文档数据库，两者的数据模型、查询语法（Cosmos DB的SQL API是类SQL但并非标准SQL）、客户端API完全不兼容，直接替换成H2肯定无法正常工作。给你几个可行的解决思路：

1. 使用官方Cosmos DB Emulator（最推荐） #

这是最贴近生产环境的测试方案，完全兼容所有Cosmos DB特性：

• 下载安装Azure Cosmos DB Emulator，它会在本地启动一个内存中的Cosmos DB实例
• 在测试配置文件application-test.yml中配置连接信息指向模拟器：

azure:
  cosmos:
    uri: https://localhost:8081
    key: C2y6yDjf5/R+ob0N8A7Cgv30VRDJIWEHLM+4QDU5DE2nQ9nDuVTqobD4b8mGGyPMbIZnqyMsEcaGQy67XIw/Jw==
    database: test-db
    consistency-level: Session

• 在测试类上添加@ActiveProfiles("test")，确保加载测试配置。

2. 模拟Cosmos DB客户端（适合单元测试） #

如果不想启动模拟器，可以用Mockito模拟CosmosTemplate或者自定义的Cosmos Repository：

@SpringBootTest
public class UserRepositoryTest {
    @MockBean
    private CosmosTemplate cosmosTemplate;

    @Autowired
    private UserRepository userRepository;

    @Test
    public void testSaveUser() {
        User testUser = new User("user1", "John Doe");
        // 模拟save方法的返回值
        when(cosmosTemplate.save(any(User.class))).thenReturn(testUser);
        
        User savedUser = userRepository.save(testUser);
        assertEquals("John Doe", savedUser.getName());
    }
}

这种方式不需要依赖任何数据库，纯内存模拟，适合快速验证业务逻辑。

3. 使用Spring Boot的Cosmos测试切片 #

Spring Boot提供了@CosmosTest测试切片，可以只加载Cosmos DB相关的Bean，减少测试上下文的加载时间：

@CosmosTest
@ActiveProfiles("test")
public class UserRepositoryTest {
    @Autowired
    private UserRepository userRepository;

    @Test
    public void testFindUserById() {
        // 如果搭配模拟器，这里可以直接执行真实的CRUD操作
        User testUser = userRepository.save(new User("user2", "Jane Doe"));
        Optional<User> foundUser = userRepository.findById("user2");
        
        assertTrue(foundUser.isPresent());
        assertEquals("Jane Doe", foundUser.get().getName());
    }
}

4. 排除H2相关的自动配置 #

如果你之前的配置中不小心引入了H2的依赖，记得在测试类中排除H2的自动配置，避免Spring混淆数据源：

@SpringBootTest(exclude = {DataSourceAutoConfiguration.class, H2ConsoleAutoConfiguration.class})
@ActiveProfiles("test")
public class IntegrationTest {
    // 测试逻辑
}

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