.NET: Update OpenTelemetry Agent - Added Demo, AIAgentMetadata and Agent.GetService() (#356)

* OTEL Demo * updating telemetry sample * OTEL updates * WIP * Adding GetService and Agent Metadata for Telemetry * WIP * Add UT for OTEL System behavior * Address Unicode problem * Add logging * Adjust for new extensions * Change Logger to LoggerFactory for the extension method * Address AI Feedback * Simplify script and readme just for Azure OpenAI * Increase code converage * Address merge conflict * Another slnx fix * Address PR comments * Address PR feedback + Add UT * Added Hosting UT to the solution * Address PR comments * Address missing sensitivity tests * Remove unecessary override * Address PR comments
2026-06-16 21:04:09 +08:00 · 2025-08-15 15:41:21 +00:00
parent dea736e550
commit 531132057b
32 changed files with 4148 additions and 451 deletions
@@ -1,5 +1,6 @@
 // Copyright (c) Microsoft. All rights reserved.

+using System;
 using System.Collections.Generic;
 using System.Linq;
 using System.Text.Json;
@@ -97,4 +98,313 @@ public class InMemoryChatMessageStoreTests

        Assert.Empty(store);
    }
+
+    [Fact]
+    public async Task AddMessagesAsync_WithNullMessages_ThrowsArgumentNullExceptionAsync()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+
+        // Act & Assert
+        await Assert.ThrowsAsync<ArgumentNullException>(() => store.AddMessagesAsync(null!, CancellationToken.None));
+    }
+
+    [Fact]
+    public async Task DeserializeStateAsync_WithNullSerializedState_DoesNothingAsync()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        store.Add(new ChatMessage(ChatRole.User, "Existing message"));
+
+        // Act
+        await store.DeserializeStateAsync(null);
+
+        // Assert - Should still have the existing message
+        Assert.Single(store);
+        Assert.Equal("Existing message", store[0].Text);
+    }
+
+    [Fact]
+    public async Task DeserializeStateAsync_WithEmptyMessages_DoesNotAddMessagesAsync()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var stateWithEmptyMessages = JsonSerializer.SerializeToElement(new { Messages = new List<ChatMessage>() });
+
+        // Act
+        await store.DeserializeStateAsync(stateWithEmptyMessages);
+
+        // Assert
+        Assert.Empty(store);
+    }
+
+    [Fact]
+    public async Task DeserializeStateAsync_WithNullMessages_DoesNotAddMessagesAsync()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var stateWithNullMessages = JsonSerializer.SerializeToElement(new { Messages = (List<ChatMessage>?)null });
+
+        // Act
+        await store.DeserializeStateAsync(stateWithNullMessages);
+
+        // Assert
+        Assert.Empty(store);
+    }
+
+    [Fact]
+    public async Task DeserializeStateAsync_WithValidMessages_AddsMessagesAsync()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var messages = new List<ChatMessage>
+        {
+            new(ChatRole.User, "User message"),
+            new(ChatRole.Assistant, "Assistant message")
+        };
+        var state = new { Messages = messages };
+        var serializedState = JsonSerializer.SerializeToElement(state);
+
+        // Act
+        await store.DeserializeStateAsync(serializedState);
+
+        // Assert
+        Assert.Equal(2, store.Count);
+        Assert.Equal("User message", store[0].Text);
+        Assert.Equal("Assistant message", store[1].Text);
+    }
+
+    [Fact]
+    public void IndexerGet_ReturnsCorrectMessage()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var message1 = new ChatMessage(ChatRole.User, "First");
+        var message2 = new ChatMessage(ChatRole.Assistant, "Second");
+        store.Add(message1);
+        store.Add(message2);
+
+        // Act & Assert
+        Assert.Same(message1, store[0]);
+        Assert.Same(message2, store[1]);
+    }
+
+    [Fact]
+    public void IndexerSet_UpdatesMessage()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var originalMessage = new ChatMessage(ChatRole.User, "Original");
+        var newMessage = new ChatMessage(ChatRole.User, "Updated");
+        store.Add(originalMessage);
+
+        // Act
+        store[0] = newMessage;
+
+        // Assert
+        Assert.Same(newMessage, store[0]);
+        Assert.Equal("Updated", store[0].Text);
+    }
+
+    [Fact]
+    public void IsReadOnly_ReturnsFalse()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+
+        // Act & Assert
+        Assert.False(store.IsReadOnly);
+    }
+
+    [Fact]
+    public void IndexOf_ReturnsCorrectIndex()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var message1 = new ChatMessage(ChatRole.User, "First");
+        var message2 = new ChatMessage(ChatRole.Assistant, "Second");
+        var message3 = new ChatMessage(ChatRole.User, "Third");
+        store.Add(message1);
+        store.Add(message2);
+
+        // Act & Assert
+        Assert.Equal(0, store.IndexOf(message1));
+        Assert.Equal(1, store.IndexOf(message2));
+        Assert.Equal(-1, store.IndexOf(message3)); // Not in store
+    }
+
+    [Fact]
+    public void Insert_InsertsMessageAtCorrectIndex()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var message1 = new ChatMessage(ChatRole.User, "First");
+        var message2 = new ChatMessage(ChatRole.Assistant, "Second");
+        var insertMessage = new ChatMessage(ChatRole.User, "Inserted");
+        store.Add(message1);
+        store.Add(message2);
+
+        // Act
+        store.Insert(1, insertMessage);
+
+        // Assert
+        Assert.Equal(3, store.Count);
+        Assert.Same(message1, store[0]);
+        Assert.Same(insertMessage, store[1]);
+        Assert.Same(message2, store[2]);
+    }
+
+    [Fact]
+    public void RemoveAt_RemovesMessageAtIndex()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var message1 = new ChatMessage(ChatRole.User, "First");
+        var message2 = new ChatMessage(ChatRole.Assistant, "Second");
+        var message3 = new ChatMessage(ChatRole.User, "Third");
+        store.Add(message1);
+        store.Add(message2);
+        store.Add(message3);
+
+        // Act
+        store.RemoveAt(1);
+
+        // Assert
+        Assert.Equal(2, store.Count);
+        Assert.Same(message1, store[0]);
+        Assert.Same(message3, store[1]);
+    }
+
+    [Fact]
+    public void Clear_RemovesAllMessages()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        store.Add(new ChatMessage(ChatRole.User, "First"));
+        store.Add(new ChatMessage(ChatRole.Assistant, "Second"));
+
+        // Act
+        store.Clear();
+
+        // Assert
+        Assert.Empty(store);
+    }
+
+    [Fact]
+    public void Contains_ReturnsTrueForExistingMessage()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var message1 = new ChatMessage(ChatRole.User, "First");
+        var message2 = new ChatMessage(ChatRole.Assistant, "Second");
+        store.Add(message1);
+
+        // Act & Assert
+        Assert.Contains(message1, store);
+        Assert.DoesNotContain(message2, store);
+    }
+
+    [Fact]
+    public void CopyTo_CopiesMessagesToArray()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var message1 = new ChatMessage(ChatRole.User, "First");
+        var message2 = new ChatMessage(ChatRole.Assistant, "Second");
+        store.Add(message1);
+        store.Add(message2);
+        var array = new ChatMessage[4];
+
+        // Act
+        store.CopyTo(array, 1);
+
+        // Assert
+        Assert.Null(array[0]);
+        Assert.Same(message1, array[1]);
+        Assert.Same(message2, array[2]);
+        Assert.Null(array[3]);
+    }
+
+    [Fact]
+    public void Remove_RemovesSpecificMessage()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var message1 = new ChatMessage(ChatRole.User, "First");
+        var message2 = new ChatMessage(ChatRole.Assistant, "Second");
+        var message3 = new ChatMessage(ChatRole.User, "Third");
+        store.Add(message1);
+        store.Add(message2);
+        store.Add(message3);
+
+        // Act
+        var removed = store.Remove(message2);
+
+        // Assert
+        Assert.True(removed);
+        Assert.Equal(2, store.Count);
+        Assert.Same(message1, store[0]);
+        Assert.Same(message3, store[1]);
+    }
+
+    [Fact]
+    public void Remove_ReturnsFalseForNonExistentMessage()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var message1 = new ChatMessage(ChatRole.User, "First");
+        var message2 = new ChatMessage(ChatRole.Assistant, "Second");
+        store.Add(message1);
+
+        // Act
+        var removed = store.Remove(message2);
+
+        // Assert
+        Assert.False(removed);
+        Assert.Single(store);
+    }
+
+    [Fact]
+    public void GetEnumerator_Generic_ReturnsAllMessages()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var message1 = new ChatMessage(ChatRole.User, "First");
+        var message2 = new ChatMessage(ChatRole.Assistant, "Second");
+        store.Add(message1);
+        store.Add(message2);
+
+        // Act
+        var messages = new List<ChatMessage>();
+        messages.AddRange(store);
+
+        // Assert
+        Assert.Equal(2, messages.Count);
+        Assert.Same(message1, messages[0]);
+        Assert.Same(message2, messages[1]);
+    }
+
+    [Fact]
+    public void GetEnumerator_NonGeneric_ReturnsAllMessages()
+    {
+        // Arrange
+        var store = new InMemoryChatMessageStore();
+        var message1 = new ChatMessage(ChatRole.User, "First");
+        var message2 = new ChatMessage(ChatRole.Assistant, "Second");
+        store.Add(message1);
+        store.Add(message2);
+
+        // Act
+        var messages = new List<ChatMessage>();
+        var enumerator = ((System.Collections.IEnumerable)store).GetEnumerator();
+        while (enumerator.MoveNext())
+        {
+            messages.Add((ChatMessage)enumerator.Current);
+        }
+
+        // Assert
+        Assert.Equal(2, messages.Count);
+        Assert.Same(message1, messages[0]);
+        Assert.Same(message2, messages[1]);
+    }
 }