Add wait for healthy in cosmos tests.

dotnet · Jan 11, 2025 · 884b995 · 884b995
1 parent 968fd3e
commit 884b995
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/Aspire.Hosting.Azure.Tests/AzureCosmosDBEmulatorFunctionalTests.cs b/tests/Aspire.Hosting.Azure.Tests/AzureCosmosDBEmulatorFunctionalTests.cs
@@ -93,6 +93,9 @@ public async Task VerifyCosmosResource(bool usePreview)
 
         await app.StartAsync();
 
+        var rns = app.Services.GetRequiredService<ResourceNotificationService>();
+        await rns.WaitForResourceHealthyAsync(db.Resource.Name, cts.Token);
+
         var hb = Host.CreateApplicationBuilder();
         hb.Configuration[$"ConnectionStrings:{db.Resource.Name}"] = await db.Resource.ConnectionStringExpression.GetValueAsync(default);
         hb.AddAzureCosmosClient(db.Resource.Name);
@@ -169,6 +172,9 @@ public async Task WithDataVolumeShouldPersistStateBetweenUsages(bool usePreview)
         {
             await app.StartAsync();
 
+            var rns = app.Services.GetRequiredService<ResourceNotificationService>();
+            await rns.WaitForResourceHealthyAsync(db1.Resource.Name, cts.Token);
+
             try
             {
                 var hb = Host.CreateApplicationBuilder();
@@ -214,6 +220,9 @@ await pipeline.ExecuteAsync(async token =>
         {
             await app.StartAsync();
 
+            var rns = app.Services.GetRequiredService<ResourceNotificationService>();
+            await rns.WaitForResourceHealthyAsync(db2.Resource.Name, cts.Token);
+
             try
             {
                 var hb = Host.CreateApplicationBuilder();