46 lines
2.6 KiB
C#
46 lines
2.6 KiB
C#
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); }
|
|
}
|
|
}
|