2016-07-06 112 views
0

我想在我的项目中测试控制器。我有以下类(不是重要的代码被跳过):模拟春季Junit测试中注入/设置对象

首页控制器:

@Autowired 
private SessionPreferences sessionPreferences; 

//controller methods 

private Pager<EventSuggestedAndPopular> discoverSuggested() { 
    return SecurityUtils.isUserLogged() 
       ? eventSuggestedService.getSuggestedPopularEventsV2(sessionPreferences.getUser().getUserId(), 10, new Date(), 1, 5) 
       : eventSuggestedService.getSuggestedPopularEventsV2(321, 10, new Date(), 1, 5); 
} 

和测试类:

@Autowired 
    WebApplicationContext wac; 
    @Autowired 
    MockHttpSession session; 
    @Autowired 
    MockHttpServletRequest request; 

    @InjectMocks 
    private HomeController controller; 

    @Mock 
    private SessionPreferences sessionPreferences; 

    @Autowired 
    private UserService userService; 

    private MockMvc mockMvc; 

    @Before 
    public void setup() { 
     MockitoAnnotations.initMocks(this); 
     this.mockMvc = MockMvcBuilders.webAppContextSetup(this.wac).build(); 
    } 

    @Test 
    public void testIndexWithLoggedUser() throws Exception { 
     this.mockMvc.perform(get("/login").header("Referer", "")); 
     this.mockMvc.perform(post("/login").param("email", "[email protected]").param("password", "Test1!").param("rememberMe", "true")) 
       .andExpect(status().isFound()).andExpect(redirectedUrl("/home")); 
     UserSummary user = userService.findByEmail("[email protected]"); 
     this.sessionPreferences.setUser(user); 
//  when(sessionPreferences.getUser()).thenReturn(user); //I also used this, but it didn't work. 
     this.mockMvc.perform(get("/").param("lat", "0.0").param("lon", "0.0").accept(MediaType.TEXT_HTML)).andExpect(status().isOk()) 
       .andExpect(view().name("home")); 
     this.mockMvc.perform(get("/").accept(MediaType.TEXT_HTML)).andExpect(status().isOk()).andExpect(view().name("home")); 
     this.mockMvc.perform(get("/home/").accept(MediaType.TEXT_HTML)).andExpect(status().isOk()).andExpect(view().name("home")); 
    } 

控制器Web应用程序的正常执行过程中效果很好。我使用User Interceptor在会话首选项类中设置用户,该类具有作用域会话并且对所有视图都可见。 当我写单元测试它给我的空指针,当它试图访问用户ID,因为用户显然没有在sessionPreferences中设置。

问题是,如何在Junit中设置autowired类中的对象?我看到我的用户拦截器在Junit测试期间不工作。 我希望能够做这样的事在测试类:

this.sessionPreferences.setUser(user); 

为了调用这个方法没有空指针:

eventSuggestedService.getSuggestedPopularEventsV2(sessionPreferences.getUser().getUserId(), 10, new Date(), 1, 5) 

感谢您的帮助。

回答

1

sessoinPreferences是一个模拟的,所以你可以添加以下到您的setup或测试方法:

User user=new User(); 
user.setUserId(...);  
when(sessionPreferences.getUser()).thenReturn(user);