mock模拟insert请求 junit4mockito.java.ft

Question

各位大师，有个问题，在写单元测试时发现业务逻辑中.有一段是这种写法
if(item.getId()==null){
mapper.insert(item);
String newId=item.getId();
if(newId!=null){
……
}
}
这种id来自于自增处理，我在模拟时如何能补充这个id，让他能够继续执行后续逻辑呢

Answer 1

在使用 Mockito 进行单元测试时，如果需要模拟自增 ID 的行为，你可以使用 thenAnswer 方法，通过一个自定义的 Answer 对象来处理。

假设你有一个 ItemMapper 接口和对应的 ItemService 服务，可以按照以下方式来模拟自增 ID 的行为：

import static org.mockito.Mockito.*;

public class ItemServiceTest {

    @Test
    public void testInsertWithGeneratedId() {
        // 创建一个 mock 的 ItemMapper
        ItemMapper itemMapper = mock(ItemMapper.class);

        // 创建一个 ItemService，并将 mock 的 ItemMapper 注入
        ItemService itemService = new ItemService(itemMapper);

        // 创建一个 Item 对象，这里不设置 ID
        Item item = new Item();
        item.setName("Example Item");

        // 模拟插入操作，并设置 Answer 来模拟生成自增 ID
        when(itemMapper.insert(any(Item.class))).thenAnswer(new Answer<Integer>() {
            private int idCounter = 1;

            @Override
            public Integer answer(InvocationOnMock invocation) throws Throwable {
                Item insertedItem = invocation.getArgument(0);
                insertedItem.setId(idCounter++); // 模拟生成自增 ID
                return 1; // 假设插入成功
            }
        });

        // 调用业务逻辑
        itemService.insertItem(item);

        // 验证插入是否被调用
        verify(itemMapper, times(1)).insert(any(Item.class));

        // 验证后续逻辑，可以根据实际情况进行具体验证
        // ...

        // 可以通过 item.getId() 获取模拟生成的自增 ID
        assertNotNull(item.getId());
        System.out.println("Generated ID: " + item.getId());
    }
}

在这个例子中，我们使用 thenAnswer 方法来设置 ItemMapper.insert 方法的行为。Answer 接口允许我们编写一段逻辑，来模拟方法调用的返回值。

不过呢。。这只是一个示例，实际情况可能需要根据你的业务逻辑和具体情况进行调整。