add save state test

Author: bassmang

dotnet/test/Microsoft.AutoGen.Core.Tests/AgentTests.cs

@@ -145,4 +145,25 @@ await runtime.RegisterAgentFactoryAsync("MyAgent", (id, runtime) =>
 
         Assert.True(agent.ReceivedItems.Count == 1);
     }
+
+    [Fact]
+    public async Task AgentShouldSaveStateCorrectlyTest()
+    {
+        var runtime = new InProcessRuntime();
+        await runtime.StartAsync();
+
+        Logger<BaseAgent> logger = new(new LoggerFactory());
+        TestAgent agent = new TestAgent(new AgentId("TestType", "TestKey"), runtime, logger);
+
+        var state = await agent.SaveStateAsync();
+
+        // Ensure state is a dictionary
+        state.Should().NotBeNull();
+        state.Should().BeOfType<Dictionary<string, object>>();
+        state.Should().BeEmpty("Default SaveStateAsync should return an empty dictionary.");
+
+        // Add a sample value and verify it updates correctly
+        state["testKey"] = "testValue";
+        state.Should().ContainKey("testKey").WhoseValue.Should().Be("testValue");
+    }
 }