diff --git a/firebase-ai/src/test/java/com/google/firebase/ai/util/tests.kt b/firebase-ai/src/test/java/com/google/firebase/ai/util/tests.kt index 4c325faa884..dff254b7adf 100644 --- a/firebase-ai/src/test/java/com/google/firebase/ai/util/tests.kt +++ b/firebase-ai/src/test/java/com/google/firebase/ai/util/tests.kt @@ -210,15 +210,16 @@ internal fun goldenUnaryFile( httpStatusCode: HttpStatusCode = HttpStatusCode.OK, backend: GenerativeBackend = GenerativeBackend.vertexAI(), block: CommonTest, -) = +) = doBlocking { commonTest(httpStatusCode, backend = backend) { val goldenFile = loadGoldenFile(name) val message = goldenFile.readText() - channel.send(message.toByteArray()) + launch { channel.send(message.toByteArray()) } block() } +} /** * A variant of [goldenUnaryFile] for vertexai tests Loads the *Golden File* and automatically diff --git a/firebase-vertexai/src/test/java/com/google/firebase/vertexai/util/tests.kt b/firebase-vertexai/src/test/java/com/google/firebase/vertexai/util/tests.kt index 6af4339dbb4..3640821771e 100644 --- a/firebase-vertexai/src/test/java/com/google/firebase/vertexai/util/tests.kt +++ b/firebase-vertexai/src/test/java/com/google/firebase/vertexai/util/tests.kt @@ -189,15 +189,16 @@ internal fun goldenUnaryFile( name: String, httpStatusCode: HttpStatusCode = HttpStatusCode.OK, block: CommonTest, -) = +) = doBlocking { commonTest(httpStatusCode) { val goldenFile = loadGoldenFile(name) val message = goldenFile.readText() - channel.send(message.toByteArray()) + launch { channel.send(message.toByteArray()) } block() } +} /** * A variant of [goldenUnaryFile] for vertexai tests Loads the *Golden File* and automatically