add NoFoundError to CreateReply

1 files changed, 6 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/pdsuidomain/services/EligibilityArmService.scala b/src/main/scala/xyz/driver/pdsuidomain/services/EligibilityArmService.scala
index a56e30a..e250e8f 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/services/EligibilityArmService.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/services/EligibilityArmService.scala
@@ -19,6 +19,10 @@ object EligibilityArmService {
     def userMessage: String = "EligibilityArm not found"
   }
 
+  trait SlotArmNotFoundError {
+    def userMessage: String = "SlotArm not found"
+  }
+
   sealed trait GetByIdReply
   object GetByIdReply {
 
@@ -78,6 +82,8 @@ object EligibilityArmService {
     case object AuthorizationError
         extends CreateReply with DefaultAccessDeniedError with DomainError.AuthorizationError
 
+    case object NotFoundError extends CreateReply with SlotArmNotFoundError with DomainError.NotFoundError
+
     final case class CommonError(userMessage: String) extends CreateReply with DomainError
 
     final case class AlreadyExistsError(x: EligibilityArmWithDiseases) extends CreateReply with DomainError {