package cn.iocoder.yudao.module.wms.service.md.warehouse;

import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
import cn.iocoder.yudao.module.wms.controller.admin.md.warehouse.vo.WmsWarehouseSaveReqVO;
import cn.iocoder.yudao.module.wms.dal.dataobject.md.warehouse.WmsWarehouseDO;
import cn.iocoder.yudao.module.wms.dal.mysql.md.warehouse.WmsWarehouseMapper;
import cn.iocoder.yudao.module.wms.service.inventory.WmsInventoryService;
import cn.iocoder.yudao.module.wms.service.order.check.WmsCheckOrderService;
import cn.iocoder.yudao.module.wms.service.order.movement.WmsMovementOrderService;
import cn.iocoder.yudao.module.wms.service.order.receipt.WmsReceiptOrderService;
import cn.iocoder.yudao.module.wms.service.order.shipment.WmsShipmentOrderService;
import jakarta.annotation.Resource;
import org.junit.jupiter.api.Test;
import org.springframework.context.annotation.Import;
import org.springframework.test.context.bean.override.mockito.MockitoBean;

import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertServiceException;
import static cn.iocoder.yudao.module.wms.enums.ErrorCodeConstants.WAREHOUSE_CODE_DUPLICATE;
import static cn.iocoder.yudao.module.wms.enums.ErrorCodeConstants.WAREHOUSE_HAS_INVENTORY;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.mockito.Mockito.when;

@Import(WmsWarehouseServiceImpl.class)
public class WmsWarehouseServiceImplTest extends BaseDbUnitTest {

    @Resource
    private WmsWarehouseServiceImpl warehouseService;

    @Resource
    private WmsWarehouseMapper warehouseMapper;

    @MockitoBean
    private WmsInventoryService inventoryService;
    @MockitoBean
    private WmsReceiptOrderService receiptOrderService;
    @MockitoBean
    private WmsShipmentOrderService shipmentOrderService;
    @MockitoBean
    private WmsMovementOrderService movementOrderService;
    @MockitoBean
    private WmsCheckOrderService checkOrderService;

    @Test
    public void testCreateWarehouse_codeDuplicate() {
        // mock 数据
        WmsWarehouseDO warehouse = createWarehouse("WH001", "成品仓");
        warehouseMapper.insert(warehouse);

        // 调用，并断言
        WmsWarehouseSaveReqVO reqVO = createWarehouseSaveReqVO();
        assertServiceException(() -> warehouseService.createWarehouse(reqVO), WAREHOUSE_CODE_DUPLICATE);
    }

    @Test
    public void testUpdateWarehouse_codeDuplicate() {
        // mock 数据
        WmsWarehouseDO warehouse = createWarehouse("WH001", "成品仓");
        warehouseMapper.insert(warehouse);
        WmsWarehouseDO updateWarehouse = createWarehouse("WH002", "原料仓");
        warehouseMapper.insert(updateWarehouse);

        // 调用，并断言
        WmsWarehouseSaveReqVO reqVO = createWarehouseSaveReqVO();
        reqVO.setId(updateWarehouse.getId());
        assertServiceException(() -> warehouseService.updateWarehouse(reqVO), WAREHOUSE_CODE_DUPLICATE);
    }

    @Test
    public void testDeleteWarehouse_hasInventory() {
        // mock 数据
        WmsWarehouseDO warehouse = createWarehouse();
        warehouseMapper.insert(warehouse);
        when(inventoryService.getInventoryCountByWarehouseId(warehouse.getId())).thenReturn(1L);

        // 调用，并断言
        assertServiceException(() -> warehouseService.deleteWarehouse(warehouse.getId()), WAREHOUSE_HAS_INVENTORY);
        assertNotNull(warehouseMapper.selectById(warehouse.getId()));
    }

    private static WmsWarehouseDO createWarehouse() {
        return createWarehouse("WH001", "成品仓");
    }

    private static WmsWarehouseDO createWarehouse(String code, String name) {
        return WmsWarehouseDO.builder()
                .code(code)
                .name(name)
                .sort(1)
                .build();
    }

    private static WmsWarehouseSaveReqVO createWarehouseSaveReqVO() {
        WmsWarehouseSaveReqVO reqVO = new WmsWarehouseSaveReqVO();
        reqVO.setCode("WH001");
        reqVO.setName("半成品仓");
        reqVO.setSort(1);
        return reqVO;
    }

}