test

dotnet/tests/Microsoft.Agents.AI.Workflows.Declarative.IntegrationTests/AzureAgentProviderTest.cs

@@ -0,0 +1,43 @@
+// Copyright (c) Microsoft. All rights reserved.
+
+using System.Linq;
+using System.Threading.Tasks;
+using Azure.Identity;
+using Microsoft.Agents.AI.Workflows.Declarative.IntegrationTests.Framework;
+using Microsoft.Extensions.AI;
+using Xunit.Abstractions;
+
+namespace Microsoft.Agents.AI.Workflows.Declarative.IntegrationTests;
+
+public sealed class AzureAgentProviderTest(ITestOutputHelper output) : IntegrationTest(output)
+{
+    [Fact]
+    public async Task ConversationTestAsync()
+    {
+        // Arrange
+        AzureAgentProvider provider = new(this.TestEndpoint, new AzureCliCredential());
+        // Act
+        string conversationId = await provider.CreateConversationAsync();
+        // Assert
+        Assert.NotEmpty(conversationId);
+
+        // Arrange & Act
+        for (int index = 0; index < 3; ++index)
+        {
+            await provider.CreateMessageAsync(conversationId, new ChatMessage(ChatRole.User, $"Message #{index * 2}"));
+            await provider.CreateMessageAsync(conversationId, new ChatMessage(ChatRole.Assistant, $"Message #{(index * 2) + 1}"));
+        }
+
+        // Act
+        ChatMessage[] messages = await provider.GetMessagesAsync(conversationId).ToArrayAsync();
+        // Assert
+        Assert.Equal(6, messages.Length);
+        Assert.NotNull(messages[3].MessageId);
+
+        // Act
+        ChatMessage message = await provider.GetMessageAsync(conversationId, messages[3].MessageId!);
+        // Assert
+        Assert.NotNull(message);
+        Assert.Equal(messages[3].Text, message.Text);
+    }
+}