使用异步作业和请求范围Bean进行Spring Boot集成测试

时间:2018-07-01 14:04:37

标签: java spring multithreading spring-boot testing

我面临的问题包括三个因素:

  • 使用Maven和JUnit进行Spring Boot集成测试
  • 应用程序上下文中的请求范围Bean
  • 在应用启动时运行的异步作业。

仅在测试中会出现问题,在不进行测试的情况下应用程序会按预期运行。

简化的应用程序流程:

  • 在Spring Boot启动时,一个 async 作业会同时从“数据源”中获取数据并将其存储在缓存中(Guava CacheLoader)
  • 当用户请求此数据时,拦截请求,对标头令牌进行身份验证,将用户信息存储在请求范围 bean中,然后继续。
  • 从缓存中获取数据并返回给用户。

尝试运行Maven测试时出现错误:

  

org.springframework.beans.factory.BeanCreationException:创建名称为'scopedTarget.requestBean'的bean时出错:作用域'request'对于当前线程无效;如果您打算从单例中引用它,请考虑为此bean定义作用域代理。嵌套异常为java.lang.IllegalStateException:未找到线程绑定的请求:您是在实际Web请求之外引用请求属性,还是在原始接收线程之外处理请求?如果您实际上是在Web请求中操作并且仍然收到此消息,则您的代码可能在DispatcherServlet / DispatcherPortlet之外运行:在这种情况下,请使用RequestContextListener或RequestContextFilter公开当前请求。

解决此问题的重要说明和线索:当控制器直接从DAO(数据源)获取数据时,测试通过!!!只有当它从Guava缓存中获取时,它才会失败。

简化代码:

/* Part of the controller: */
 @Autowired AsyncCacheService cacheService;
 @RequestMapping("/resellers")
 public HashMap<String, Reseller> getAllResellers() throws Exception {
   return cacheService.getAllResellers();
   //When I switch to get directly from DAO below, tests pass.
   //return partnerDao.getAllResellers(); <-- get directly from DAO.
 }




/* The service which the controller calls */
@Service
public class AsyncCacheService {
  private LoadingCache<String, List<Reseller>> resellersCache;
  public AsyncCacheService() {

    resellersCache = CacheBuilder.newBuilder().build(new CacheLoader<String, 
    HashMap<String, Reseller>>() {
      public HashMap<String, Reseller> load(String key) throws Exception {
        return partnerDao.getAllResellers();
      }
    });
  }
  @PostConstruct
  private void refreshCache() {
    /* Refresh cache concurrently */    
    Executors.newSingleThreadScheduledExecutor().scheduleWithFixedDelay(
                () -> resellersCache.refresh(this.getClass().toString()), 0, 
                cacheRefreshTimeInterval, TimeUnit.SECONDS);
  }
  /* Return resellers from cache */
  public HashMap<String, Reseller> getAllResellers() {
    return resellersCache.getUnchecked(this.getClass().toString());
  }
}

拦截器代码简单明了:

@Component
public class AuthInterceptor extends HandlerInterceptorAdapter {
  private @Autowired RequestBean requestBean;
  @Override
  public boolean preHandle(HttpServletRequest request, HttpServletResponse 
       response, Object handler) throws Exception {
    //Verify request and requestBean.set(email, ip, foo)
  }
}

我们如何实例化请求bean:

@Bean
@RequestScope
public RequestBean requestBean() {
    return new RequestBean();
}

最后,测试:

@RunWith(SpringRunner.class)
@SpringBootTest
@AutoConfigureMockMvc
tester {
  @Test
  public void allResellersSizeTest() throws Exception {

    //MvcResult r = 

      mockMvc.perform(get("/api/resellers").header(authHeaderName, jwtToken))
            .andExpect(status().isOk());//.andReturn();
  }
}

1 个答案:

答案 0 :(得分:0)

我找不到适当的修复程序,所以我最终在服务器启动后延迟了几秒钟的异步作业:

 @PostConstruct
  private void refreshCache() {
    /* Refresh cache concurrently */    
    Executors.newSingleThreadScheduledExecutor().scheduleWithFixedDelay(
                () -> resellersCache.refresh(this.getClass().toString()), 33,
                cacheRefreshTimeInterval, TimeUnit.SECONDS);
  }