diff options
author | Aleksandr <ognelisar@gmail.com> | 2017-10-13 14:13:28 +0700 |
---|---|---|
committer | Aleksandr <ognelisar@gmail.com> | 2017-10-13 14:13:28 +0700 |
commit | 2a1bba7d8b5fbc6a2d665c10ffea110bc7d89b46 (patch) | |
tree | 2c3536e446744b82ca4a02744643ca5dc2d98bf9 /src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala | |
parent | f466bbbdb2cea1f64f7b3781795f890aaf390acf (diff) | |
download | rest-query-2a1bba7d8b5fbc6a2d665c10ffea110bc7d89b46.tar.gz rest-query-2a1bba7d8b5fbc6a2d665c10ffea110bc7d89b46.tar.bz2 rest-query-2a1bba7d8b5fbc6a2d665c10ffea110bc7d89b46.zip |
Added swagger annotations for new entities
Diffstat (limited to 'src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala')
-rw-r--r-- | src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala | 47 |
1 files changed, 47 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..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 + ) } |