path: root/src/main/scala/xyz/driver/pdsuidomain/services/fake/FakeEligibilityVerificationService.scala

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.users.AuthUserInfo
import xyz.driver.pdsuidomain.entities.eligibility.MismatchRankedLabels
import xyz.driver.pdsuidomain.entities.{Arm, Patient, 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)
      ))
}