Making EligibilityVerificationService to accept AuthorizedServiceRequestContextv0.5.5

1 files changed, 5 insertions, 3 deletions

diff --git a/src/main/scala/xyz/driver/pdsuidomain/services/rest/RestEligibilityVerificationService.scala b/src/main/scala/xyz/driver/pdsuidomain/services/rest/RestEligibilityVerificationService.scala
index 07757e3..5a5ede7 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/services/rest/RestEligibilityVerificationService.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/services/rest/RestEligibilityVerificationService.scala
@@ -4,8 +4,9 @@ import akka.http.scaladsl.model.Uri
 import akka.stream.Materializer
 import spray.json.DefaultJsonProtocol
 import xyz.driver.core.Id
-import xyz.driver.core.rest.{RestService, ServiceRequestContext, ServiceTransport}
+import xyz.driver.core.rest.{AuthorizedServiceRequestContext, RestService, ServiceTransport}
 import xyz.driver.entities.patient
+import xyz.driver.entities.users.AuthUserInfo
 import xyz.driver.pdsuidomain.entities.eligibility.{MatchedPatient, MismatchRankedLabels}
 import xyz.driver.pdsuidomain.entities.{Arm, Patient, eligibility}
 import xyz.driver.pdsuidomain.services.EligibilityVerificationService
@@ -23,7 +24,8 @@ class RestEligibilityVerificationService(transport: ServiceTransport, baseUri: U
   import akka.http.scaladsl.marshallers.sprayjson.SprayJsonSupport._
   import xyz.driver.pdsuidomain.formats.json.sprayformats.eligibility._
 
-  override def getMatchedPatients()(implicit ctx: ServiceRequestContext): ListT[Future, eligibility.MatchedPatient] = {
+  override def getMatchedPatients()(
+          implicit ctx: AuthorizedServiceRequestContext[AuthUserInfo]): ListT[Future, eligibility.MatchedPatient] = {
     val request = get(baseUri, s"/v1/patients")
     listResponse[MatchedPatient](transport.sendRequest(ctx)(request))
   }
@@ -31,7 +33,7 @@ class RestEligibilityVerificationService(transport: ServiceTransport, baseUri: U
   override def getMismatchRankedLabels(patientId: Id[Patient],
                                        cancerType: patient.CancerType,
                                        excludedArms: Seq[Id[Arm]])(
-          implicit ctx: ServiceRequestContext): Future[eligibility.MismatchRankedLabels] = {
+          implicit ctx: AuthorizedServiceRequestContext[AuthUserInfo]): Future[eligibility.MismatchRankedLabels] = {
 
     val query =
       Seq("disease" -> cancerType.toString.toUpperCase, "ineligible_arms" -> excludedArms.map(_.value).mkString(","))