使用 Mockito 模拟枚举？

Question

我需要模拟以下枚举：

public enum PersonStatus
{
    WORKING,
    HOLIDAY,
    SICK      
}

这是因为它在我正在测试的以下类中使用：

被测类：

public interface PersonRepository extends CrudRepository<Person, Integer>
{
    List<Person> findByStatus(PersonStatus personStatus);
}

这是我目前的测试尝试：

当前测试：

public class PersonRepositoryTest {

    private final Logger LOGGER = LoggerFactory.getLogger(PersonRepositoryTest.class);

    //Mock the PersonRepository class
    @Mock
    private PersonRepository PersonRepository;

    @Mock
    private PersonStatus personStatus;

    @Before
    public void setUp() throws Exception {

        MockitoAnnotations.initMocks(this);
        assertThat(PersonRepository, notNullValue());
        assertThat(PersonStatus, notNullValue());
    }

    @Test
    public void testFindByStatus() throws ParseException {

        List<Person> personlist = PersonRepository.findByStatus(personStatus);
        assertThat(personlist, notNullValue());
    }
}

这给出了以下错误：

错误：

org.mockito.exceptions.base.MockitoException: 
Cannot mock/spy class PersonStatus
Mockito cannot mock/spy following:
  - final classes
  - anonymous classes
  - primitive types

我该如何解决这个问题？

Answer 1

您的testFindByStatus试图断言findByStatus不返回null。

如果方法以相同的方式工作，无论personStatus param的值如何，只需传递其中一个：

@Test
public void testFindByStatus() throws ParseException {
    List<Person> personlist = PersonRepository.findByStatus(WORKING);
    assertThat(personlist, notNullValue());
}

如果其他可能值的行为可能不同，您可以测试每个值：

@Test
public void testFindByStatus() throws ParseException {
    for (PersonStatus status : PersonStatus.values()) {
        List<Person> personlist = PersonRepository.findByStatus(status);
        assertThat(personlist, notNullValue());
    }
}

Answer 2

只是为了完成图片：

最新版本的Mockito 2非常支持模拟最终课程。 但是你必须首先明确启用这个新的实验功能！

（请参阅此处了解如何执行此操作 - 归结为将文件mockito-extensions/org.mockito.plugins.MockMaker到类路径中，其中包含值mock-maker-inline ）

但当然：如果必须，你只会嘲笑某事。 你想要模拟Enum实例的愿望很可能是由于不了解 - 或者因为你在这里创建了难以测试的代码。 从这个意义上说，真正的答案是首先考虑避免这种嘲弄的方法。

Answer 3

如前所述，使用 Mockito 2 并启用实验性功能。

实际上缺少的是一个示例片段来演示如何。 考虑到一个名为LicenseHistoryAction的枚举已经存在 4 个值，这将正确模拟一个UNSUPPORTED一个：

try (MockedStatic<LicenseHistoryAction> licenseHistoryActionMockedStatic = Mockito.mockStatic(LicenseHistoryAction.class)) {
    final LicenseHistoryAction UNSUPPORTED = Mockito.mock(LicenseHistoryAction.class);
    Mockito.doReturn(4).when(UNSUPPORTED).ordinal();

    licenseHistoryActionMockedStatic.when(LicenseHistoryAction::values)
            .thenReturn(new LicenseHistoryAction[]{
                    LicenseHistoryAction.ASSIGN,
                    LicenseHistoryAction.RELEASE,
                    LicenseHistoryAction.UNBIND,
                    LicenseHistoryAction.DENY,
                    UNSUPPORTED});
}