61 lines
1.9 KiB
C#
61 lines
1.9 KiB
C#
// Copyright (c) .NET Foundation. All rights reserved.
|
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
|
|
|
using System;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.AspNetCore.Builder;
|
|
using Microsoft.AspNetCore.Hosting;
|
|
using Microsoft.AspNetCore.Http;
|
|
using Microsoft.Extensions.Configuration;
|
|
|
|
namespace Microsoft.AspNetCore.WebSockets.Test
|
|
{
|
|
public class KestrelWebSocketHelpers
|
|
{
|
|
public static IDisposable CreateServer(Func<HttpContext, Task> app)
|
|
{
|
|
Action<IApplicationBuilder> startup = builder =>
|
|
{
|
|
builder.Use(async (ct, next) =>
|
|
{
|
|
try
|
|
{
|
|
// Kestrel does not return proper error responses:
|
|
// https://github.com/aspnet/KestrelHttpServer/issues/43
|
|
await next();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
if (ct.Response.HasStarted)
|
|
{
|
|
throw;
|
|
}
|
|
|
|
ct.Response.StatusCode = 500;
|
|
ct.Response.Headers.Clear();
|
|
await ct.Response.WriteAsync(ex.ToString());
|
|
}
|
|
});
|
|
builder.UseWebSockets();
|
|
builder.Run(c => app(c));
|
|
};
|
|
|
|
var configBuilder = new ConfigurationBuilder();
|
|
configBuilder.AddInMemoryCollection();
|
|
var config = configBuilder.Build();
|
|
config["server.urls"] = "http://localhost:54321";
|
|
|
|
var host = new WebHostBuilder()
|
|
.UseConfiguration(config)
|
|
.UseKestrel()
|
|
.Configure(startup)
|
|
.Build();
|
|
|
|
host.Start();
|
|
|
|
return host;
|
|
}
|
|
}
|
|
}
|
|
|