package cn.iocoder.yudao.module.mes.service.pro.task.schedule;

import org.junit.jupiter.api.Test;
import java.math.BigDecimal;
import java.util.Map;
import static org.junit.jupiter.api.Assertions.*;

class MesScheduleMaterialCheckTest {
    @Test void oneBatchCannotUseSameStockTwice() {
        var stock = Map.of(1L, new BigDecimal("10"));
        var first = MesScheduleMaterialCheck.allocate(stock, Map.of(1L, new BigDecimal("6")));
        assertEquals(0, first.get(1L).compareTo(new BigDecimal("4")));
        assertThrows(IllegalArgumentException.class, () -> MesScheduleMaterialCheck.allocate(first, Map.of(1L, new BigDecimal("5"))));
        assertEquals(0, stock.get(1L).compareTo(new BigDecimal("10")));
    }
    @Test void shortageOrInvalidRequirementIsRejected() {
        assertThrows(IllegalArgumentException.class, () -> MesScheduleMaterialCheck.allocate(Map.of(), Map.of(2L, BigDecimal.ONE)));
        assertThrows(IllegalArgumentException.class, () -> MesScheduleMaterialCheck.allocate(Map.of(2L, BigDecimal.TEN), Map.of(2L, BigDecimal.ZERO)));
    }
    @Test void sufficientStockCanBeAllocated() {
        assertEquals(0, MesScheduleMaterialCheck.allocate(Map.of(2L, BigDecimal.TEN), Map.of(2L, BigDecimal.TEN)).get(2L).signum());
    }
}