基于基于Springboot+Junit+Mockito做单元测试的示例做单元测试的示例
本篇文章主要介绍了基于Springboot+Junit+Mockito做单元测试的示例，小编觉得挺不错的，现在分享给大家，也给
大家做个参考。一起跟随小编过来看看吧
前言前言
这篇文章介绍如何使用Springboot+Junit+Mockito做单元测试，案例选取撮合交易的一个类来做单元测试。
单元测试前先理解需求单元测试前先理解需求
要写出好的单测，必须先理解了需求，只有知道做什么才能知道怎么测。但本文主要讲mockito的用法，无需关注具体需求。所以本
节略去具体的需求描述。
隔离外部依赖隔离外部依赖
Case1. 被测类中被@Autowired 或 @Resource 注解标注的依赖对象，如何控制其返回值
以被测方法 MatchingServiceImpl.java的matching(MatchingOrder buyOrder, MatchingOrder sellOrder)为例
被测类MatchingServiceImpl
public class MatchingServiceImpl implements MatchingService {
private static final Logger log = LoggerFactory.getLogger(MatchingServiceImpl.class);
@Autowired
private QuoteService quoteService;
...
public MatchingResult matching(MatchingOrder buyOrder, MatchingOrder sellOrder) {
int currentPrice = quoteService.getCurrentPriceByProduct(buyOrder.getProductCode());
MatchingResult result = new MatchingResult();
if (sellOrder != null && buyOrder != null &&
sellOrder.getPrice() <= buyOrder.getPrice()) {
...
}
}
matching方法中的quoteService.getCurrentPriceByProduct(buyOrder.getProductCode());要访问Redis获取当前报价，这里我们需要
把外部依赖quoteService mock掉，控制getCurrentPriceByProduct方法的返回值。使用mockito可以做到，具体如下：
测试类MatchingServiceImplTest
public class MatchingServiceImplTest extends MockitoBasedTest {
/**
* 被@Mock标注的对象会自动注入到被@InjectMocks标注的对象中
*/
@Mock
private QuoteService quoteService;
/**
* <pre>
* 被测对象，用@InjectMocks标注，那些被@mock标注的对象就会自动注入其中。
* 另一个注意点是这里的MatchingServiceImpl是直接new出来(Mockito 1.9版本后不new也可以)，而不是通过spring容器注入的。因为这里我不需要从spring容器中
* 获得其他依赖，不需要database ，redis ，zookeeper，mq，啥都不依赖，所以直接new
* </pre>
*/
@InjectMocks
private MatchingServiceImpl matchingService = new MatchingServiceImpl();
@Test
public void testMatching_SuccessWhenCurrentPriceBetweenBuyPriceAndSellPrice() {
MatchingOrder buyOrder = new MatchingOrder();
buyOrder.setPrice(1000);
buyOrder.setCount(23);
MatchingOrder sellOrder = new MatchingOrder();
sellOrder.setPrice(800);
sellOrder.setCount(20);
// 方法打桩(Method stubbing)
// when(x).thenReturn(y) ：当指定方法被调用时返回指定值
Mockito.when(quoteService.getCurrentPriceByProduct(Mockito.anyString())).thenReturn(900);
MatchingResult result = matchingService.matching(buyOrder, sellOrder);
org.junit.Assert.assertEquals(true, result.isSuccess());// 断言撮合是否成功
org.junit.Assert.assertEquals(20, result.getTradeCount());// 断言成交数量
org.junit.Assert.assertEquals(900, result.getTradePrice()); // 断言最新报价是否符合预期
}
Case2. 被测函数A调用被测类其他函数B，怎么控制函数B的返回值？
比如，MatchingServiceImpl中有个函数startBuyProcess，它里面调用了该类中的其他函数，如getTopSellOrder，matching，如何
控制这两个函数的返回值？
这里要解决的问题其实是怎么对一个类”部分mock”–被测类的被测方法（如startBuyProcess）要真实执行，而另一些方法（如
getTopSellOrder）则是要打桩（不真正进去执行）。
评论1
最新资源