diff --git a/gma/integration_test/src/integration_test.cc b/gma/integration_test/src/integration_test.cc index e7bc556ced..d65577c79c 100644 --- a/gma/integration_test/src/integration_test.cc +++ b/gma/integration_test/src/integration_test.cc @@ -2660,12 +2660,14 @@ TEST_F(FirebaseGmaUmpTest, TestUmpLoadForm) { EXPECT_EQ(consent_info_->GetConsentFormStatus(), firebase::gma::ump::kConsentFormStatusAvailable); - // Load the form. - firebase::Future future = consent_info_->LoadConsentForm(); + // Load the form. Run this step with retry in case of timeout. + WaitForCompletion(RunWithRetry( + [&]() { return consent_info_->LoadConsentForm(); }, "LoadConsentForm", + firebase::gma::ump::kConsentFormErrorTimeout), "LoadconsentForm"); - EXPECT_TRUE(future == consent_info_->LoadConsentFormLastResult()); + firebase::Future future = consent_info_->LoadConsentFormLastResult(); - WaitForCompletion(future, "LoadConsentForm"); + EXPECT_TRUE(future == consent_info_->LoadConsentFormLastResult()); EXPECT_EQ(consent_info_->GetConsentFormStatus(), firebase::gma::ump::kConsentFormStatusAvailable);