diff options
author | Vlad Uspensky <v.uspenskiy@icloud.com> | 2017-10-18 00:23:58 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-10-18 00:23:58 -0400 |
commit | 2a9bada86658ef15793f99f1867e23f3b31e7032 (patch) | |
tree | e10018be9eac9640d67c576d9f81e1021ae8edb6 /src/main/scala/xyz/driver/pdsuidomain/fakes/entities/trialcuration.scala | |
parent | eb9a75c226b005e5989d2a6494160ebe0dd9d9e7 (diff) | |
parent | 9997ad1a778d6ed32f5dd90634be371986e4cca9 (diff) | |
download | rest-query-2a9bada86658ef15793f99f1867e23f3b31e7032.tar.gz rest-query-2a9bada86658ef15793f99f1867e23f3b31e7032.tar.bz2 rest-query-2a9bada86658ef15793f99f1867e23f3b31e7032.zip |
Merge pull request #46 from drivergroup/PDSUI-additional-formats
Implemented format for EligibilityArmWithDiseases class; Fixed formats for other classes
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 | 38 |
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 + ) } |