java - 我如何使用 Mockito 进行模拟？

Question

我正在进行单元测试，我无法对 Mockito 进行编程以覆盖部分代码。
如何让 Mockito 返回有效的东西？当我得到 spec 时，我得到一个 IllegalArgumentExpection 代码。很抱歉，如果这是一个无知的问题，我最近开始编写测试。

我的测试

@Bean
        public SpecDBDAO getSpecDBDAO() {
            SpecDBDAO dao = Mockito.mock(SpecDBDAO.class);
            when(dao.findLastOne(new BasicDBObject("_id", "erro"))).thenReturn(new BasicDBObject());
            return dao;
        }

 @Test
    public void testAddLinha_validId() throws Exception {
        planilhaService.addLinha("123", new BasicDBObject("_id", "erro"));
    }

我的代码

public Planilha addLinha(String id, BasicDBObject body) {
        String idSpec = body.getString("_id", "");
        Planilha planilha = specDBPlanilhasDAO.get(id);
        if (planilha == null) {
            throw new NotFoundException("Planilha não encontrada.");
        }

        try {
            BasicDBObject spec = specDBDAO.findLastOne(new BasicDBObject("_id", new ObjectId(idSpec)));
            if (spec.isEmpty()) {
                throw new NotFoundException("Especificação não encontrada.");
            }
            planilha.addLinha(spec);
            planilha = specDBPlanilhasDAO.update(planilha);

            return planilha;
        } catch (IllegalArgumentException e) {
            throw new BadRequestException("Id inválido.");
        }
    }

覆盖范围

Answer 1

您在此处使用的 BasicDBObject 实例

BasicDBObject spec = specDBDAO.findLastOne(new BasicDBObject("_id", new ObjectId(idSpec)));

与您在此处使用的 BasicDBObject 实例不同

when(dao.findLastOne(new BasicDBObject("_id", "erro"))).thenReturn(new BasicDBObject());

解决方案

1 在 BasicDBObject 上重写 equals() 和 hashCode() 因此相等性基于例如 id 和 errno 以及任何其他必要的值。

2 例如，在设置期望时使用 org.mockito.Matchers 类为您提供匹配器

when(dao.findLastOne(Matchers.any(BasicDBObject.class).thenReturn(new BasicDBObject()); 
// any BasicDBObject instance will trigger the expectation

或

when(dao.findLastOne(Matchers.eq(new BasicDBObject("_id", "erro")).thenReturn(new BasicDBObject()); 
// any BasicDBObject equal to the used here instance will trigger the expectation.  Equality is given by your overridden equals method

您可以找到有关此 here 的更多信息.