EPMRPP-95844 fix find by id

reportportal · Jan 3, 2025 · 52b4cd5 · 52b4cd5
1 parent d3856df
commit 52b4cd5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/java/com/epam/ta/reportportal/core/user/impl/UserInvitationHandlerImpl.java b/src/main/java/com/epam/ta/reportportal/core/user/impl/UserInvitationHandlerImpl.java
@@ -28,6 +28,7 @@
 import com.epam.reportportal.api.model.Invitation;
 import com.epam.reportportal.api.model.InvitationRequest;
 import com.epam.reportportal.api.model.InvitationRequestOrganizationsInner;
+import com.epam.reportportal.rules.exception.ErrorType;
 import com.epam.reportportal.rules.exception.ReportPortalException;
 import com.epam.ta.reportportal.commons.ReportPortalUser;
 import com.epam.ta.reportportal.core.integration.GetIntegrationHandler;
@@ -148,7 +149,9 @@ public Invitation createUserInvitation(InvitationRequest request, ReportPortalUs
 
   @Override
   public Invitation getInvitation(String invitationId, String baseUrl) {
-    var bid = userCreationBidRepository.getById(invitationId);
+    var bid = userCreationBidRepository.findById(invitationId)
+        .orElseThrow(() -> new ReportPortalException(ErrorType.NOT_FOUND, "User invitation id"));
+
     var invitation = InvitationConverter.TO_INVITATION.apply(bid);
     invitation.setLink(getEmailLink(baseUrl, bid.getUuid()));
     return invitation;