如何使用mockito为同一个类的私有方法建立返回值存根

Question

我正在开发基于spring的项目，并使用JUnit+Mockito编写单元测试用例。当将布尔值存根到同一个测试类的私有方法时，我面临一个问题（在将访问级别更改为public后，我仍然未能存根布尔值）。

下面的代码片段显示了同样问题的模拟

class ABC {

    public String method1(User userObj){
        String result = "";

        if(!isValidUser(userObj.getSessionID())){
            return "InvalidUser";
        } else {
           // execute some logic
        }

        return result;
    }


    private boolean isValidUser(String sessionId) {
        // Here it calls some other class to validate the user 
        if (sessionId == null || UserSessionPool.getInstance().getSessionUser(sessionId) == null) {
            return false;
        } else {
            return true;
        }
    } 
}

在这里，我想为method1（）编写一个测试用例。在类ABC中，我有一个名为IsValidUser()的方法，它通过查看一个全局会话池来帮助识别会话中的用户，该会话池保存所有登录的使用过的详细信息，即usersessionpool.getInstance（）.getsessionUser(sessionId)。

在测试method1()时，当测试控制器触发ISValidUser(userobj.getsessionID())时，我希望从ISValidUser()方法返回true，以便继续测试实现逻辑的其余部分。

PowerMockito.doNothing().when(spyed_ABC_ClassObject, "isValidUser", true);

PowerMockito.doReturn(true).when(cntrl, "isValidUser", Mockito.anyString());

Whitebox.invokeMethod(spyed_ABC_ClassObject, "isValidUser", Mockito.anyString());

when(spyed_ABC_ClassObject.isValidUser(Mockito.anyString())).thenReturn(true);

或

Mockito.doNothing().when(spyed_ABC_ClassObject).isValidUser(Mockito.anyString());

Answer 1

你能试试这个吗。

@Before
public void setUp() {
    UserSessionPool mockConnectionPool = Mockito.mock(UserSessionPool.class);
}

@Test
public void testName() throws Exception {
    //given
    PowerMockito.mockStatic(UserSessionPool.class);
    BDDMockito.given(UserSessionPool.getInstance()(...)).willReturn(mockConnectionPool);
    Mockito.when(mockConnectionPool.getSessionUser(Mockito.anylong())).thenReturn(something);


    //then
    PowerMockito.verifyStatic();

}

希望这能有所帮助。快乐编码！