欢迎来到代码驿站!

JAVA代码

当前位置:首页 > 软件编程 > JAVA代码

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

时间:2020-12-24 12:08:58|栏目:JAVA代码|点击:

概述

当我们使用单元测试来验证应用程序代码时,如果代码中需要访问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掉其他操作,自行加上即可。

总结

上一篇:Java基本数据类型与封装类型详解(int和Integer区别)

栏    目:JAVA代码

下一篇:java实现的汉字转五笔功能实例

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

本文地址:http://www.codeinn.net/misctech/36171.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有