add EligibilityArmDisease entity

1 files changed, 11 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/pdsuidomain/entities/EligibilityArm.scala b/src/main/scala/xyz/driver/pdsuidomain/entities/EligibilityArm.scala
index 2f5924c..bf81e2c 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/entities/EligibilityArm.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/entities/EligibilityArm.scala
@@ -4,6 +4,7 @@ import java.time.LocalDateTime
 
 import xyz.driver.pdsuicommon.domain.{LongId, StringId}
 import xyz.driver.pdsuicommon.logging._
+import xyz.driver.pdsuidomain.entities.Trial.Condition
 
 final case class EligibilityArm(id: LongId[EligibilityArm],
                                 name: String,
@@ -18,3 +19,13 @@ object EligibilityArm {
     phi"Arm(id=$id, name=${Unsafe(x.name)}, trialId=${Unsafe(x.trialId)})"
   }
 }
+
+final case class EligibilityArmDisease(eligibilityArmId: LongId[EligibilityArmDisease],
+                                       disease: Condition)
+
+object EligibilityArmDisease {
+
+  implicit def toPhiString(x: EligibilityArmDisease): PhiString = {
+    phi"EligibilityArmDisease(eligibilityArmId=${Unsafe(x.eligibilityArmId)}, disease=${Unsafe(x.disease)})"
+  }
+}