package cn.iocoder.yudao.module.mes.service.pro.task.schedule;
|
|
import org.junit.jupiter.api.Test;
|
import java.math.BigDecimal;
|
import java.time.*;
|
import java.util.*;
|
import static org.junit.jupiter.api.Assertions.*;
|
|
class MesProScheduleEngineTest {
|
private final MesProScheduleEngine.Calendar calendar = new MesProScheduleEngine.Calendar(Set.of(), LocalDate.of(2027, 12, 31));
|
private LocalDateTime at(String text) { return LocalDateTime.parse(text); }
|
|
@Test void openingClosingAndWeekend() {
|
assertEquals(at("2026-10-16T09:00"), calendar.nextWorkingTime(at("2026-10-16T07:20")));
|
assertEquals(at("2026-10-19T09:00"), calendar.nextWorkingTime(at("2026-10-16T18:00")));
|
assertEquals(at("2026-10-19T09:00"), calendar.nextWorkingTime(at("2026-10-17T12:00")));
|
}
|
@Test void exactShiftAndCrossWeekend() {
|
assertEquals(at("2026-10-16T18:00"), calendar.endTime(at("2026-10-16T09:00"), 9));
|
assertEquals(at("2026-10-19T10:00"), calendar.endTime(at("2026-10-16T17:00"), 2));
|
}
|
@Test void skipsConfiguredHoliday() {
|
var custom = new MesProScheduleEngine.Calendar(Set.of(LocalDate.of(2026, 10, 19)), LocalDate.of(2027, 1, 1));
|
assertEquals(at("2026-10-20T10:00"), custom.endTime(at("2026-10-16T17:00"), 2));
|
}
|
@Test void rejectsNoWorkingWindowAndInvalidCapacity() {
|
var closed = new MesProScheduleEngine.Calendar(Set.of(LocalDate.of(2026, 10, 16)), LocalDate.of(2026, 10, 16));
|
assertThrows(IllegalArgumentException.class, () -> closed.nextWorkingTime(at("2026-10-16T09:00")));
|
assertThrows(IllegalArgumentException.class, () -> MesProScheduleEngine.duration(BigDecimal.ONE, BigDecimal.ZERO));
|
assertThrows(IllegalArgumentException.class, () -> MesProScheduleEngine.duration(BigDecimal.ZERO, BigDecimal.ONE));
|
}
|
@Test void quantitiesRoundUpWithoutDroppingDecimals() {
|
assertEquals(2, MesProScheduleEngine.duration(new BigDecimal("100.01"), new BigDecimal("100")));
|
assertEquals(1, MesProScheduleEngine.duration(new BigDecimal("0.01"), new BigDecimal("100")));
|
}
|
@Test void sharedMachineAndStationCannotOverlap() {
|
var busy = List.of(new MesProScheduleEngine.Busy(1L, Set.of(10L), at("2026-10-15T09:00"), at("2026-10-15T11:00")));
|
var sharedMachine = MesProScheduleEngine.findSlot(2L, Set.of(10L), BigDecimal.TEN, BigDecimal.TEN, at("2026-10-15T09:00"), busy, calendar);
|
assertEquals(at("2026-10-15T11:00"), sharedMachine.start());
|
var sameStation = MesProScheduleEngine.findSlot(1L, Set.of(11L), BigDecimal.TEN, BigDecimal.TEN, at("2026-10-15T09:00"), busy, calendar);
|
assertEquals(at("2026-10-15T11:00"), sameStation.start());
|
var independent = MesProScheduleEngine.findSlot(2L, Set.of(11L), BigDecimal.TEN, BigDecimal.TEN, at("2026-10-15T09:00"), busy, calendar);
|
assertEquals(at("2026-10-15T09:00"), independent.start());
|
}
|
@Test void boundaryTouchAndMultipleConflicts() {
|
var busy = List.of(
|
new MesProScheduleEngine.Busy(1L, Set.of(), at("2026-10-15T10:00"), at("2026-10-15T12:00")),
|
new MesProScheduleEngine.Busy(1L, Set.of(), at("2026-10-15T12:00"), at("2026-10-15T15:00")));
|
var shortSlot = MesProScheduleEngine.findSlot(1L, Set.of(), BigDecimal.ONE, BigDecimal.ONE, at("2026-10-15T09:00"), busy, calendar);
|
assertEquals(at("2026-10-15T10:00"), shortSlot.end());
|
var longSlot = MesProScheduleEngine.findSlot(1L, Set.of(), BigDecimal.TEN, new BigDecimal("5"), at("2026-10-15T09:00"), busy, calendar);
|
assertEquals(at("2026-10-15T15:00"), longSlot.start());
|
}
|
@Test void randomizedPlansAlwaysStayWithinShiftsAndDoNotOverlap() {
|
var random = new Random(17);
|
List<MesProScheduleEngine.Busy> busy = new ArrayList<>();
|
for (int i = 0; i < 150; i++) {
|
var slot = MesProScheduleEngine.findSlot(1L, Set.of(10L), BigDecimal.valueOf(random.nextInt(400) + 1),
|
new BigDecimal("100"), at("2026-10-15T09:00"), busy, calendar);
|
assertTrue(calendar.isWorkingDay(slot.start().toLocalDate()));
|
assertTrue(calendar.isWorkingDay(slot.end().toLocalDate()));
|
assertFalse(slot.start().toLocalTime().isBefore(LocalTime.of(9, 0)));
|
assertFalse(slot.end().toLocalTime().isAfter(LocalTime.of(18, 0)));
|
assertTrue(busy.stream().noneMatch(interval -> MesProScheduleEngine.overlaps(slot.start(), slot.end(), interval)));
|
busy.add(new MesProScheduleEngine.Busy(1L, Set.of(10L), slot.start(), slot.end()));
|
}
|
}
|
}
|
