Merge pull request #46 from drivergroup/PDSUI-additional-formats

 Implemented format for EligibilityArmWithDiseases class; Fixed formats for other classes

1 files changed, 38 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala b/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala
index ade0115..84dcb88 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala
@@ -1,15 +1,19 @@
 package xyz.driver.pdsuidomain.fakes.entities
 
+import xyz.driver.core.generators._
 import xyz.driver.entities.labels.{Label, LabelCategory}
 import xyz.driver.pdsuicommon.domain.{LongId, User}
 import xyz.driver.pdsuidomain.entities._
 import xyz.driver.pdsuidomain.services.CriterionService.RichCriterion
 
 object trialcuration {
+
   import common._
   import xyz.driver.core.generators
   import xyz.driver.pdsuidomain.entities.InterventionType._
 
+  private val maxItemsInCollectionNumber: Int = 5
+
   def nextTrial(): Trial = Trial(
     id = nextStringId[Trial],
     externalId = nextUuidId[Trial],
@@ -144,4 +148,38 @@ object trialcuration {
     SurgeryProcedure
   )
 
+  def nextEligibilityArm(): EligibilityArm = EligibilityArm(
+    id = nextLongId,
+    name = nextString(),
+    originalName = nextString(),
+    trialId = nextStringId
+  )
+
+  def nextEligibilityArmDisease(): EligibilityArmDisease = EligibilityArmDisease(
+    eligibilityArmId = nextLongId,
+    disease = nextCancerType
+  )
+
+  private def nextEligibilityArmDiseaseCollection(count: Int): Seq[EligibilityArmDisease] =
+    Seq.fill(count)(nextEligibilityArmDisease())
+
+  def nextEligibilityArmWithDiseases(): EligibilityArmWithDiseases = {
+    val entity = nextEligibilityArm()
+    val id     = entity.id
+    val collection = nextEligibilityArmDiseaseCollection(
+      nextInt(maxItemsInCollectionNumber, minValue = 0)
+    ).map(_.copy(eligibilityArmId = id))
+
+    EligibilityArmWithDiseases(
+      entity,
+      collection
+    )
+  }
+
+  def nextSlotArm(): SlotArm = SlotArm(
+    id = nextLongId,
+    name = nextString(),
+    originalName = nextString(),
+    trialId = nextStringId
+  )
 }