如何使用HashMap模拟实现MongoDB Repository的Mock仓库？

嘿，这个思路真的很聪明——用HashMap模拟MongoDB Repository，把数据库当成黑盒来验证最终结果，确实比逐个Mock每个数据库调用要高效得多！我来分享几个实用的实现方案，你可以根据自己的项目情况调整：

本质就是实现你的MongoDB Repository接口，内部用HashMap作为内存存储，把MongoDB的CRUD操作映射成HashMap的对应方法。这样业务代码调用Repository时，其实是在操作内存里的HashMap，测试最后直接检查HashMap里的数据就能验证结果，完全不用关心中间的调用细节。

假设你有一个简单的用户实体和对应的Repository接口：

// 实体类
@Document
public class User {
    @Id
    private String id;
    private String name;
    private int age;

    // 构造器、getter、setter省略
}

// Repository接口
public interface UserRepository extends MongoRepository<User, String> {
    List<User> findByName(String name);
    Optional<User> findByAge(int age);
}

接下来创建Mock版本的Repository：

// 可以标记为@Repository让Spring注入，或者测试时手动实例化
@Repository
public class MockUserRepository implements UserRepository {

    // 用HashMap存储数据，key对应MongoDB的ID
    private final Map<String, User> dataStore = new HashMap<>();

    @Override
    public <S extends User> S save(S entity) {
        // 模拟MongoDB自动生成ID的逻辑
        if (entity.getId() == null) {
            entity.setId(new ObjectId().toHexString());
        }
        dataStore.put(entity.getId(), entity);
        return entity;
    }

    @Override
    public Optional<User> findById(String id) {
        return Optional.ofNullable(dataStore.get(id));
    }

    @Override
    public List<User> findAll() {
        return new ArrayList<>(dataStore.values());
    }

    @Override
    public void deleteById(String id) {
        dataStore.remove(id);
    }

    // 实现自定义查询方法
    @Override
    public List<User> findByName(String name) {
        return dataStore.values().stream()
                .filter(user -> name.equals(user.getName()))
                .collect(Collectors.toList());
    }

    @Override
    public Optional<User> findByAge(int age) {
        return dataStore.values().stream()
                .filter(user -> user.getAge() == age)
                .findFirst();
    }

    // 实现基础统计方法
    @Override
    public long count() {
        return dataStore.size();
    }

    // 自定义方法：方便测试时直接获取存储的数据，用于验证
    public Map<String, User> getDataStore() {
        return Collections.unmodifiableMap(dataStore);
    }

    // 其他不需要的方法可以抛出UnsupportedOperationException，或者按需实现
    @Override
    public void delete(User entity) {
        throw new UnsupportedOperationException("未实现该方法");
    }
}

在测试类里，注入这个Mock Repository代替真实的MongoDB Repository，执行业务逻辑后直接检查HashMap里的数据：

@SpringBootTest
public class UserServiceTest {

    @Autowired
    private UserService userService; // 你的业务类，依赖UserRepository

    @Autowired
    private MockUserRepository mockUserRepository;

    @Test
    public void testCreateAndQueryUser() {
        // 执行业务逻辑
        User user = new User(null, "Alice", 25);
        userService.createUser(user);

        // 直接验证存储的数据
        Map<String, User> storedData = mockUserRepository.getDataStore();
        assertEquals(1, storedData.size());
        User savedUser = storedData.values().iterator().next();
        assertEquals("Alice", savedUser.getName());
        assertEquals(25, savedUser.getAge());

        // 测试自定义查询
        List<User> foundUsers = userService.findUsersByName("Alice");
        assertEquals(1, foundUsers.size());
    }
}

• 并发支持：如果测试涉及并发场景，可以把HashMap换成ConcurrentHashMap
• 通用基类：可以封装一个通用的BaseMockMongoRepository基类，实现CRUD的通用逻辑，让其他实体的Mock Repository继承，减少重复代码：

public abstract class BaseMockMongoRepository<T, ID> implements MongoRepository<T, ID> {
    protected final Map<ID, T> dataStore = new HashMap<>();

    @Override
    public <S extends T> S save(S entity) {
        ID id = getId(entity);
        if (id == null) {
            id = generateId();
            setId(entity, id);
        }
        dataStore.put(id, entity);
        return entity;
    }

    // 抽象方法，子类实现ID的获取、设置和生成逻辑
    protected abstract ID getId(T entity);
    protected abstract void setId(T entity, ID id);
    protected abstract ID generateId();

    // 通用方法实现
    @Override
    public Optional<T> findById(ID id) {
        return Optional.ofNullable(dataStore.get(id));
    }

    @Override
    public List<T> findAll() {
        return new ArrayList<>(dataStore.values());
    }

    // 其他通用方法按需实现...
}

然后你的MockUserRepository就可以简化很多：

@Repository
public class MockUserRepository extends BaseMockMongoRepository<User, String> {

    @Override
    protected String getId(User entity) {
        return entity.getId();
    }

    @Override
    protected void setId(User entity, String id) {
        entity.setId(id);
    }

    @Override
    protected String generateId() {
        return new ObjectId().toHexString();
    }

    // 只需要实现自定义查询方法
    @Override
    public List<User> findByName(String name) {
        return dataStore.values().stream()
                .filter(user -> name.equals(user.getName()))
                .collect(Collectors.toList());
    }
}

• 模拟高级特性：如果需要测试分页、排序等MongoDB特性，可以在Mock类里实现对应的逻辑，比如对HashMap的值进行分页截取。

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