1 files changed, 27 insertions, 1 deletions
diff --git a/tests/Agent.Core.Tests/McpHttpTests.cs b/tests/Agent.Core.Tests/McpHttpTests.cs
index 7e3f1cf..9edb133 100644
--- a/tests/Agent.Core.Tests/McpHttpTests.cs
+++ b/tests/Agent.Core.Tests/McpHttpTests.cs
@@ -6,6 +6,9 @@ using Agent.Core.Runtime;
 using Agent.Mcp;
 using ModelContextProtocol;
 using ModelContextProtocol.Client;
+using ModelContextProtocol.Extensions.Tasks;
+using ModelContextProtocol.Protocol;
+using System.Text.Json;
 
 namespace Agent.Core.Tests;
 
@@ -15,9 +18,10 @@ public sealed class McpHttpTests
     public async Task Loopback_streamable_http_works_and_get_sse_legacy_sse_are_rejected()
     {
         int port = FreePort();
+        await using var ollamaServer = new SharedFakeOllamaServer();
         using var stop = CancellationTokenSource.CreateLinkedTokenSource(TestContext.Current.CancellationToken);
         LocaleCatalog locale = LocaleCatalog.Load(Path.GetFullPath("config/locales"), "en");
-        var settings = new OllamaSettings(new Uri("http://127.0.0.1:1"), "unused", false, "0", new Dictionary<string, System.Text.Json.JsonElement>());
+        var settings = new OllamaSettings(ollamaServer.Endpoint, "unused", false, "0", new Dictionary<string, System.Text.Json.JsonElement>());
         Task server = McpHost.RunHttpAsync(locale, settings, new AgentRuntimeSettings("", "en", [], "sandbox"), new VisionLimits(), "127.0.0.1", port, stop.Token);
         using var http = new HttpClient();
         Uri endpoint = new($"http://127.0.0.1:{port}/mcp");
@@ -33,10 +37,32 @@ public sealed class McpHttpTests
         {
             IList<McpClientTool> tools = await client.ListToolsAsync(cancellationToken: TestContext.Current.CancellationToken);
             Assert.Equal(["agent_run", "agent_status"], tools.Select(static tool => tool.Name).Order(StringComparer.Ordinal));
+
+            ResultOrCreatedTask<CallToolResult> taskCall = await client.CallToolAsTaskAsync(new CallToolRequestParams
+            {
+                Name = "agent_run",
+                Arguments = new Dictionary<string, JsonElement> { ["request"] = JsonSerializer.SerializeToElement("task extension test") }
+            }, TestContext.Current.CancellationToken);
+            Assert.True(taskCall.IsTask);
+            CreateTaskResult created = taskCall.TaskCreated!;
+            Assert.True(Guid.TryParse(created.TaskId, out _));
+            GetTaskResult? task = null;
+            for (int attempt = 0; attempt < 400; attempt++)
+            {
+                task = await client.GetTaskAsync(created.TaskId, TestContext.Current.CancellationToken);
+                if (task.Status is McpTaskStatus.Completed or McpTaskStatus.Failed or McpTaskStatus.Cancelled) break;
+                await Task.Delay(25, TestContext.Current.CancellationToken);
+            }
+            Assert.NotNull(task);
+            Assert.Equal(McpTaskStatus.Completed, task.Status);
+            Assert.Contains(created.TaskId, JsonSerializer.Serialize(task), StringComparison.OrdinalIgnoreCase);
         }
 
         using HttpResponseMessage get = await http.GetAsync(endpoint, TestContext.Current.CancellationToken);
         Assert.Equal(HttpStatusCode.MethodNotAllowed, get.StatusCode);
+        Assert.False(get.Headers.Contains("Mcp-Session-Id"));
+        using HttpResponseMessage delete = await http.DeleteAsync(endpoint, TestContext.Current.CancellationToken);
+        Assert.Equal(HttpStatusCode.MethodNotAllowed, delete.StatusCode);
         using var hostileOrigin = new HttpRequestMessage(HttpMethod.Post, endpoint);
         hostileOrigin.Headers.Add("Origin", "https://example.invalid");
         using HttpResponseMessage forbidden = await http.SendAsync(hostileOrigin, TestContext.Current.CancellationToken);