如何使用Mockito通过注释注入模拟集合

时间:2020-02-11 13:31:22

标签: java mockito junit4

我已经创建了一个带有两个参数的参数化类。一种是字符串类型,另一种是Abstract类的List类型。类构造函数看起来像下面的代码。

public TestService(Tenant tenant, List<AbstractService> testServices) {
        testServicesMap = testServices.stream().collect(Collectors.toMap(AbstractService::getType, Function.identity()));
}

现在,我想为此类编写Junit测试用例,为此,我有以下代码。

    @Mock
    protected Tenant tenant;

    @Mock
    private List<AbstractService> testServices;

    @InjectMocks
    private TestService testService;

    @Before
    public void setup() {
        testServices.add(new JobService(new JobEventService()));
        testServices.add(new ApplicationService(new ApplicationEventService()));
        testServices.add(new UserService(new UserEventService()));
//      notificationService = new NotificationService(tenant, notificationServices);
//      MockitoAnnotations.initMocks(notificationService);
    }

我也尝试启用两个注释行,但是现在可以正常工作了。以下是系统引发的错误。

org.mockito.exceptions.base.MockitoException: 
Cannot instantiate @InjectMocks field named 'notificationService' of type 'class com.test.TestService'.
You haven't provided the instance at field declaration so I tried to construct the instance.   
However the constructor or the initialization block threw an exception : `null`.    

有人可以帮忙吗?

2 个答案:

答案 0 :(得分:1)

您正在将模拟与真实对象混合在一起,因为您创建了一个列表模拟,但随后在该列表上调用add方法,然后期望stream()照常工作。

Mockito模拟默认情况下不执行任何操作,因此您必须告诉它:

Mockito.when(testServices.stream())
       .thenReturn(Stream.of(new JobService(new JobEventService())));

或者更好的情况是从@Mock中删除testServices并为其分配一个新的ArrayList

答案 1 :(得分:0)

问题是您尝试模拟列表,然后调用list.stream(),在模拟默认情况下,此方法返回null。

重复问题的常见解决方案是使用列表的@Spy。