Java中多个条件的单元测试？

Question

我是单元测试的新手，有时会遇到具有多种条件的情况。但是，我不确定是否为每个测试重新模拟或验证相同的情况。

例如，我正在尝试为以下服务方法编写单元测试：

public void create(Request request) {
    
    // code omitted

    if (!employeeService.existsByUuid(uuid)) {
        throw new EntityNotFoundException("Not found");
    }

    EmployeeDTO employee = employeeService.save(...);
    
    if (!departmentService.existsByUuid(employee.getDepartment())) {
        throw new EntityNotFoundException("Not found");
    }
}

我认为我需要为以下场景编写测试：

1. 当employeeService.existsByUuid(uuid) == false，然后抛出新的EntityNotFoundException。然后验证 employeeService.save() 和 departmentService.existsByUuid() 从未被调用。

2. 当 employeeService.existsByUuid(uuid) == true 时，调用 employeeService.save() 并断言这些值。然后验证 employeeService.save() 和 departmentService.existsByUuid() 从未被调用。

3.当departmentService.existsByUuid() == false时抛出新的EntityNotFoundException。在此阶段，我还将 employeeService.existsByUuid(uuid) 模拟为 true，以便测试通过第一个条件。但是，我不确定是否需要断言第二部分；调用 employeeService.save() 并断言这些值。我是否断言返回值或仅验证该方法被调用 1 次。因为我已经断言了它的值，而第三个测试只是针对第三个条件。

当我们有多种条件并且可能需要一次又一次地重新测试相同的条件时，对于这种情况有什么想法吗？

Answer 1

您不应该尝试逐行测试代码，而应该使用涵盖单个有意义场景的案例。因此，如果您已经有一个检查条件的案例，则不必在其他测试案例中重复这些断言。

在您的示例中，我认为这些可能是核心情况：

1. 如果 UUID 不存在，则会引发异常，并且
2. 如果 UUID 存在，则不会保存员工，
3. 如果保存了员工，则所有员工字段都将正确保存，但员工的部门不存在，抛出异常

要测试它们，您可以执行以下操作：

EmployeeService employeeService = mock(EmployeeService.class);

case 1:

when(employeeService.existsByUuid(employeeUuid)).thenReturn(false);   
try {
    testObject.create(request);
    fail();
}
catch(EntityNotFoundException e) {
    verify(employeeService, never()).save(...);
}

case 2:

when(employeeService.existsByUuid(employeeUuid)).thenReturn(true); 
when(employeeService.existsByUuid(departmentUuid)).thenReturn(true);    
testObject.create(request);
verify(employeeService).save(field1, field2, ...);

case 3:

when(employeeService.existsByUuid(employeeUuid)).thenReturn(true);   
when(employeeService.existsByUuid(departmentUuid)).thenReturn(false);   
try {
    testObject.create(request);
    fail();
}
catch(EntityNotFoundException e) {
    // success
}

顺便说一句，您还可以在 @Test 注释中指示预期的异常，但随后您无法对结果：

@Test(expected = EntityNotFoundException.class)
public void test3() {
    when(employeeService.existsByUuid(employeeUuid)).thenReturn(true);   
    when(employeeService.existsByUuid(departmentUuid)).thenReturn(false);   
    testObject.create(request);
}

Answer 2

您可以使用mockito verify和assert throws来测试您的目标，如下所示

    @Test
    public void testOne(){
        when(employeeService.existsByUuid(uuid)).thenReturn(false);

        assertThrows(EntityNotFoundException.class, () -> {
            create(request);
        });

        verify(employeeService, times(0)).save(eq(empObj));
        verify(departmentService, times(0)).existsByUuid(eq(departmentObj));
    }




    @Test
    public void testTwo(){
        when(employeeService.existsByUuid(uuid)).thenReturn(true);
         when(departmentService.existsByUuid(uuid)).thenReturn(true);

        create(request);

        verify(employeeService, times(1)).save(eq(empObj));
        verify(departmentService, times(1)).existsByUuid(eq(departmentObj));
    }

    @Test
    public void testThree(){
        when(employeeService.existsByUuid(uuid)).thenReturn(true);
        when(departmentService.existsByUuid(uuid)).thenReturn(false);

        assertThrows(EntityNotFoundException.class, () -> {
            create(request);
        });

        verify(employeeService, times(1)).save(eq(empObj));
        verify(departmentService, times(1)).existsByUuid(eq(departmentObj));
    }