Add CosmosDB trigger test for Functions

dotnet · Jan 28, 2025 · 0efa174 · 0efa174
1 parent 59ba067
commit 0efa174
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 8 deletions.
diff --git a/playground/AzureFunctionsEndToEnd/AzureFunctionsEndToEnd.ApiService/Program.cs b/playground/AzureFunctionsEndToEnd/AzureFunctionsEndToEnd.ApiService/Program.cs
@@ -4,8 +4,8 @@
 using Azure.Messaging.EventHubs.Producer;
 #if !SKIP_UNSTABLE_EMULATORS
 using Azure.Messaging.ServiceBus;
-using Microsoft.Azure.Cosmos;
 #endif
+using Microsoft.Azure.Cosmos;
 using Azure.Storage.Blobs;
 using Azure.Storage.Queues;
 using Newtonsoft.Json;
@@ -17,9 +17,9 @@
 builder.AddAzureQueueClient("queue");
 builder.AddAzureBlobClient("blob");
 builder.AddAzureEventHubProducerClient("eventhubs", static settings => settings.EventHubName = "myhub");
+builder.AddAzureCosmosClient("cosmosdb");
 #if !SKIP_UNSTABLE_EMULATORS
 builder.AddAzureServiceBusClient("messaging");
-builder.AddAzureCosmosClient("cosmosdb");
 #endif
 
 var app = builder.Build();
@@ -67,6 +67,7 @@ static string RandomString(int length)
     await sender.SendMessageAsync(message, cancellationToken);
     return Results.Ok("Message sent to Azure Service Bus.");
 });
+#endif
 
 app.MapGet("/publish/cosmosdb", async (CosmosClient cosmosClient) =>
 {
@@ -78,7 +79,6 @@ static string RandomString(int length)
 
     return Results.Ok("Document created in Azure Cosmos DB.");
 });
-#endif
 
 app.MapGet("/", async (HttpClient client) =>
 {

diff --git a/playground/AzureFunctionsEndToEnd/AzureFunctionsEndToEnd.AppHost/Program.cs b/playground/AzureFunctionsEndToEnd/AzureFunctionsEndToEnd.AppHost/Program.cs
@@ -6,28 +6,28 @@
 var eventHubs = builder.AddAzureEventHubs("eventhubs").RunAsEmulator().WithHub("myhub");
 #if !SKIP_UNSTABLE_EMULATORS
 var serviceBus = builder.AddAzureServiceBus("messaging").RunAsEmulator().WithQueue("myqueue");
+#endif
 var cosmosDb = builder.AddAzureCosmosDB("cosmosdb")
     .RunAsEmulator()
     .WithDatabase("mydatabase", (database)
         => database.Containers.AddRange([new("mycontainer", "/id")]));
-#endif
 
 var funcApp = builder.AddAzureFunctionsProject<Projects.AzureFunctionsEndToEnd_Functions>("funcapp")
     .WithExternalHttpEndpoints()
     .WithReference(eventHubs).WaitFor(eventHubs)
 #if !SKIP_UNSTABLE_EMULATORS
     .WithReference(serviceBus).WaitFor(serviceBus)
-    .WithReference(cosmosDb).WaitFor(cosmosDb)
 #endif
+    .WithReference(cosmosDb).WaitFor(cosmosDb)
     .WithReference(blob)
     .WithReference(queue);
 
 builder.AddProject<Projects.AzureFunctionsEndToEnd_ApiService>("apiservice")
     .WithReference(eventHubs).WaitFor(eventHubs)
 #if !SKIP_UNSTABLE_EMULATORS
     .WithReference(serviceBus).WaitFor(serviceBus)
-    .WithReference(cosmosDb).WaitFor(cosmosDb)
 #endif
+    .WithReference(cosmosDb).WaitFor(cosmosDb)
     .WithReference(queue)
     .WithReference(blob)
     .WithReference(funcApp);

diff --git a/playground/AzureFunctionsEndToEnd/AzureFunctionsEndToEnd.Functions/MyCosmosDbTrigger.cs b/playground/AzureFunctionsEndToEnd/AzureFunctionsEndToEnd.Functions/MyCosmosDbTrigger.cs
@@ -1,4 +1,3 @@
-#if !SKIP_UNSTABLE_EMULATORS
 using System.Reflection.Metadata;
 using Microsoft.Azure.Functions.Worker;
 using Microsoft.Extensions.Logging;
@@ -17,4 +16,3 @@ public void Run([CosmosDBTrigger(
         logger.LogInformation("C# cosmosdb trigger function processed: {Count} messages", input.Count);
     }
 }
-#endif
diff --git a/tests/Aspire.Playground.Tests/ProjectSpecificTests.cs b/tests/Aspire.Playground.Tests/ProjectSpecificTests.cs
@@ -125,6 +125,15 @@ await WaitForAllTextAsync(app,
             timeoutSecs: 160);
 #endif
 
+        // Assert that CosmosDB triggers work correctly
+        await apiServiceClient.GetAsync("/publish/cosmosdb");
+        await WaitForAllTextAsync(app,
+            [
+                "Executed 'Functions.MyCosmosDbTrigger'"
+            ],
+            resourceName: "funcapp",
+            timeoutSecs: 160);
+
         // TODO: The following line is commented out because the test fails due to an erroneous log in the Functions App
         // resource that happens after the Functions host has been built. The error log shows up after the Functions
         // worker extension has been built and before the host has launched.