24c706b4ff* Add tests for tool generation * Add couple of MCPServer tests * Fix merge issues * More tests for McpServer * Some more tests * Fix merge issues
22 lines
841 B
C#
22 lines
841 B
C#
namespace McpDotNet.Tests.Utils;
|
|
|
|
public class MockHttpHandler : HttpMessageHandler
|
|
{
|
|
public Func<HttpRequestMessage, Task<HttpResponseMessage>>? RequestHandler { get; set; }
|
|
|
|
protected async override Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
|
|
{
|
|
if (RequestHandler == null)
|
|
throw new InvalidOperationException($"No {nameof(RequestHandler)} was set! Please set handler first and make request afterwards.");
|
|
|
|
if (cancellationToken.IsCancellationRequested)
|
|
throw new OperationCanceledException(cancellationToken);
|
|
|
|
var result = await RequestHandler.Invoke(request);
|
|
|
|
if (cancellationToken.IsCancellationRequested)
|
|
throw new OperationCanceledException(cancellationToken);
|
|
|
|
return result;
|
|
}
|
|
}
|