feat: add HTTP and MCP service

2026-09-07 19:07:16 +08:00
parent 6077abd1ad
commit d660c3b9e6
37 changed files with 2186 additions and 0 deletions
@@ -0,0 +1,45 @@
+using System.Net.Http.Headers;
+using System.Text.Json;
+using Microsoft.AspNetCore.TestHost;
+using Microsoft.Extensions.DependencyInjection;
+using WxAgent.Service;
+using Xunit;
+
+namespace WxAgent.Service.Tests;
+
+public sealed class SseTests
+{
+    private sealed class Backend : IAgentBackend
+    {
+        public IReadOnlyList<AgentCapability> Capabilities => [new("listener-events", true, true, true, true, false, "read", false, 30, null, [])];
+        public Task<object> StatusAsync(CancellationToken ct) => Task.FromResult<object>(new { ok = true });
+    }
+
+    [Fact]
+    public async Task EventStreamUsesRealSseFramesAndRedactsMessageContent()
+    {
+        var dir = Path.Combine(Path.GetTempPath(), Guid.NewGuid().ToString("N")); Directory.CreateDirectory(dir);
+        var token = new string('A', 43);
+        var options = new ServiceOptions { CredentialFile = Path.Combine(dir, "credentials.json"), DataDirectory = dir };
+        File.WriteAllText(options.CredentialFile, JsonSerializer.Serialize(new[] { new ServiceCredential("alice", ServiceOptions.HashToken(token), ["read"], []) }));
+        await using var app = ServiceHost.Build(options, new Backend(), b => b.WebHost.UseTestServer());
+        try
+        {
+            await app.StartAsync(); using var client = app.GetTestClient(); client.BaseAddress = new Uri("http://localhost:5088");
+            client.DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Bearer", token);
+            using var response = await client.GetAsync("/api/v1/events", HttpCompletionOption.ResponseHeadersRead);
+            Assert.Equal(200, (int)response.StatusCode);
+            var hub = app.Services.GetRequiredService<EventHub>();
+            hub.Publish("alice", new AgentEvent("evt-1", "ui-current", "test", "message", "Text", DateTimeOffset.UtcNow));
+            using var timeout = new CancellationTokenSource(TimeSpan.FromSeconds(5));
+            using var reader = new StreamReader(await response.Content.ReadAsStreamAsync(timeout.Token));
+            Assert.Equal("id: evt-1", await reader.ReadLineAsync(timeout.Token));
+            Assert.Equal("event: message", await reader.ReadLineAsync(timeout.Token));
+            var data = await reader.ReadLineAsync(timeout.Token);
+            Assert.StartsWith("data: ", data);
+            Assert.DoesNotContain("content", data, StringComparison.OrdinalIgnoreCase);
+            Assert.Equal(string.Empty, await reader.ReadLineAsync(timeout.Token));
+        }
+        finally { await app.StopAsync(); Microsoft.Data.Sqlite.SqliteConnection.ClearAllPools(); Directory.Delete(dir, true); }
+    }
+}