Java Mockito - 验证使用引用类型参数调用的方法

Question

我是 Mockito 的新手，我正在验证应该使用特定参数调用某些方法，同时可以验证所有值类型参数（int、String、enum 等），但是 reference/class 类型参数好像没有，举个例子

// my actual class
public class MyActualClass {
   public void processRequest() {
       User curUser = MyUtils.getUserFromOtherPlace(UserType.ADMIN);
       processAnotherRequest(1, "address", curUser);
   }
   public void processAnotherRequest(int userId, String address, User user) { ... }
}
public static class MyUtils{
   public static getUserFromOtherPlace(UserType userType) {
       User newUser = new User();
       if (userType == UserType.ADMIN) {
          newUser.setAccess(1);
       }
       //...
       return newUser
   }
}

// my test class
public class MyActualClassTest{
   @Mock
   private MyActualClass myActualClass;

   @Test
   public void testIfMethodBeingCalledCorrectly() {
      User adminUser = new User();
      adminUser.setAccess(1);
      doCallRealMethod().when(myActualClass).processRequest();
      myActualClass.processRequest();
      verify(myActualClass).processAnotherRequest(1, "address", adminUser);
   }
}

我知道这可能是因为我的测试方法中设置的 adminUser 与通过我的实际方法 getUserFromOtherPlace -> MyUtils.getUserFromOtherPlace 生成的参考对象不同，我还尝试模拟return 对象与我的静态方法如

// tried 1
when(MyUtils.getUserFromOtherPlace(UserType.ADMIN).thenReturn(adminUser);  // this throws error like "You cannot use argument matchers outside of verification or stubbing", and suggest using eq()
// tried 2
when(MyUtils.getUserFromOtherPlace(eq(UserType.ADMIN)).thenReturn(adminUser); //this throws NullPointer exception in getUserFromOtherPlace when check the input enum parameter "userType"

那么我如何将引用对象传递到我的入口方法并在此处将其模拟为我的内部方法的 return 值？顺便说一句，如果我的方法只包含值类型参数，它将起作用...

Answer 1

首先，除非不使用 powermock，否则不能模拟静态方法。相反，您可以将用户对象作为参数传递给 processRequest。

public void processRequest(User curUser) {
    processAnotherRequest(1, "address", curUser);
}

public void processAnotherRequest(int userId, String address, User user) {
    //...
}

然后，您可以在测试代码中使用对象引用。

User adminUser = MyUtils.getUserFromOtherPlace(UserType.ADMIN);
adminUser.setAccess(1);
doCallRealMethod().when(myActualClass).processRequest(adminUser);
myActualClass.processRequest(adminUser);
verify(myActualClass).processAnotherRequest(1, "address", adminUser);

Answer 2

因此，为了模拟您的静态方法，您需要：

@RunWith(PowerMockRunner.class)
@PrepareForTest({ MyUtils.class })

//your test class here

@Before
public void setup() {
    PowerMockito.mockStatic(MyUtils.class);
    when(MyUtils.getUserFromOtherPlace(UserType.ADMIN).thenReturn(adminUser);
}

但我总是不喜欢模拟静态类如果还有其他选择，那么也许你可以尝试一个参数捕获器：

//test class

@Captor
private ArgumentCaptor<User> captor;

//your test
@Test
public void testIfMethodBeingCalledCorrectly() {
    doCallRealMethod().when(myActualClass).processRequest();
    myActualClass.processRequest();
    verify(myActualClass, times(1)).processAnotherRequest(1, "address", 
        captor.capture());
    Assert.assertEquals(1, captor.getValue().getAccess());
}

Java Mockito - 验证使用引用类型参数调用的方法

Java Mockito - verify method called with reference type parameter

java

unit-testing

nullpointerexception

mockito

reference-type