fix: Run test methods in InProcessLifeOfAReportIntegrationTest (#2005)

Closes #2004 --------- Co-authored-by: Tristan Vuong <[email protected]>
world-federation-of-advertisers · Jan 24, 2025 · 162f9ad · 162f9ad
1 parent 78997f8
commit 162f9ad
Show file tree

Hide file tree

Showing 3 changed files with 125 additions and 193 deletions.
diff --git a/src/main/kotlin/org/wfanet/measurement/integration/common/InProcessCmmsComponents.kt b/src/main/kotlin/org/wfanet/measurement/integration/common/InProcessCmmsComponents.kt
@@ -19,6 +19,7 @@ package org.wfanet.measurement.integration.common
 import com.google.protobuf.ByteString
 import com.google.protobuf.TypeRegistry
 import java.security.cert.X509Certificate
+import java.util.concurrent.TimeUnit
 import kotlinx.coroutines.runBlocking
 import org.junit.rules.TestRule
 import org.junit.runner.Description
@@ -153,6 +154,8 @@ class InProcessCmmsComponents(
   }
   private val publicAccountsClient by lazy {
     AccountsGrpcKt.AccountsCoroutineStub(kingdom.publicApiChannel)
+      // Fail faster.
+      .withDeadlineAfter(5L, TimeUnit.SECONDS)
   }
   private val publicApiKeysClient by lazy {
     ApiKeysGrpcKt.ApiKeysCoroutineStub(kingdom.publicApiChannel)
@@ -308,6 +311,12 @@ class InProcessCmmsComponents(
     populationRequisitionFulfiller.stop()
   }
 
+  fun stopDaemons() {
+    stopEdpSimulators()
+    stopDuchyDaemons()
+    stopPopulationRequisitionFulfillerDaemon()
+  }
+
   override fun apply(statement: Statement, description: Description): Statement {
     return ruleChain.apply(statement, description)
   }