129 lines
3.5 KiB
C#
129 lines
3.5 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.IO.Pipelines;
|
|
using System.Text;
|
|
using System.Net;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.AspNetCore.Server.Kestrel;
|
|
using Microsoft.AspNetCore.Server.Kestrel.Internal;
|
|
using Microsoft.AspNetCore.Server.Kestrel.Internal.Http;
|
|
using Microsoft.AspNetCore.Testing;
|
|
using Microsoft.Extensions.Internal;
|
|
using MemoryPool = Microsoft.AspNetCore.Server.Kestrel.Internal.Infrastructure.MemoryPool;
|
|
|
|
namespace Microsoft.AspNetCore.Server.KestrelTests
|
|
{
|
|
class TestInput : IConnectionControl, IFrameControl, IDisposable
|
|
{
|
|
private MemoryPool _memoryPool;
|
|
private PipeFactory _pipelineFactory;
|
|
|
|
public TestInput()
|
|
{
|
|
var trace = new KestrelTrace(new TestKestrelTrace());
|
|
var serviceContext = new ServiceContext
|
|
{
|
|
DateHeaderValueManager = new DateHeaderValueManager(),
|
|
ServerOptions = new KestrelServerOptions(),
|
|
HttpParser = new KestrelHttpParser(trace),
|
|
};
|
|
var listenerContext = new ListenerContext(serviceContext)
|
|
{
|
|
ListenOptions = new ListenOptions(new IPEndPoint(IPAddress.Loopback, 5000))
|
|
};
|
|
var connectionContext = new ConnectionContext(listenerContext)
|
|
{
|
|
ConnectionControl = this
|
|
};
|
|
var context = new Frame<object>(null, connectionContext);
|
|
FrameContext = context;
|
|
FrameContext.FrameControl = this;
|
|
FrameContext.ConnectionContext.ListenerContext.ServiceContext.Log = trace;
|
|
|
|
_memoryPool = new MemoryPool();
|
|
_pipelineFactory = new PipeFactory();
|
|
FrameContext.Input = _pipelineFactory.Create();;
|
|
}
|
|
|
|
public Frame FrameContext { get; set; }
|
|
|
|
public void Add(string text, bool fin = false)
|
|
{
|
|
var data = Encoding.ASCII.GetBytes(text);
|
|
FrameContext.Input.Writer.WriteAsync(data).Wait();
|
|
if (fin)
|
|
{
|
|
FrameContext.Input.Writer.Complete();
|
|
}
|
|
}
|
|
|
|
public void ProduceContinue()
|
|
{
|
|
}
|
|
|
|
public void Pause()
|
|
{
|
|
}
|
|
|
|
public void Resume()
|
|
{
|
|
}
|
|
|
|
public void End(ProduceEndType endType)
|
|
{
|
|
}
|
|
|
|
public void SetTimeout(long milliseconds, TimeoutAction timeoutAction)
|
|
{
|
|
}
|
|
|
|
public void ResetTimeout(long milliseconds, TimeoutAction timeoutAction)
|
|
{
|
|
}
|
|
|
|
public void CancelTimeout()
|
|
{
|
|
}
|
|
|
|
public void Abort()
|
|
{
|
|
}
|
|
|
|
public void Write(ArraySegment<byte> data, Action<Exception, object> callback, object state)
|
|
{
|
|
}
|
|
|
|
void IFrameControl.ProduceContinue()
|
|
{
|
|
}
|
|
|
|
void IFrameControl.Write(ArraySegment<byte> data)
|
|
{
|
|
}
|
|
|
|
Task IFrameControl.WriteAsync(ArraySegment<byte> data, CancellationToken cancellationToken)
|
|
{
|
|
return TaskCache.CompletedTask;
|
|
}
|
|
|
|
void IFrameControl.Flush()
|
|
{
|
|
}
|
|
|
|
Task IFrameControl.FlushAsync(CancellationToken cancellationToken)
|
|
{
|
|
return TaskCache.CompletedTask;
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
_pipelineFactory.Dispose();
|
|
_memoryPool.Dispose();
|
|
}
|
|
}
|
|
}
|
|
|