Add retry to LoadConsentForm for timeouts.

firebase · Oct 19, 2023 · 465b8e1 · 465b8e1
1 parent d8c5099
commit 465b8e1
Showing 1 changed file with 6 additions and 4 deletions.
diff --git 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<void> 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<void> future = consent_info_->LoadConsentFormLastResult();
 
-  WaitForCompletion(future, "LoadConsentForm");
+  EXPECT_TRUE(future == consent_info_->LoadConsentFormLastResult());
 
   EXPECT_EQ(consent_info_->GetConsentFormStatus(),
             firebase::gma::ump::kConsentFormStatusAvailable);