test(bigtable): Add retries while getting app profile (#8751)

googleapis · Oct 25, 2023 · 7bb630b · 7bb630b
1 parent b1409e3
commit 7bb630b
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/bigtable/integration_test.go b/bigtable/integration_test.go
@@ -2862,11 +2862,14 @@ func TestIntegration_InstanceAdminClient_AppProfile(t *testing.T) {
 			continue
 		}
 
-		got, _ := iAdminClient.GetAppProfile(ctx, adminClient.instance, profileID)
+		// Retry to see if the update has been completed
+		testutil.Retry(t, 10, 10*time.Second, func(r *testutil.R) {
+			got, _ := iAdminClient.GetAppProfile(ctx, adminClient.instance, profileID)
 
-		if !proto.Equal(got, test.want) {
-			t.Fatalf("%s : got profile : %v, want profile: %v", test.desc, gotProfile, test.want)
-		}
+			if !proto.Equal(got, test.want) {
+				r.Errorf("%s : got profile : %v, want profile: %v", test.desc, gotProfile, test.want)
+			}
+		})
 
 	}