From 0c336d5c5c0596691c9b33c08643c03486d47d5f Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期一, 27 四月 2026 18:56:39 +0800
Subject: [PATCH] refactor: move station buffer capacity to bas station
---
src/test/java/com/zy/ai/service/impl/AutoTuneSnapshotServiceImplTest.java | 30 ++++++++++++++++++++++++------
1 files changed, 24 insertions(+), 6 deletions(-)
diff --git a/src/test/java/com/zy/ai/service/impl/AutoTuneSnapshotServiceImplTest.java b/src/test/java/com/zy/ai/service/impl/AutoTuneSnapshotServiceImplTest.java
index 80cc4f7..6619623 100644
--- a/src/test/java/com/zy/ai/service/impl/AutoTuneSnapshotServiceImplTest.java
+++ b/src/test/java/com/zy/ai/service/impl/AutoTuneSnapshotServiceImplTest.java
@@ -52,6 +52,19 @@
}
@Test
+ void buildStationOutBufferCapacityMapPreservesConfiguredCapacity() {
+ Map<String, Integer> result = service.buildStationOutBufferCapacityMap(Arrays.asList(
+ station(101, 1, null),
+ station(102, 1, 0),
+ station(103, 1, 3)
+ ));
+
+ assertNull(result.get("101"));
+ assertEquals(0, result.get("102"));
+ assertEquals(3, result.get("103"));
+ }
+
+ @Test
void loadActiveTasksIncludesNullWrkStatusInGroupedCondition() {
WrkMastService wrkMastService = mock(WrkMastService.class);
when(wrkMastService.list(any(Wrapper.class))).thenReturn(Collections.emptyList());
@@ -68,7 +81,7 @@
}
@Test
- void loadStationOutTaskLimitsOnlyQueriesBasDevpOutStations() {
+ void loadOutStationListOnlyQueriesBasDevpOutStations() {
BasStationService basStationService = mock(BasStationService.class);
BasDevpService basDevpService = mock(BasDevpService.class);
ReflectionTestUtils.setField(service, "basStationService", basStationService);
@@ -82,11 +95,11 @@
station(102, 3)
));
- Map<String, Integer> result = ReflectionTestUtils.invokeMethod(service, "loadStationOutTaskLimits");
+ java.util.List<BasStation> result = ReflectionTestUtils.invokeMethod(service, "loadOutStationList");
assertEquals(2, result.size());
- assertEquals(2, result.get("101"));
- assertEquals(3, result.get("102"));
+ assertEquals(101, result.get(0).getStationId());
+ assertEquals(102, result.get(1).getStationId());
ArgumentCaptor<Wrapper<BasDevp>> basDevpWrapperCaptor = ArgumentCaptor.forClass(Wrapper.class);
verify(basDevpService).list(basDevpWrapperCaptor.capture());
@@ -98,23 +111,28 @@
}
@Test
- void loadStationOutTaskLimitsDoesNotQueryStationsWhenBasDevpOutStationsAreEmpty() {
+ void loadOutStationListDoesNotQueryStationsWhenBasDevpOutStationsAreEmpty() {
BasStationService basStationService = mock(BasStationService.class);
BasDevpService basDevpService = mock(BasDevpService.class);
ReflectionTestUtils.setField(service, "basStationService", basStationService);
ReflectionTestUtils.setField(service, "basDevpService", basDevpService);
when(basDevpService.list(any(Wrapper.class))).thenReturn(Collections.singletonList(basDevp(1)));
- Map<String, Integer> result = ReflectionTestUtils.invokeMethod(service, "loadStationOutTaskLimits");
+ java.util.List<BasStation> result = ReflectionTestUtils.invokeMethod(service, "loadOutStationList");
assertTrue(result.isEmpty());
verify(basStationService, never()).list(any(Wrapper.class));
}
private BasStation station(Integer stationId, Integer outTaskLimit) {
+ return station(stationId, outTaskLimit, null);
+ }
+
+ private BasStation station(Integer stationId, Integer outTaskLimit, Integer outBufferCapacity) {
BasStation station = new BasStation();
station.setStationId(stationId);
station.setOutTaskLimit(outTaskLimit);
+ station.setOutBufferCapacity(outBufferCapacity);
return station;
}
--
Gitblit v1.9.1