Spring/Springboot/SpringCloud项目快速基础Mockito单元测试

最新推荐文章于 2025-02-03 22:26:00 发布

lvyuanj

最新推荐文章于 2025-02-03 22:26:00 发布

阅读量372

点赞数 2

本文链接：https://blog.csdn.net/lvyuanj/article/details/137065702

版权

本文介绍了如何在Java项目中使用PowerMock和Mockito进行单元测试，包括引入相关jar包，Mock和Spy的区别，以及Mockito的自动注入机制。详细展示了如何在测试类中使用这些工具进行对象的模拟和替换。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Powermock+junit4单元测试

1、引入jar包

	  <dependency>
            <groupId>org.powermock</groupId>
            <artifactId>powermock-api-mockito2</artifactId>
            <version>${org.powermock.version}</version>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.powermock</groupId>
            <artifactId>powermock-module-junit4</artifactId>
            <version>${org.powermock.version}</version>
            <scope>test</scope>
        </dependency>

2、对于Mockito而言，有两种方式创建：

1. mock为一个interface提供一个虚拟的实现，
2. spy为object加一个动态代理，实现部分方法的虚拟化

3、 py 和 mock不同，不同点是：

spy 的参数是对象示例，mock 的参数是 class。
被 spy 的对象，调用其方法时默认会走真实方法。mock 对象不会

4、mockito 会将 `@Mock`、`@Spy` 修饰的对象自动注入到 `@InjectMocks` 修饰的对象中。

注入方式有多种，mockito 会按照下面的顺序尝试注入：

1.  构造函数注入
1.  设值函数注入（set函数）
1.  属性注入

@RunWith(PowerMockRunner.class)
public class StockAvailableDetailRecordTest {

    @InjectMocks
    private RabbitProducer rabbitProducer;  // 待测试的实例对象

    @Mock
    private RabbitTemplate rabbitTemplate;  // 需要Mock的对象

    @InjectMocks
    private TenantOmsImsServiceImpl tenantOmsImsServiceImpl;

    @Mock
    private IWarehouseService warehouseService;

    @Mock
    private IStockAvailableDetailService stockAvailableDetailService;

    //spy 和 mock不同，不同点是：
    //    spy 的参数是对象示例，mock 的参数是 class。
    //    被 spy 的对象，调用其方法时默认会走真实方法。mock 对象不会。


    @Test
    public void testStockDetailRecord(){
        List<StockAvailableDetailRecord> records = new ArrayList<>();
        rabbitProducer.sendMqMessage(QueueDeclareEnum.STOCK_AVAILABLE_DETAIL_RECORD, records,"000000");
    }

    @Test
    public void testListAvailableStock(){
        OmsAvailableStockReqDTO omsAvailableStockReqDTO = preOmsAvailableStockReqDto();

        Warehouse warehouse = new Warehouse();
        warehouse.setId(227L);
        String warehouseCode = "USWEST1227";
        warehouse.setWarehouseCode(warehouseCode);
        Mockito.when(warehouseService.getWarehouseByCode(warehouseCode)).thenReturn(warehouse);

        GetAvailableByShopQuery query = new GetAvailableByShopQuery();
        query.setProductAuthId("000000");
        query.setProductOwnerId(0L);
        query.setWarehouseId(227L);
        query.setShopCode("amazon_US_05TK");
        query.setProductSku("QP80X928J9H");
        query.setOutputQty(1);

        Mockito.when(stockAvailableDetailService.addOverseaOwoAvailable(Mockito.any(), Mockito.any())).thenReturn(1);

        List<OwoAvailableDetailDto> availableDetails = new ArrayList<>();
        OwoAvailableDetailDto owoAvailableDetailDto = new OwoAvailableDetailDto();
        owoAvailableDetailDto.setStockId(20220106000001L);
        availableDetails.add(owoAvailableDetailDto);

        R<OmsAvailableStockReqDTO> res = tenantOmsImsServiceImpl.listAvailableStock(omsAvailableStockReqDTO);
        Assert.isTrue(res.isSuccess(), "fail");
        OmsAvailableStockReqDTO data = res.getData();

    }


    private OmsAvailableStockReqDTO preOmsAvailableStockReqDto(){

        OmsAvailableStockReqDTO omsAvailableStockReqDTO = new OmsAvailableStockReqDTO();
        ArrayList<OmsAvailableStockReqDTO.Detail> details = new ArrayList<>();

        omsAvailableStockReqDTO.setProductAuthId("000000");
        omsAvailableStockReqDTO.setDetails(details);

        OmsAvailableStockReqDTO.Detail detail = new OmsAvailableStockReqDTO.Detail();
        detail.setProductOwnerId(0L);
        detail.setWarehouseCode("USWEST1227");
        detail.setShopCode("amazon_US_05TK");
        detail.setProductSku("QP80X928J9H");
        detail.setProductQty(1);

        details.add(detail);
        return omsAvailableStockReqDTO;
    }

}