1 files changed, 72 insertions, 0 deletions
diff --git a/tests/Agent.Mcp.TestHost/Program.cs b/tests/Agent.Mcp.TestHost/Program.cs
new file mode 100644
index 0000000..3447beb
--- /dev/null
+++ b/tests/Agent.Mcp.TestHost/Program.cs
@@ -0,0 +1,72 @@
+using Agent.Core.Localization;
+using Agent.Core.Ollama;
+using Agent.Core.Runtime;
+using Agent.Mcp;
+using ModelContextProtocol.Server;
+using ModelContextProtocol.Protocol;
+using System.Reflection;
+using Microsoft.Extensions.DependencyInjection;
+using ModelContextProtocol;
+
+namespace Agent.Mcp.TestHost;
+
+public static class Marker;
+
+internal static class Program
+{
+    public static async Task Main(string[] args)
+    {
+        if (args.Length is not (2 or 4)) throw new ArgumentException("Expected <locale> <task-state> or http <port> <locale> <task-state>.");
+        bool http = args.Length == 4;
+        int offset = http ? 2 : 0;
+        LocaleCatalog locale = LocaleCatalog.Load(args[offset], "en");
+        var ollama = new OllamaSettings(new Uri("http://127.0.0.1:1"), "unused", false, "0", new Dictionary<string, System.Text.Json.JsonElement>());
+        var store = new McpTaskStoreSettings(args[offset + 1], TimeSpan.FromHours(24), 1000);
+        var runtime = new AgentRuntimeSettings("", "en", [], "sandbox");
+        McpServerTool[] fixtures =
+        [
+            McpServerTool.Create((string name) => $"Hello, {name}!", new McpServerToolCreateOptions { Name = "greet" }),
+            McpServerTool.Create(async (int seconds, CancellationToken cancellationToken) =>
+            {
+                await Task.Delay(TimeSpan.FromSeconds(seconds), cancellationToken).ConfigureAwait(false);
+                return $"Computed after {seconds} seconds";
+            }, new McpServerToolCreateOptions { Name = "slow_compute" }),
+            McpServerTool.Create(async () =>
+            {
+                await Task.Delay(1000).ConfigureAwait(false);
+                return new CallToolResult { IsError = true, Content = [new TextContentBlock { Text = "intentional tool error" }] };
+            }, new McpServerToolCreateOptions { Name = "failing_job" }),
+            McpServerTool.Create((Func<string>)(() => throw new InvalidOperationException("intentional protocol error")),
+                new McpServerToolCreateOptions { Name = "protocol_error_job" }),
+            .. typeof(ConformanceTools).GetMethods(BindingFlags.Public | BindingFlags.Instance | BindingFlags.DeclaredOnly)
+                .Where(static method => method.GetCustomAttribute<McpServerToolAttribute>() is not null)
+                .Select(method => McpServerTool.Create(method, new ConformanceTools()))
+        ];
+        var taskTools = new HashSet<string>(["agent_run", "slow_compute", "confirm_delete", "multi_input", "protocol_error_job"], StringComparer.Ordinal);
+        var requiredTaskTools = new HashSet<string>(["failing_job", "test_tool_with_task"], StringComparer.Ordinal);
+        if (http)
+            await McpHost.RunHttpAsync(locale, ollama, runtime, new VisionLimits(), "127.0.0.1", int.Parse(args[1], System.Globalization.CultureInfo.InvariantCulture), CancellationToken.None,
+                taskStoreSettings: store, additionalTools: fixtures, taskCapableTools: taskTools, requiredTaskTools: requiredTaskTools,
+                configureBuilder: ConfigureConformanceFilters).ConfigureAwait(false);
+        else
+            await McpHost.RunStdioAsync(locale, ollama, runtime, new VisionLimits(), CancellationToken.None, taskStoreSettings: store).ConfigureAwait(false);
+    }
+
+    private static void ConfigureConformanceFilters(IMcpServerBuilder builder)
+    {
+        builder.WithRequestFilters(filters => filters.AddCallToolFilter(next => async (context, cancellationToken) =>
+        {
+            if (context.Params?.Name == "test_missing_capability")
+            {
+#pragma warning disable MCP9005 // Frozen alpha conformance probes the legacy sampling capability shape.
+                throw new MissingRequiredClientCapabilityException(
+                    new ClientCapabilities { Sampling = new SamplingCapability() },
+                    "The conformance diagnostic requires the sampling capability.");
+#pragma warning restore MCP9005
+            }
+            if (context.Params?.Name == "protocol_error_job")
+                throw new McpProtocolException("intentional protocol error", McpErrorCode.InternalError);
+            return await next(context, cancellationToken).ConfigureAwait(false);
+        }));
+    }
+}