From d59ef922e4412f4762145bfdee9bb781aa0ce4d4 Mon Sep 17 00:00:00 2001
From: Janet Gainer-Dewar <jdewar@broadinstitute.org>
Date: Thu, 17 Oct 2024 09:09:42 -0400
Subject: [PATCH] AN-138 Enable cost endpoint in CromIAM (#7572)

---
 .../webservice/CromIamApiService.scala        |  3 +-
 .../webservice/CromIamApiServiceSpec.scala    | 45 +++++++++++++++++++
 2 files changed, 47 insertions(+), 1 deletion(-)

diff --git a/CromIAM/src/main/scala/cromiam/webservice/CromIamApiService.scala b/CromIAM/src/main/scala/cromiam/webservice/CromIamApiService.scala
index d046694b0f9..47187fd8157 100644
--- a/CromIAM/src/main/scala/cromiam/webservice/CromIamApiService.scala
+++ b/CromIAM/src/main/scala/cromiam/webservice/CromIamApiService.scala
@@ -81,7 +81,7 @@ trait CromIamApiService
 
   val workflowRoutes: Route = queryGetRoute ~ queryPostRoute ~ workflowOutputsRoute ~ submitRoute ~
     workflowLogsRoute ~ abortRoute ~ metadataRoute ~ timingRoute ~ statusRoute ~ backendRoute ~ labelPatchRoute ~
-    callCacheDiffRoute ~ labelGetRoute ~ releaseHoldRoute
+    callCacheDiffRoute ~ labelGetRoute ~ releaseHoldRoute ~ costRoute
 
   val allRoutes: Route = handleExceptions(CromIamExceptionHandler)(workflowRoutes ~ engineRoutes ~ womtoolRoutes)
 
@@ -114,6 +114,7 @@ trait CromIamApiService
   def timingRoute: Route = workflowGetRouteWithId("timing")
   def statusRoute: Route = workflowGetRouteWithId("status")
   def labelGetRoute: Route = workflowGetRouteWithId(Labels)
+  def costRoute: Route = workflowGetRouteWithId("cost")
 
   def labelPatchRoute: Route =
     path("api" / "workflows" / Segment / Segment / Labels) { (_, workflowId) =>
diff --git a/CromIAM/src/test/scala/cromiam/webservice/CromIamApiServiceSpec.scala b/CromIAM/src/test/scala/cromiam/webservice/CromIamApiServiceSpec.scala
index e93acd51a2a..45948563d0d 100644
--- a/CromIAM/src/test/scala/cromiam/webservice/CromIamApiServiceSpec.scala
+++ b/CromIAM/src/test/scala/cromiam/webservice/CromIamApiServiceSpec.scala
@@ -520,4 +520,49 @@ class CromIamApiServiceSpec
       rejection shouldEqual MissingHeaderRejection("OIDC_CLAIM_user_id")
     }
   }
+
+  behavior of "Cost endpoint"
+  it should "return 200 for authorized user who has collection associated with root workflow" in {
+    Get(s"/api/workflows/$version/${cromwellClient.rootWorkflowIdWithCollection}/cost")
+      .withHeaders(goodAuthHeaders) ~> allRoutes ~> check {
+      status shouldBe OK
+      responseAs[String] shouldBe "Response from Cromwell"
+      contentType should be(ContentTypes.`text/plain(UTF-8)`)
+    }
+  }
+
+  it should "return 200 for authorized user who has collection associated with subworkflow" in {
+    Get(s"/api/workflows/$version/${cromwellClient.subworkflowId}/cost")
+      .withHeaders(goodAuthHeaders) ~> allRoutes ~> check {
+      status shouldBe OK
+      responseAs[String] shouldBe "Response from Cromwell"
+      contentType should be(ContentTypes.`text/plain(UTF-8)`)
+    }
+  }
+
+  it should "return 500 for authorized user who doesn't have collection associated with workflow" in {
+    Get(s"/api/workflows/$version/${cromwellClient.workflowIdWithoutCollection}/cost")
+      .withHeaders(goodAuthHeaders) ~> allRoutes ~> check {
+      status shouldBe InternalServerError
+      responseAs[
+        String
+      ] shouldBe s"CromIAM unexpected error: java.lang.IllegalArgumentException: Workflow ${cromwellClient.workflowIdWithoutCollection} has no associated collection"
+      contentType should be(ContentTypes.`text/plain(UTF-8)`)
+    }
+  }
+
+  it should "return SamDenialException for user who doesn't have view permissions" in {
+    Get(s"/api/workflows/$version/${cromwellClient.subworkflowId}/cost")
+      .withHeaders(badAuthHeaders) ~> allRoutes ~> check {
+      status shouldBe Forbidden
+      responseAs[String] shouldBe "Access Denied"
+      contentType should be(ContentTypes.`text/plain(UTF-8)`)
+    }
+  }
+
+  it should "reject request if it doesn't contain OIDC_CLAIM_user_id in header" in {
+    Get(s"/api/workflows/$version/${cromwellClient.rootWorkflowIdWithCollection}/cost") ~> allRoutes ~> check {
+      rejection shouldEqual MissingHeaderRejection("OIDC_CLAIM_user_id")
+    }
+  }
 }