From 319d2494453c4c5606e3adabbbff66c865b467a5 Mon Sep 17 00:00:00 2001 From: Brian Neumann-Fopiano Date: Tue, 29 Sep 2026 00:33:13 -0400 Subject: [PATCH] History sync test drain stub - Close test stubs the admission drain with the deadline signature - Stubbed drain reports every request drained --- .../arcane/iris/world/pregen/AsyncPregenHistorySyncTest.java | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/core/src/test/java/art/arcane/iris/world/pregen/AsyncPregenHistorySyncTest.java b/core/src/test/java/art/arcane/iris/world/pregen/AsyncPregenHistorySyncTest.java index 50a6b2f3b..487237cd2 100644 --- a/core/src/test/java/art/arcane/iris/world/pregen/AsyncPregenHistorySyncTest.java +++ b/core/src/test/java/art/arcane/iris/world/pregen/AsyncPregenHistorySyncTest.java @@ -37,7 +37,7 @@ public class AsyncPregenHistorySyncTest { method.close(); } InOrder order = inOrder(admission, engine); - order.verify(admission).awaitDrain(anyLong(), any(), any()); + order.verify(admission).awaitDrain(anyLong(), any(), anyLong(), any(), any()); order.verify(engine).syncGenerationHistory(); } @@ -63,6 +63,8 @@ public class AsyncPregenHistorySyncTest { set(method, "world", world); set(method, "closing", new AtomicBoolean()); set(method, "holdsWorkerBoost", new AtomicBoolean()); + when(admission.awaitDrain(anyLong(), any(), anyLong(), any(), any())) + .thenReturn(new PregenAdmissionGate.Drain(true, false, 0)); set(method, "admission", admission); set(method, "mantleCleanup", new PregenSerialWorker("Test Mantle Cleanup", "world")); set(method, "chunkFlush", new PregenSerialWorker("Test Chunk Flush", "world"));