From 2a1bba7d8b5fbc6a2d665c10ffea110bc7d89b46 Mon Sep 17 00:00:00 2001 From: Aleksandr Date: Fri, 13 Oct 2017 14:13:28 +0700 Subject: Added swagger annotations for new entities --- .../pdsuidomain/fakes/entities/trialcuration.scala | 47 ++++++++++++++++++++++ 1 file changed, 47 insertions(+) (limited to 'src/main/scala/xyz/driver/pdsuidomain/fakes') 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..d323e86 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,20 @@ package xyz.driver.pdsuidomain.fakes.entities +import xyz.driver.core.generators._ import xyz.driver.entities.labels.{Label, LabelCategory} +import xyz.driver.entities.patient.CancerType 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 +149,46 @@ object trialcuration { SurgeryProcedure ) + EligibilityArmDisease + + def nextEligibilityArm(): EligibilityArm = EligibilityArm( + id = nextLongId, + name = nextString(), + originalName = nextString(), + trialId = nextStringId + ) + + def nextPatientCancerType(): CancerType = generators.oneOf[CancerType]( + CancerType.Lung, + CancerType.Breast, + CancerType.Prostate + ) + + def nextEligibilityArmDisease(): EligibilityArmDisease = EligibilityArmDisease( + eligibilityArmId = nextLongId, + disease = nextPatientCancerType() + ) + + 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 + ) } -- cgit v1.2.3 From 9997ad1a778d6ed32f5dd90634be371986e4cca9 Mon Sep 17 00:00:00 2001 From: Aleksandr Date: Tue, 17 Oct 2017 16:43:15 +0700 Subject: Removed unused import --- src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala | 1 - 1 file changed, 1 deletion(-) (limited to 'src/main/scala/xyz/driver/pdsuidomain/fakes') 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 e618305..84dcb88 100644 --- a/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala +++ b/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala @@ -2,7 +2,6 @@ package xyz.driver.pdsuidomain.fakes.entities import xyz.driver.core.generators._ import xyz.driver.entities.labels.{Label, LabelCategory} -import xyz.driver.entities.patient.CancerType import xyz.driver.pdsuicommon.domain.{LongId, User} import xyz.driver.pdsuidomain.entities._ import xyz.driver.pdsuidomain.services.CriterionService.RichCriterion -- cgit v1.2.3