fix mock class

broadinstitute · Aug 21, 2020 · 93d8de4 · 93d8de4
1 parent dc42d94
commit 93d8de4
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/google/src/test/scala/org/broadinstitute/dsde/workbench/google/mock/MockGoogleIamDAO.scala b/google/src/test/scala/org/broadinstitute/dsde/workbench/google/mock/MockGoogleIamDAO.scala
@@ -79,6 +79,15 @@ class MockGoogleIamDAO extends GoogleIamDAO {
       Future.failed(new Exception(s"Unknown service account $userEmail"))
     }
 
+  override def addWorkloadIdentityUserRoleForUser(googleProject: GoogleProject,
+                                                  serviceAccountEmail: WorkbenchEmail,
+                                                  email: WorkbenchEmail): Future[Unit] =
+    if (serviceAccounts.contains(serviceAccountEmail)) {
+      Future.successful(())
+    } else {
+      Future.failed(new Exception(s"Unknown service account $email"))
+    }
+
   override def createServiceAccountKey(serviceAccountProject: GoogleProject,
                                        serviceAccountEmail: WorkbenchEmail): Future[ServiceAccountKey] = {
     val keyId = ServiceAccountKeyId(UUID.randomUUID().toString)