chore: import upstream snapshot with attribution

dotnet/tests/Microsoft.Agents.AI.DurableTask.UnitTests/DurableAgentSessionTests.cs

@@ -0,0 +1,79 @@
+// Copyright (c) Microsoft. All rights reserved.
+
+using System.Text.Json;
+
+namespace Microsoft.Agents.AI.DurableTask.UnitTests;
+
+public sealed class DurableAgentSessionTests
+{
+    [Fact]
+    public void BuiltInSerialization()
+    {
+        AgentSessionId sessionId = AgentSessionId.WithRandomKey("test-agent");
+        DurableAgentSession session = new(sessionId);
+
+        JsonElement serializedSession = session.Serialize();
+
+        // Expected format: "{\"sessionId\":\"@dafx-test-agent@<random-key>\"}"
+        string expectedSerializedSession = $"{{\"sessionId\":\"@dafx-{sessionId.Name}@{sessionId.Key}\",\"stateBag\":{{}}}}";
+        Assert.Equal(expectedSerializedSession, serializedSession.ToString());
+
+        DurableAgentSession deserializedSession = DurableAgentSession.Deserialize(serializedSession);
+        Assert.Equal(sessionId, deserializedSession.SessionId);
+    }
+
+    [Fact]
+    public void STJSerialization()
+    {
+        AgentSessionId sessionId = AgentSessionId.WithRandomKey("test-agent");
+        AgentSession session = new DurableAgentSession(sessionId);
+
+        // Need to specify the type explicitly because STJ, unlike other serializers,
+        // does serialization based on the static type of the object, not the runtime type.
+        string serializedSession = JsonSerializer.Serialize(session, typeof(DurableAgentSession));
+
+        // Expected format: "{\"sessionId\":\"@dafx-test-agent@<random-key>\"}"
+        string expectedSerializedSession = $"{{\"sessionId\":\"@dafx-{sessionId.Name}@{sessionId.Key}\",\"stateBag\":{{}}}}";
+        Assert.Equal(expectedSerializedSession, serializedSession);
+
+        DurableAgentSession? deserializedSession = JsonSerializer.Deserialize<DurableAgentSession>(serializedSession);
+        Assert.NotNull(deserializedSession);
+        Assert.Equal(sessionId, deserializedSession.SessionId);
+    }
+
+    [Fact]
+    public void BuiltInSerialization_RoundTrip_PreservesStateBag()
+    {
+        // Arrange
+        AgentSessionId sessionId = AgentSessionId.WithRandomKey("test-agent");
+        DurableAgentSession session = new(sessionId);
+        session.StateBag.SetValue("durableKey", "durableValue");
+
+        // Act
+        JsonElement serializedSession = session.Serialize();
+        DurableAgentSession deserializedSession = DurableAgentSession.Deserialize(serializedSession);
+
+        // Assert
+        Assert.Equal(sessionId, deserializedSession.SessionId);
+        Assert.True(deserializedSession.StateBag.TryGetValue<string>("durableKey", out var value));
+        Assert.Equal("durableValue", value);
+    }
+
+    [Fact]
+    public void STJSerialization_RoundTrip_PreservesStateBag()
+    {
+        // Arrange
+        AgentSessionId sessionId = AgentSessionId.WithRandomKey("test-agent");
+        DurableAgentSession session = new(sessionId);
+        session.StateBag.SetValue("stjKey", "stjValue");
+
+        // Act
+        string serializedSession = JsonSerializer.Serialize(session, typeof(DurableAgentSession));
+        DurableAgentSession? deserializedSession = JsonSerializer.Deserialize<DurableAgentSession>(serializedSession);
+
+        // Assert
+        Assert.NotNull(deserializedSession);
+        Assert.True(deserializedSession.StateBag.TryGetValue<string>("stjKey", out var value));
+        Assert.Equal("stjValue", value);
+    }
+}