[MODORDSTOR-435] Update unit tests

folio-org · Dec 19, 2024 · 91e4e32 · 91e4e32
1 parent f5583b9
commit 91e4e32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/test/java/org/folio/service/pieces/flows/update/PieceUpdateFlowManagerTest.java b/src/test/java/org/folio/service/pieces/flows/update/PieceUpdateFlowManagerTest.java
@@ -347,7 +347,7 @@ void shouldUpdatePiecesStatusesSuccessfully() {
     doReturn(succeededFuture(poLine2)).when(purchaseOrderLineService).getOrderLineById(poLine2.getId(), requestContext);
     doReturn(succeededFuture(purchaseOrder)).when(purchaseOrderStorageService).getPurchaseOrderById(purchaseOrder.getId(), requestContext);
     doReturn(succeededFuture()).when(pieceUpdateFlowPoLineService).updatePoLine(any(), eq(requestContext));
-    doReturn(succeededFuture()).when(pieceStorageService).updatePiece(any(), eq(requestContext));
+    doReturn(succeededFuture()).when(pieceStorageService).updatePiecesBatch(any(), eq(requestContext));
     doNothing().when(pieceService).receiptConsistencyPiecePoLine(anyString(), eq(requestContext));
 
     Future<Void> result = pieceUpdateFlowManager.updatePiecesStatuses(pieceIds, receivingStatus, requestContext);