72 lines
2.0 KiB
C#
72 lines
2.0 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.Text;
|
|
using Microsoft.AspNetCore.Server.Kestrel.Core.Internal.Http;
|
|
using Microsoft.AspNetCore.Server.Kestrel.Core.Internal.Infrastructure;
|
|
using Microsoft.AspNetCore.Server.Kestrel.Internal.System.IO.Pipelines;
|
|
using Microsoft.AspNetCore.Testing;
|
|
using Moq;
|
|
|
|
namespace Microsoft.AspNetCore.Server.Kestrel.Core.Tests
|
|
{
|
|
class TestInput : IDisposable
|
|
{
|
|
private MemoryPool _memoryPool;
|
|
private PipeFactory _pipelineFactory;
|
|
|
|
public TestInput()
|
|
{
|
|
_memoryPool = new MemoryPool();
|
|
_pipelineFactory = new PipeFactory();
|
|
Pipe = _pipelineFactory.Create();
|
|
|
|
FrameContext = new FrameContext
|
|
{
|
|
ServiceContext = new TestServiceContext(),
|
|
Input = Pipe.Reader,
|
|
ConnectionInformation = new MockConnectionInformation
|
|
{
|
|
PipeFactory = _pipelineFactory
|
|
},
|
|
TimeoutControl = Mock.Of<ITimeoutControl>()
|
|
};
|
|
|
|
Frame = new Frame<object>(null, FrameContext);
|
|
Frame.FrameControl = Mock.Of<IFrameControl>();
|
|
}
|
|
|
|
public IPipe Pipe { get; }
|
|
|
|
public PipeFactory PipeFactory => _pipelineFactory;
|
|
|
|
public FrameContext FrameContext { get; }
|
|
|
|
public Frame Frame { get; set; }
|
|
|
|
public void Add(string text)
|
|
{
|
|
var data = Encoding.ASCII.GetBytes(text);
|
|
Pipe.Writer.WriteAsync(data).Wait();
|
|
}
|
|
|
|
public void Fin()
|
|
{
|
|
Pipe.Writer.Complete();
|
|
}
|
|
|
|
public void Cancel()
|
|
{
|
|
Pipe.Reader.CancelPendingRead();
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
_pipelineFactory.Dispose();
|
|
_memoryPool.Dispose();
|
|
}
|
|
}
|
|
}
|
|
|