单元测试编写最佳实践(ChatGPT+Mockito+JUnit)

背景

基于 springboot 微服务架构给单元测试带来的问题:

  1. springboot 单元测试启动家长过程非常缓慢,后期服务启动达到分钟级,非常影响效率
  2. 服务之间相互依赖非常严重,单元测试的运行非常依赖其它服务稳定性
  3. 第三方服务和中间件,测试过程产生大量垃圾数据,污染环境,非常笨重,甚至产生资损。

解决办法

采用 EasyMock, PowerMock,Mockito 等mock 框架, 屏蔽外部依赖,还原单元测试本身。

Mockito使用

由于spring-boot-starter-test 默认集成了 Mockito的依赖,本文优先介绍 Mockito框架的使用。

依赖

一般不需要手动指定

        <dependency>
            <groupId>org.mockito</groupId>
            <artifactId>mockito-core</artifactId>
            <version>3.3.3</version>
        </dependency>

导入

导入常用静态方法

import static org.mockito.Mockito.*;
import static org.junit.Assert.*;

模拟对象

Mock 对象的创建

语法: mock(class or interface)

用例:

OrderService orderService = mock(OrderService.class);

设置预期返回值

语法: when(mock.someMethod()).thenReturn(value)

用例:

when(productService.getById(1L)).thenReturn(new Product(1L,"mate50", new BigDecimal("9000")));

验证被测试类方法

语法: verify(mock).someMethod(…),verify(mock,times(integer)).someMethod(…)

用例:

verify(orderMapper).insert(any(Order.class));
verify(stockService, times(2)).deduct(anyLong(), eq(1));

配合注解使用

语法: @Mock, mock 一个对象

语法: @InjectMocks , 依赖注入 mock对象

用例:

    /**
     * orderMapper, stockService等为 OrderServiceImpl的依赖对象.
     */
    @InjectMocks
    private OrderService orderService = new OrderServiceImpl(); 

    @Mock
    private OrderMapper orderMapper;

    @Mock
    private StockService stockService;

    @Mock
    private AmountService amountService;

    @Mock
    private ProductService productService;

完整案列

    @Test
    public void testSubmitOrder() {
        Long userId = 1L;
        List<Product> productList = Lists.newArrayList(
                new Product(1L, 1),
                new Product(2L, 1)
        );

        // 模拟productService.getById方法,返回一个Product对象
        when(productService.getById(1L)).thenReturn(new Product(1L,"mate50", new BigDecimal("9000")));
        when(productService.getById(2L)).thenReturn(new Product(2L,"充电宝", new BigDecimal("120")));

        boolean result = orderService.submitOrder(userId, productList);

        // submitOrder 验证返回结果为true
        assertTrue(result);

        // 验证stockService.deduct方法被调用了2次
        verify(stockService, times(2)).deduct(anyLong(), eq(1));

        // 验证orderMapper.insert方法被调用了1次
        verify(orderMapper, times(1)).insert(any(Order.class));
    }

使用codeGPT 生成单元测试

prompt: Write JUnit Tests for the following code by using java, use mockito, use Chinese comment:

完整代码

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;

import java.math.BigDecimal;
import java.util.Arrays;
import java.util.List;

import static org.junit.Assert.assertEquals;
import static org.mockito.ArgumentMatchers.anyLong;
import static org.mockito.Mockito.*;

@RunWith(MockitoJUnitRunner.class)
public class OrderServiceTest {

    @Mock
    private ProductService productService;

    @Mock
    private StockService stockService;

    @Mock
    private OrderMapper orderMapper;

    @InjectMocks
    private OrderService orderService;

    @Test
    public void testSubmitOrder() {
        // given
        List<Product> products = Arrays.asList(
                new Product(1L, "product1", BigDecimal.valueOf(10), 2),
                new Product(2L, "product2", BigDecimal.valueOf(20), 1)
        );
        when(productService.getById(1L)).thenReturn(new Product(1L, "product1", BigDecimal.valueOf(10), 10));
        when(productService.getById(2L)).thenReturn(new Product(2L, "product2", BigDecimal.valueOf(20), 10));

        // when
        boolean result = orderService.submitOrder(123L, products);

        // then
        assertEquals(true, result);
        verify(productService, times(2)).getById(anyLong());
        verify(stockService, times(3)).deduct(anyLong(), anyInt());
        verify(orderMapper).insert(any(Order.class));
    }
}

结论:通过观察,gpt 生成的单元测试跟手动写的单元测试非常相近,几乎直接可以使用。



一点心得

不要去纠结GPT 哪方面做的不好,要多思考,利用GPT能为我们做什么。文章来源地址https://uudwc.com/A/PmJWx

原文地址:https://blog.csdn.net/wfh45678/article/details/130374290

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请联系站长进行投诉反馈,一经查实,立即删除!

上一篇 2023年09月19日 08:11
Learn Prompt-ChatGPT 精选案例:学习各国语言
下一篇 2023年09月19日 09:31