106ab8064fNo meaningful code changes, just lots of namespace shuffling to have fewer namespaces and to put things in better namespaces.
27 lines
1.1 KiB
C#
27 lines
1.1 KiB
C#
using ModelContextProtocol.Server;
|
|
using ModelContextProtocol.Tests.Utils;
|
|
using System.IO.Pipelines;
|
|
|
|
namespace ModelContextProtocol.Tests.Transport;
|
|
|
|
public class SseResponseStreamTransportTests(ITestOutputHelper testOutputHelper) : LoggedTest(testOutputHelper)
|
|
{
|
|
[Fact]
|
|
public async Task Can_Customize_MessageEndpoint()
|
|
{
|
|
var responsePipe = new Pipe();
|
|
|
|
await using var transport = new SseResponseStreamTransport(responsePipe.Writer.AsStream(), "/my-message-endpoint");
|
|
var transportRunTask = transport.RunAsync(TestContext.Current.CancellationToken);
|
|
|
|
using var responseStreamReader = new StreamReader(responsePipe.Reader.AsStream());
|
|
var firstLine = await responseStreamReader.ReadLineAsync(TestContext.Current.CancellationToken);
|
|
Assert.Equal("event: endpoint", firstLine);
|
|
|
|
var secondLine = await responseStreamReader.ReadLineAsync(TestContext.Current.CancellationToken);
|
|
Assert.Equal("data: /my-message-endpoint", secondLine);
|
|
|
|
responsePipe.Reader.Complete();
|
|
responsePipe.Writer.Complete();
|
|
}
|
|
}
|