[MODORDERS-963] - Fixed ProtectedEntityTestBase

folio-org · Nov 28, 2023 · 72a3dea · 72a3dea
1 parent b2d458f
commit 72a3dea
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/test/java/org/folio/TestUtils.java b/src/test/java/org/folio/TestUtils.java
@@ -175,6 +175,10 @@ public static CompositePoLine getMinimalContentCompositePoLine(String orderId) {
       .withPurchaseOrderId(orderId);
   }
 
+  public static Title getMinimalContentTitle() {
+    return new Title().withTitle("Test title").withId(SAMPLE_TITLE_ID);
+  }
+
   public static CompositePoLine getMinimalPackageCompositePoLine(String orderId) {
     return new CompositePoLine().withSource(CompositePoLine.Source.EDI)
       .withId(MIN_PO_LINE_ID)

diff --git a/src/test/java/org/folio/rest/impl/protection/ProtectedEntityTestBase.java b/src/test/java/org/folio/rest/impl/protection/ProtectedEntityTestBase.java
@@ -3,10 +3,12 @@
 import static org.folio.TestUtils.getMinimalContentCompositePoLine;
 import static org.folio.TestUtils.getMinimalContentCompositePurchaseOrder;
 import static org.folio.TestUtils.getMinimalContentPiece;
+import static org.folio.TestUtils.getMinimalContentTitle;
 import static org.folio.TestUtils.getRandomId;
 import static org.folio.orders.utils.ResourcePathResolver.PIECES_STORAGE;
 import static org.folio.orders.utils.ResourcePathResolver.PO_LINES_STORAGE;
 import static org.folio.orders.utils.ResourcePathResolver.PURCHASE_ORDER_STORAGE;
+import static org.folio.orders.utils.ResourcePathResolver.TITLES;
 import static org.folio.rest.RestVerticle.OKAPI_USERID_HEADER;
 import static org.folio.rest.impl.MockServer.addMockEntry;
 import static org.hamcrest.MatcherAssert.assertThat;
@@ -24,6 +26,7 @@
 import org.folio.rest.jaxrs.model.CompositePoLine;
 import org.folio.rest.jaxrs.model.CompositePurchaseOrder;
 import org.folio.rest.jaxrs.model.Piece;
+import org.folio.rest.jaxrs.model.Title;
 import org.hamcrest.Matcher;
 
 import io.restassured.http.Header;
@@ -73,9 +76,18 @@ public CompositePoLine preparePoLine(List<String> acqUnitsIds,
     return poLine;
   }
 
+  public Title prepareTitle(List<String> acqUnitIds) {
+    Title title = getMinimalContentTitle();
+    title.setAcqUnitIds(acqUnitIds);
+    addMockEntry(TITLES, JsonObject.mapFrom(title));
+    return title;
+  }
+
   public Piece preparePiece(List<String> acqUnitsIds) {
-    CompositePoLine poLine = preparePoLine(acqUnitsIds, CompositePurchaseOrder.WorkflowStatus.OPEN);
+    CompositePoLine poLine = preparePoLine(new ArrayList<>(), CompositePurchaseOrder.WorkflowStatus.OPEN);
+    Title title = prepareTitle(acqUnitsIds);
     Piece piece = getMinimalContentPiece(poLine.getId());
+    piece.setTitleId(title.getId());
     addMockEntry(PIECES_STORAGE, JsonObject.mapFrom(piece));
 
     return piece;