Fix reserved expansion (e.g. "{+var}") when matching resources (#1142)
This commit is contained in:
parent
6ca7cde13c
commit
81664c3fce
1 changed files with 967 additions and 50 deletions
|
|
@ -13,36 +13,55 @@ public abstract class ClientServerTestBase : LoggedTest, IAsyncDisposable
|
|||
{
|
||||
private readonly Pipe _clientToServerPipe = new();
|
||||
private readonly Pipe _serverToClientPipe = new();
|
||||
private readonly IMcpServerBuilder _builder;
|
||||
private readonly CancellationTokenSource _cts;
|
||||
private readonly Task _serverTask;
|
||||
private readonly CancellationTokenSource _cts = CancellationTokenSource.CreateLinkedTokenSource(TestContext.Current.CancellationToken);
|
||||
private Task _serverTask = Task.CompletedTask;
|
||||
|
||||
public ClientServerTestBase(ITestOutputHelper testOutputHelper)
|
||||
public ClientServerTestBase(ITestOutputHelper testOutputHelper, bool startServer = true)
|
||||
: base(testOutputHelper)
|
||||
{
|
||||
ServiceCollection sc = new();
|
||||
sc.AddLogging();
|
||||
sc.AddSingleton(XunitLoggerProvider);
|
||||
sc.AddSingleton<ILoggerProvider>(MockLoggerProvider);
|
||||
_builder = sc
|
||||
ServiceCollection.AddLogging();
|
||||
ServiceCollection.AddSingleton(XunitLoggerProvider);
|
||||
ServiceCollection.AddSingleton<ILoggerProvider>(MockLoggerProvider);
|
||||
McpServerBuilder = ServiceCollection
|
||||
.AddMcpServer()
|
||||
.WithStreamServerTransport(_clientToServerPipe.Reader.AsStream(), _serverToClientPipe.Writer.AsStream());
|
||||
ConfigureServices(sc, _builder);
|
||||
ServiceProvider = sc.BuildServiceProvider(validateScopes: true);
|
||||
|
||||
_cts = CancellationTokenSource.CreateLinkedTokenSource(TestContext.Current.CancellationToken);
|
||||
Server = ServiceProvider.GetRequiredService<McpServer>();
|
||||
_serverTask = Server.RunAsync(_cts.Token);
|
||||
ConfigureServices(ServiceCollection, McpServerBuilder);
|
||||
|
||||
if (startServer)
|
||||
{
|
||||
StartServer();
|
||||
}
|
||||
}
|
||||
|
||||
protected McpServer Server { get; }
|
||||
protected ServiceCollection ServiceCollection { get; } = [];
|
||||
|
||||
protected IServiceProvider ServiceProvider { get; }
|
||||
protected IMcpServerBuilder McpServerBuilder { get; }
|
||||
|
||||
protected McpServer Server
|
||||
{
|
||||
get => field ?? throw new InvalidOperationException("You must call StartServer first.");
|
||||
private set => field = value;
|
||||
}
|
||||
|
||||
protected ServiceProvider ServiceProvider
|
||||
{
|
||||
get => field ?? throw new InvalidOperationException("You must call StartServer first.");
|
||||
private set => field = value;
|
||||
}
|
||||
|
||||
protected virtual void ConfigureServices(ServiceCollection services, IMcpServerBuilder mcpServerBuilder)
|
||||
{
|
||||
}
|
||||
|
||||
protected McpServer StartServer()
|
||||
{
|
||||
ServiceProvider = ServiceCollection.BuildServiceProvider(validateScopes: true);
|
||||
Server = ServiceProvider.GetRequiredService<McpServer>();
|
||||
_serverTask = Server.RunAsync(_cts.Token);
|
||||
return Server;
|
||||
}
|
||||
|
||||
public async ValueTask DisposeAsync()
|
||||
{
|
||||
await _cts.CancelAsync();
|
||||
|
|
|
|||
Loading…
Reference in a new issue