如果我们在方法内部调用私有方法，该如何模拟或准备测试方法？

Question

虽然为Controller类编写测试用例，但私有方法是getServiceContext（）。有不同的对象，因为我们从testclass传递serviceContext，并且控制器类内部的其他对象本身调用自身。由于此Foo对象为null。该如何解决。

公共类控制器{

@Refernce
private FooService fooService;

public CustomData getDetails(String id){

      Foo foo = fooService.getFoo(id ,**getServiceContext()**); 
         //getServiceContext() is different object
    System.out.println("foo data>>>> "+foo); // **Throwing null pointer exceptions** 

      CustomData customData = new CustomData();
      customData.setStudentName(foo.getName);
      customData.setStudentName(foo.getId);
      ...
      ...
      ...
    return  customData;

}

private ServiceContext getServiceContext() {

 ServiceContext serviceContext = new ServiceContext();
    serviceContext.setCompanyId(context..);
    serviceContext.setUserId(context..);
    ...
    ....    
retrn serviceContext;

}

}

---

公共类ControllerTest {

@InjectMocks
private Controller controller;

@Mock
private FooService fooService;

private Foo foo;

@BeforeEach
public void setUp() throws PortalException {

foo = mock(Foo.class);

}

@Test
public void getDetailsTest() throws Exception {

    ServiceContext **serviceContext** = new ServiceContext();
    serviceContext.setCompanyId(context..);
    serviceContext.setUserId(context..);
    ...
    ....    
    Mockito.when(fooService.getFoo("testId",serviceContext)).thenReturn(foo);

    System.out.println("Service context>>>> "+**serviceContext**); // different serviceContext object
    CustomData customData = controller.getDetails("testId");

    Assertions.assertThat(ss).isNotNull();
}

}

Answer 1

有多种方法可以做到这一点。

首先，我们可以使用anyOf(Type.class)进行模拟，这实际上将匹配对象类型而不是值。

Mockito
 .when(fooService.getFoo(Mockit.eq("testId"), Mockito.any(ServiceContext.class)))
 .thenReturn(foo);

这将按预期工作，并返回所需的值。

另外，如果要检查在服务方法中以arg形式传递了哪些数据serviceContext对象（（我们只是检查了对象类型而不是值），我们可以使用ArgumentCaptor。

它基本上捕获在方法调用中传递的参数数据。

让我们为服务上下文创建ArgumentCaptor

@Mock
private FooService fooService;

@Captor
private ArgumentCaptor<ServiceContext> captor;

现在，让我们在验证期间捕获参数。

Mockito.verify(fooService).getFoo(Mockit.eq("testId"), captor.capture());

Assertions.assertEquals("value of x in context", captor.getValue().getX());

[基本上在这里，captor.getValue()返回正在传递的服务上下文对象。因此，您可以在该对象中验证所有要验证的数据。

Alternate，方法将是Spy，它将基本上监视被测类，并且您可以控制测试类本身中私有方法的行为。

为此，我们需要在测试类上添加@Spy注释以及@InjectMocks。

@Spy
@InjectMocks
private Controller controller;

现在，您可以模拟私有方法并返回期望值。

Mockito.doReturn(serviceContextValue).when(controller).getServiceContext();

并使用该对象模拟fooService。

Mockito.verify(fooService).getFoo("testId", serviceContextValue);

但是使用Spy时，请不要忘记为私有方法编写单元测试，因为它是被模拟的，因此上述测试用例将不会对它的业务逻辑进行测试。这是一个原因，不建议这样做。

我建议使用ArgumentCaptor方法。