Merge branch 'add-slot-eligibility-arms' into trial-46-disease-entity

1 files changed, 31 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/pdsuidomain/services/fake/FakeEligibilityVerificationService.scala b/src/main/scala/xyz/driver/pdsuidomain/services/fake/FakeEligibilityVerificationService.scala
new file mode 100644
index 0000000..1e1a3ea
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuidomain/services/fake/FakeEligibilityVerificationService.scala
@@ -0,0 +1,31 @@
+package xyz.driver.pdsuidomain.services.fake
+
+import xyz.driver.core.rest.AuthorizedServiceRequestContext
+import xyz.driver.core.{Id, generators}
+import xyz.driver.entities.patient
+import xyz.driver.entities.patient.Patient
+import xyz.driver.entities.users.AuthUserInfo
+import xyz.driver.pdsuidomain.entities.eligibility.MismatchRankedLabels
+import xyz.driver.pdsuidomain.entities.{Arm, eligibility}
+import xyz.driver.pdsuidomain.services.EligibilityVerificationService
+
+import scala.concurrent.Future
+import scalaz.ListT
+
+class FakeEligibilityVerificationService extends EligibilityVerificationService {
+
+  override def getMatchedPatients()(
+          implicit ctx: AuthorizedServiceRequestContext[AuthUserInfo]): ListT[Future, eligibility.MatchedPatient] =
+    ListT.listT[Future](
+      Future.successful(List(xyz.driver.pdsuidomain.fakes.entities.eligibility.nextMatchedPatient())))
+
+  override def getMismatchRankedLabels(patientId: Id[Patient],
+                                       cancerType: patient.CancerType,
+                                       excludedArms: Seq[Id[Arm]])(
+          implicit ctx: AuthorizedServiceRequestContext[AuthUserInfo]): Future[eligibility.MismatchRankedLabels] =
+    Future.successful(
+      MismatchRankedLabels(
+        generators.seqOf(xyz.driver.pdsuidomain.fakes.entities.eligibility.nextLabelMismatchRank()),
+        labelVersion = generators.nextInt(10)
+      ))
+}