aa75f6f37e* Enable netfx testing. * Skip impacted tests and re-enable netfx testing. * Remove redundant comment. * Remove unnecessary project configuration. * Simplify test server copying.
35 lines
1.1 KiB
C#
35 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(
|
|
#if NET
|
|
TestContext.Current.CancellationToken
|
|
#endif
|
|
);
|
|
Assert.Equal("event: endpoint", firstLine);
|
|
|
|
var secondLine = await responseStreamReader.ReadLineAsync(
|
|
#if NET
|
|
TestContext.Current.CancellationToken
|
|
#endif
|
|
);
|
|
Assert.Equal("data: /my-message-endpoint", secondLine);
|
|
|
|
responsePipe.Reader.Complete();
|
|
responsePipe.Writer.Complete();
|
|
}
|
|
}
|