diff --git a/firebase-dataconnect/src/androidTest/kotlin/com/google/firebase/dataconnect/AuthIntegrationTest.kt b/firebase-dataconnect/src/androidTest/kotlin/com/google/firebase/dataconnect/AuthIntegrationTest.kt index ba314d0f9db..a53463ae75a 100644 --- a/firebase-dataconnect/src/androidTest/kotlin/com/google/firebase/dataconnect/AuthIntegrationTest.kt +++ b/firebase-dataconnect/src/androidTest/kotlin/com/google/firebase/dataconnect/AuthIntegrationTest.kt @@ -127,6 +127,7 @@ class AuthIntegrationTest : DataConnectIntegrationTestBase() { grpcServer.metadatas.map { it.get(firebaseAuthTokenHeader) }.toCollection(authTokens) } val dataConnect = dataConnectFactory.newInstance(auth.app, grpcServer) + (dataConnect as FirebaseDataConnectInternal).awaitAuthReady() val operationName = Arb.dataConnect.operationName().next(rs) val queryRef = dataConnect.query(operationName, Unit, serializer(), serializer()) @@ -155,6 +156,7 @@ class AuthIntegrationTest : DataConnectIntegrationTestBase() { grpcServer.metadatas.map { it.get(firebaseAuthTokenHeader) }.toCollection(authTokens) } val dataConnect = dataConnectFactory.newInstance(auth.app, grpcServer) + (dataConnect as FirebaseDataConnectInternal).awaitAuthReady() val operationName = Arb.dataConnect.operationName().next(rs) val mutationRef = dataConnect.mutation(operationName, Unit, serializer(), serializer())