系列文章目录

文章目录

系列文章目录
前言

前言

SpringBoot支持集成Mockito做单元测试，我们在本地做单元测试测试的时候，经常因为环境等问题需要mock掉外部方法（远程调用、DB查询等），在Mock掉的同时，如果也想根据入参条件返回mock结果，需要怎样做呢？

一、本文要点

接前文，我们已经已介绍SringBoot如果做单元测试了，本文介绍在两种mock模式下，怎样根据入参返回自定义mock结果。
系列文章完整目录

SpringBooot 单元测试
Mockito 单元测试
MockBean、SpyBean 单元测试
根据入参自定义mock结果

二、开发环境

jdk 1.8
maven 3.6.2
springboot 2.4.3
idea 2020
junit5
Mockito

三、编写真实Service类TagService.java

编写一个方法，无实际意义，模拟真实应用场景。

@Slf4j
@Getter
@Service("tagService")
public class TagService {

	// 这里模拟超级复杂的远程方法调用
    public String buildTagName(String bId) {
        return "";
    }
}

四、编写测试类

1、@MockBean 通常使用when().thenAnswer()模式来测试。

@Slf4j
@ActiveProfiles("dev")
@ExtendWith(SpringExtension.class)
@SpringBootTest
class TagServiceTest {

    /**
     * 这里使用MockBean，tagService中所有方法都会被mock掉.
     */
    @MockBean
    private TagService tagService;
    
    @Test
    void buildTagName() {

        String bid = "45345xclm34kxjL";

        // 这段代码的含义：当执行tagService.buildTagName(bid)方法时，将会被thenAnswer()中的内容替换
        // thenAnswer()中的方法：根据入参，判断是否为正确bId，是的话返回正确的mock结果，否则返回错误的mock结果
        // 这里只是随便举例，可以根据实际情况编写mock逻辑
        Mockito.when(tagService.buildTagName(bid)).thenAnswer(
                invocation -> Optional.of(invocation.getArgument(0))
                        .filter(x -> x.equals(bid)).map(x -> "TAG" + x).orElse("ERROR"));

        // 模拟调用buildTagName()方法
        Assertions.assertEquals(tagService.buildTagName(bid), "TAG" + bid);
        Assertions.assertNotEquals(tagService.buildTagName("error bid"), "TAG" + bid);

    }
}

2、@SpyBean 需要使用doAnswer().when()模式来测试。

@Slf4j
@ActiveProfiles("dev")
@ExtendWith(SpringExtension.class)
@SpringBootTest
class TagServiceTest {

    /**
     * 这里使用MockBean，tagService中所有方法都会被mock掉.
     */
    @MockBean
    private TagService tagService;
    
    @Test
    void buildTagName() {

        String bid = "45345xclm34kxjL";

        // 这段代码的含义：当执行tagService.buildTagName(bid)方法时，将会被thenAnswer()中的内容替换
        // thenAnswer()中的方法：根据入参，判断是否为正确bId，是的话返回正确的mock结果，否则返回错误的mock结果
        // 这里只是随便举例，可以根据实际情况编写mock逻辑
        Mockito.doAnswer(invocation -> Optional.of(invocation.getArgument(0))
                .filter(x -> x.equals(bid)).map(x -> "TAG" + x).orElse("ERROR"))
                .when(tagService).buildTagName(bid);

        // 模拟调用buildTagName()方法
        Assertions.assertEquals(tagService.buildTagName(bid), "TAG" + bid);
        Assertions.assertNotEquals(tagService.buildTagName("error bid"), "TAG" + bid);

    }
}