feat(openapi): Enhance batch processing operations

Ahoo-Wang · Nov 17, 2024 · 2b2cf2e · 2b2cf2e
1 parent bba6bca
commit 2b2cf2e
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/wow-webflux/src/test/kotlin/me/ahoo/wow/webflux/route/BatchResultsKtTest.kt b/wow-webflux/src/test/kotlin/me/ahoo/wow/webflux/route/BatchResultsKtTest.kt
@@ -42,14 +42,17 @@ class BatchResultsKtTest {
 
     @Test
     fun toBatchResultWhenError() {
-        // Arrange
-        val flux = Flux.error<AggregateId>(RuntimeException("error"))
+        val flux = Flux.create<AggregateId> { sink ->
+            sink.next(MOCK_AGGREGATE_METADATA.aggregateId("id1"))
+            sink.error(RuntimeException("error"))
+            sink.next(MOCK_AGGREGATE_METADATA.aggregateId("id2"))
+        }
         val afterId = "id0"
 
         // Act
         flux.toBatchResult(afterId)
             .test()
-            .expectNext(BatchResult(afterId, 0, ErrorCodes.BAD_REQUEST, "error"))
+            .expectNext(BatchResult("id1", 1, ErrorCodes.BAD_REQUEST, "error"))
             .verifyComplete()
     }
 }