From 8f5a6ec4a1896e5432eb800bafdf4ad53d2b6c20 Mon Sep 17 00:00:00 2001
From: Roger Barreto <19890735+rogerbarreto@users.noreply.github.com>
Date: Thu, 16 Apr 2026 13:35:10 +0100
Subject: [PATCH] Fix AgentFrameworkResponseHandlerTests: implement session
 methods in test mock agents

---
 .../AgentFrameworkResponseHandlerTests.cs     | 26 ++++++++++---------
 1 file changed, 14 insertions(+), 12 deletions(-)

diff --git a/dotnet/tests/Microsoft.Agents.AI.Foundry.UnitTests/Hosting/AgentFrameworkResponseHandlerTests.cs b/dotnet/tests/Microsoft.Agents.AI.Foundry.UnitTests/Hosting/AgentFrameworkResponseHandlerTests.cs
index 277156c9bf..169dc75e22 100644
--- a/dotnet/tests/Microsoft.Agents.AI.Foundry.UnitTests/Hosting/AgentFrameworkResponseHandlerTests.cs
+++ b/dotnet/tests/Microsoft.Agents.AI.Foundry.UnitTests/Hosting/AgentFrameworkResponseHandlerTests.cs
@@ -699,19 +699,19 @@ public class AgentFrameworkResponseHandlerTests
 
         protected override ValueTask<AgentSession> CreateSessionCoreAsync(
             CancellationToken cancellationToken = default) =>
-            throw new NotImplementedException();
+            new(new SimpleAgentSession());
 
         protected override ValueTask<JsonElement> SerializeSessionCoreAsync(
             AgentSession session,
             JsonSerializerOptions? jsonSerializerOptions,
             CancellationToken cancellationToken = default) =>
-            throw new NotImplementedException();
+            new(JsonDocument.Parse("{}").RootElement);
 
         protected override ValueTask<AgentSession> DeserializeSessionCoreAsync(
             JsonElement serializedState,
             JsonSerializerOptions? jsonSerializerOptions,
             CancellationToken cancellationToken = default) =>
-            throw new NotImplementedException();
+            new(new SimpleAgentSession());
     }
 
     private sealed class ThrowingAgent(Exception exception) : AIAgent
@@ -732,19 +732,19 @@ public class AgentFrameworkResponseHandlerTests
 
         protected override ValueTask<AgentSession> CreateSessionCoreAsync(
             CancellationToken cancellationToken = default) =>
-            throw new NotImplementedException();
+            new(new SimpleAgentSession());
 
         protected override ValueTask<JsonElement> SerializeSessionCoreAsync(
             AgentSession session,
             JsonSerializerOptions? jsonSerializerOptions,
             CancellationToken cancellationToken = default) =>
-            throw new NotImplementedException();
+            new(JsonDocument.Parse("{}").RootElement);
 
         protected override ValueTask<AgentSession> DeserializeSessionCoreAsync(
             JsonElement serializedState,
             JsonSerializerOptions? jsonSerializerOptions,
             CancellationToken cancellationToken = default) =>
-            throw new NotImplementedException();
+            new(new SimpleAgentSession());
     }
 
     private sealed class CapturingAgent : AIAgent
@@ -776,19 +776,19 @@ public class AgentFrameworkResponseHandlerTests
 
         protected override ValueTask<AgentSession> CreateSessionCoreAsync(
             CancellationToken cancellationToken = default) =>
-            throw new NotImplementedException();
+            new(new SimpleAgentSession());
 
         protected override ValueTask<JsonElement> SerializeSessionCoreAsync(
             AgentSession session,
             JsonSerializerOptions? jsonSerializerOptions,
             CancellationToken cancellationToken = default) =>
-            throw new NotImplementedException();
+            new(JsonDocument.Parse("{}").RootElement);
 
         protected override ValueTask<AgentSession> DeserializeSessionCoreAsync(
             JsonElement serializedState,
             JsonSerializerOptions? jsonSerializerOptions,
             CancellationToken cancellationToken = default) =>
-            throw new NotImplementedException();
+            new(new SimpleAgentSession());
     }
 
     private sealed class CancellationCheckingAgent : AIAgent
@@ -813,18 +813,20 @@ public class AgentFrameworkResponseHandlerTests
 
         protected override ValueTask<AgentSession> CreateSessionCoreAsync(
             CancellationToken cancellationToken = default) =>
-            throw new NotImplementedException();
+            new(new SimpleAgentSession());
 
         protected override ValueTask<JsonElement> SerializeSessionCoreAsync(
             AgentSession session,
             JsonSerializerOptions? jsonSerializerOptions,
             CancellationToken cancellationToken = default) =>
-            throw new NotImplementedException();
+            new(JsonDocument.Parse("{}").RootElement);
 
         protected override ValueTask<AgentSession> DeserializeSessionCoreAsync(
             JsonElement serializedState,
             JsonSerializerOptions? jsonSerializerOptions,
             CancellationToken cancellationToken = default) =>
-            throw new NotImplementedException();
+            new(new SimpleAgentSession());
     }
+
+    private sealed class SimpleAgentSession : AgentSession { }
 }