JsonRpcServerTestBase.cs

using System;
using System.IO.Pipelines;
using System.Threading.Tasks;
using MediatR;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;

namespace OmniSharp.Extensions.JsonRpc.Testing
{
    /// <summary>
    /// This is a test class that is designed to allow you configure an in memory lsp client and server to do testing of handlers or behaviors.
    /// </summary>
    public abstract class JsonRpcServerTestBase : JsonRpcTestBase
    {
        private JsonRpcServer _client= null!;
        private JsonRpcServer _server = null!;

        public JsonRpcServerTestBase(JsonRpcTestOptions testOptions) : base(testOptions)
        {
        }

        protected virtual void ConfigureClientInputOutput(PipeReader inMemoryReader, PipeWriter inMemoryWriter, JsonRpcServerOptions options) =>
            options.WithInput(inMemoryReader).WithOutput(inMemoryWriter);

        protected virtual void ConfigureServerInputOutput(PipeReader inMemoryReader, PipeWriter inMemoryWriter, JsonRpcServerOptions options) =>
            options.WithInput(inMemoryReader).WithOutput(inMemoryWriter);

        protected virtual async Task<(JsonRpcServer client, JsonRpcServer server)> Initialize(
            Action<JsonRpcServerOptions> clientOptionsAction,
            Action<JsonRpcServerOptions> serverOptionsAction
        )
        {
            var clientPipe = new Pipe(TestOptions.DefaultPipeOptions);
            var serverPipe = new Pipe(TestOptions.DefaultPipeOptions);

            var clientTask = JsonRpcServer.From(
                options => {
                    options
                       .WithLoggerFactory(TestOptions.ClientLoggerFactory)
                       .WithServices(
                            services => services
                                       .AddTransient(typeof(IPipelineBehavior<,>), typeof(SettlePipeline<,>))
                                       .AddSingleton(ClientEvents as IRequestSettler)
                                       .AddLogging(
                                            x => {
                                                x.SetMinimumLevel(LogLevel.Trace);
                                            }
                                        )
                        );
                    ConfigureClientInputOutput(serverPipe.Reader, clientPipe.Writer, options);
                    clientOptionsAction(options);
                }, CancellationToken
            );

            var serverTask = JsonRpcServer.From(
                options => {
                    options
                       .WithServices(
                            services => services
                                       .AddTransient(typeof(IPipelineBehavior<,>), typeof(SettlePipeline<,>))
                                       .AddSingleton(ServerEvents as IRequestSettler)
                                       .AddLogging(
                                            x => {
                                                x.SetMinimumLevel(LogLevel.Trace);
                                                x.Services.AddSingleton(TestOptions.ServerLoggerFactory);
                                            }
                                        )
                        );
                    ConfigureServerInputOutput(clientPipe.Reader, serverPipe.Writer, options);
                    serverOptionsAction(options);
                }, CancellationToken
            );

            await Task.WhenAll(clientTask, serverTask).ConfigureAwait(false);
#pragma warning disable VSTHRD103
            _client = clientTask.Result;
            _server = serverTask.Result;
#pragma warning restore VSTHRD103

            Disposable.Add(_client);
            Disposable.Add(_server);

            return ( _client, _server );
        }
    }
}