当前位置: 移动技术网 > IT编程>开发语言>Java > Spring Boot单元测试中使用mockito框架mock掉整个RedisTemplate的示例

Spring Boot单元测试中使用mockito框架mock掉整个RedisTemplate的示例

2019年07月19日  | 移动技术网IT编程  | 我要评论

概述

当我们使用单元测试来验证应用程序代码时,如果代码中需要访问redis,那么为了保证单元测试不依赖redis,需要将整个redis mock掉。在spring boot中结合mockito很容易做到这一点,如下代码:

import org.mockito.mockito;
import org.springframework.context.annotation.bean;
import org.springframework.context.annotation.configuration;
import org.springframework.data.redis.connection.redisconnection;
import org.springframework.data.redis.connection.redisconnectionfactory;
import org.springframework.data.redis.core.*;
import org.springframework.test.context.activeprofiles;
import static org.mockito.mockito.when;
/**
 * mock掉整个redistemplate
 */
@activeprofiles("uttest")
@configuration
public class redistemplatemocker {
  @bean
  public redistemplate redistemplate() {
    redistemplate redistemplate = mockito.mock(redistemplate.class);
    valueoperations valueoperations = mockito.mock(valueoperations.class);
    setoperations setoperations = mockito.mock(setoperations.class);
    hashoperations hashoperations = redistemplate.opsforhash();
    listoperations listoperations = redistemplate.opsforlist();
    zsetoperations zsetoperations = redistemplate.opsforzset();
    when(redistemplate.opsforset()).thenreturn(setoperations);
    when(redistemplate.opsforvalue()).thenreturn(valueoperations);
    when(redistemplate.opsforhash()).thenreturn(hashoperations);
    when(redistemplate.opsforlist()).thenreturn(listoperations);
    when(redistemplate.opsforzset()).thenreturn(zsetoperations);
    redisoperations redisoperations = mockito.mock(redisoperations.class);
    redisconnection redisconnection = mockito.mock(redisconnection.class);
    redisconnectionfactory redisconnectionfactory = mockito.mock(redisconnectionfactory.class);
    when(redistemplate.getconnectionfactory()).thenreturn(redisconnectionfactory);
    when(valueoperations.getoperations()).thenreturn(redisoperations);
    when(redistemplate.getconnectionfactory().getconnection()).thenreturn(redisconnection);
    return redistemplate;
  }
}

上面的代码已经mock掉大部分的redis操作了,网友想mock掉其他操作,自行加上即可。

总结

以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,谢谢大家对移动技术网的支持。如果你想了解更多相关内容请查看下面相关链接

如对本文有疑问, 点击进行留言回复!!

相关文章:

验证码:
移动技术网