diff --git a/backend/src/test/java/ca/bc/gov/hlth/hnweb/controller/MaintenanceControllerTest.java b/backend/src/test/java/ca/bc/gov/hlth/hnweb/controller/MaintenanceControllerTest.java index 3a628c16..89737670 100644 --- a/backend/src/test/java/ca/bc/gov/hlth/hnweb/controller/MaintenanceControllerTest.java +++ b/backend/src/test/java/ca/bc/gov/hlth/hnweb/controller/MaintenanceControllerTest.java @@ -611,7 +611,7 @@ public void testExtendCancelDate_success() throws Exception { assertEquals(MediaType.TEXT_PLAIN.toString(), recordedRequest.getHeader(CONTENT_TYPE)); assertTransactionCreated(TransactionType.EXTEND_CANCEL_DATE); - assertAffectedPartyCount(AffectedPartyDirection.INBOUND, 1); + assertAffectedPartyCount(AffectedPartyDirection.INBOUND, 2); assertAffectedPartyCount(AffectedPartyDirection.OUTBOUND, 1); } @@ -638,7 +638,7 @@ public void testExtendCancelDate_error_invalidImmigrationCode() throws Exception assertEquals(MediaType.TEXT_PLAIN.toString(), recordedRequest.getHeader(CONTENT_TYPE)); assertTransactionCreated(TransactionType.EXTEND_CANCEL_DATE); - assertAffectedPartyCount(AffectedPartyDirection.INBOUND, 1); + assertAffectedPartyCount(AffectedPartyDirection.INBOUND, 2); assertAffectedPartyCount(AffectedPartyDirection.OUTBOUND, 1); } @@ -665,7 +665,7 @@ public void testExtendCancelDate_error_expiryDateMustBeAfterIssueDate() throws E assertEquals(MediaType.TEXT_PLAIN.toString(), recordedRequest.getHeader(CONTENT_TYPE)); assertTransactionCreated(TransactionType.EXTEND_CANCEL_DATE); - assertAffectedPartyCount(AffectedPartyDirection.INBOUND, 1); + assertAffectedPartyCount(AffectedPartyDirection.INBOUND, 2); assertAffectedPartyCount(AffectedPartyDirection.OUTBOUND, 1); } @@ -693,7 +693,7 @@ public void testExtendCancelDate_error_visaDateChangeRejected() throws Exception assertEquals(MediaType.TEXT_PLAIN.toString(), recordedRequest.getHeader(CONTENT_TYPE)); assertTransactionCreated(TransactionType.EXTEND_CANCEL_DATE); - assertAffectedPartyCount(AffectedPartyDirection.INBOUND, 1); + assertAffectedPartyCount(AffectedPartyDirection.INBOUND, 2); assertAffectedPartyCount(AffectedPartyDirection.OUTBOUND, 1); }