63 lines
1.8 KiB
C#
63 lines
1.8 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;
|
|
using Microsoft.AspNetCore.Http;
|
|
using Microsoft.AspNetCore.Server.Kestrel;
|
|
using Microsoft.AspNetCore.Server.Kestrel.Http;
|
|
|
|
namespace Microsoft.AspNetCore.Server.KestrelTests
|
|
{
|
|
/// <summary>
|
|
/// Summary description for TestServer
|
|
/// </summary>
|
|
public class TestServer : IDisposable
|
|
{
|
|
private static int _nextPort = 9001;
|
|
|
|
private KestrelEngine _engine;
|
|
private IDisposable _server;
|
|
ServerAddress _address;
|
|
|
|
public TestServer(RequestDelegate app)
|
|
: this(app, new TestServiceContext())
|
|
{
|
|
}
|
|
|
|
public TestServer(RequestDelegate app, ServiceContext context)
|
|
: this(app, context, $"http://localhost:{GetNextPort()}/")
|
|
{
|
|
}
|
|
|
|
public int Port => _address.Port;
|
|
|
|
public TestServer(RequestDelegate app, ServiceContext context, string serverAddress)
|
|
{
|
|
Create(app, context, serverAddress);
|
|
}
|
|
|
|
public void Create(RequestDelegate app, ServiceContext context, string serverAddress)
|
|
{
|
|
context.FrameFactory = connectionContext =>
|
|
{
|
|
return new Frame<HttpContext>(new DummyApplication(app), connectionContext);
|
|
};
|
|
_engine = new KestrelEngine(context);
|
|
_engine.Start(1);
|
|
_address = ServerAddress.FromUrl(serverAddress);
|
|
_server = _engine.CreateServer(_address);
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
_server.Dispose();
|
|
_engine.Dispose();
|
|
}
|
|
|
|
public static int GetNextPort()
|
|
{
|
|
return Interlocked.Increment(ref _nextPort);
|
|
}
|
|
}
|
|
} |