如何使用基于Spring的单元测试模拟返回Servlet上下文的抽象类中可用的静态方法

Question

I want to mock static method which passes UtilityThreadLocal.getServletContext() as a parameter to webApplicationContextUtils.getWebApplicationContext (UtilityThreadLocal.getServletContext())).

I want to use easy mock + powermock. I am probaly tring to create mock from xml and autowring those in my test class but not able to do such. Here is my code which is not working properly. It returning Null pointer exception and sometime Illegal state exception

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration
@PrepareForTest({
    WebApplicationContextUtils.class
})
public class LoginServiceTest {
    @Rule
    public PowerMockRule rule = new PowerMockRule();
    @Autowired
    private LoginService loginService;

    @Test
    public void loginTest() throws Exception {
        String[] rights = new String[] {
        "MANAGE_TENANTS", "MANAGE_USERS", "MANAGE_APPLICATIONS", "MANAGE_SETTINGS", "MANAGE_HISTORY", "MANAGE_OFFICES", "EXPORT_TIMESHEETS", "MANAGE_POLICIES", "MANAGE_ASSETS", "MANAGE_LEAVES"
        };
        Role roleObj = new Role();
        roleObj.setRights(rights);
        WebApplicationContext webAppContextMock = createNiceMock(WebApplicationContext.class);
        RoleService roleServiceBeanMock = createNiceMock(RoleService.class);
        PowerMock.mockStatic(WebApplicationContextUtils.class);
        expect(WebApplicationContextUtils.getWebApplicationContext(UtilityThreadLocal.getServletContext())).andReturn(webAppContextMock);
        expect(webAppContextMock.getBean(RoleService.class)).andReturn(roleServiceBeanMock);
        expect(roleServiceBeanMock.get((long) 2).getRights()).andReturn(rights);
        expect(roleObj.getRights()).andReturn(roleObj.getRights());
        PowerMock.replay(WebApplicationContextUtils.class);
        replay(webAppContextMock);
        replay(roleServiceBeanMock);
    }
}

Answer 1

Mock the call to getWebApplicationContext like this

expect(WebApplicationContextUtils.getWebApplicationContext(
           EasyMock.anyObject(ServletContext.class)))
   .andReturn(webAppContextMock);

This uses non strict matching that is call with any parameter will return specified value. You may probably change setup of other mocks so that they actually match the parameters.