private static void TestUpgrade(Http2ConnectionHandler handler, IChannelHandler multiplexer)
{
IFullHttpRequest request = new DefaultFullHttpRequest(HttpVersion.Http11, HttpMethod.Options, "*");
request.Headers.Set(HttpHeaderNames.Host, "netty.io");
request.Headers.Set(HttpHeaderNames.Connection, "Upgrade, HTTP2-Settings");
request.Headers.Set(HttpHeaderNames.Upgrade, "h2c");
request.Headers.Set((AsciiString)"HTTP2-Settings", "AAMAAABkAAQAAP__");
var parent = new Mock <IServerChannel>();
EmbeddedChannel channel = new EmbeddedChannel(parent.Object, DefaultChannelId.NewInstance(), false, true, new ChannelHandlerAdapter());
IChannelHandlerContext ctx = channel.Pipeline.FirstContext();
Http2ServerUpgradeCodec codec;
if (multiplexer == null)
{
codec = new Http2ServerUpgradeCodec(handler);
}
else
{
codec = new Http2ServerUpgradeCodec((Http2FrameCodec)handler, multiplexer);
}
Assert.True(codec.PrepareUpgradeResponse(ctx, request, new DefaultHttpHeaders()));
codec.UpgradeTo(ctx, request);
// Flush the channel to ensure we write out all buffered data
channel.Flush();
channel.WriteInbound(Http2CodecUtil.ConnectionPrefaceBuf());
Http2FrameInboundWriter writer = new Http2FrameInboundWriter(channel);
writer.WriteInboundSettings(new Http2Settings());
writer.WriteInboundRstStream(Http2CodecUtil.HttpUpgradeStreamId, Http2Error.Cancel);
Assert.Same(handler, channel.Pipeline.Remove <Http2ConnectionHandler>());
Assert.Null(channel.Pipeline.Get <Http2ConnectionHandler>());
Assert.True(channel.Finish());
// Check that the preface was send (a.k.a the settings frame)
var settingsBuffer = channel.ReadOutbound <IByteBuffer>();
Assert.NotNull(settingsBuffer);
settingsBuffer.Release();
var buf = channel.ReadOutbound <IByteBuffer>();
Assert.NotNull(buf);
buf.Release();
Assert.Null(channel.ReadOutbound());
}
private void SetUp(Http2FrameCodecBuilder frameCodecBuilder, Http2Settings initialRemoteSettings)
{
// Some tests call this method twice. Once with JUnit's @Before and once directly to pass special settings.
// This call ensures that in case of two consecutive calls to setUp(), the previous channel is shutdown and
// ByteBufs are released correctly.
Dispose0();
_frameWriter = Http2TestUtil.MockedFrameWriter();
var builder = frameCodecBuilder.FrameWriter(_frameWriter.Object);
builder.FrameLogger = new Http2FrameLogger(Common.Internal.Logging.InternalLogLevel.TRACE);
builder.InitialSettings = initialRemoteSettings;
_frameCodec = frameCodecBuilder.Build();
_inboundHandler = new LastInboundHandler();
_channel = new EmbeddedChannel();
_frameInboundWriter = new Http2FrameInboundWriter(_channel);
//channel.Connect(new InetSocketAddress(0));
_channel.Pipeline.AddLast(_frameCodec);
_channel.Pipeline.AddLast(_inboundHandler);
_channel.Pipeline.FireChannelActive();
// Handshake
_frameWriter.Verify(
x => x.WriteSettingsAsync(
It.Is <IChannelHandlerContext>(v => v == _frameCodec._ctx),
It.IsAny <Http2Settings>(),
It.IsAny <IPromise>()));
_frameWriter.VerifyNoOtherCalls();
_channel.WriteInbound(Http2CodecUtil.ConnectionPrefaceBuf());
_frameInboundWriter.WriteInboundSettings(initialRemoteSettings);
_frameWriter.Verify(
x => x.WriteSettingsAckAsync(
It.Is <IChannelHandlerContext>(v => v == _frameCodec._ctx),
It.IsAny <IPromise>()));
_frameInboundWriter.WriteInboundSettingsAck();
var settingsFrame = _inboundHandler.ReadInbound <IHttp2SettingsFrame>();
Assert.NotNull(settingsFrame);
var settingsAckFrame = _inboundHandler.ReadInbound <IHttp2SettingsAckFrame>();
Assert.NotNull(settingsAckFrame);
}