src/Middleware/WebSockets/test/UnitTests/Http2WebSocketTests.cs - main - githubmirror/aspnetcore

src/Middleware/WebSockets/test/UnitTests/Http2WebSocketTests.cs
74 строки3 KB
Chris Ross
Implement Http/2 WebSockets (#41558)
08 июн 2022, 20:18
08 июн 2022, 20:18b14c9b8
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Http.Features;
using Microsoft.AspNetCore.TestHost;
using Microsoft.Extensions.Hosting;
using Microsoft.Net.Http.Headers;

namespace Microsoft.AspNetCore.WebSockets.Tests;

public class Http2WebSocketTests
{
    [Fact]
    public async Task Http2Handshake_Success()
    {
        using var host = new HostBuilder()
            .ConfigureWebHost(webHost =>
            {
                webHost.UseTestServer();
                webHost.Configure(app =>
                {
                    app.UseWebSockets();
                    app.Run(httpContext =>
                    {
                        Assert.True(httpContext.WebSockets.IsWebSocketRequest);
                        Assert.Equal(new[] { "p1", "p2" }, httpContext.WebSockets.WebSocketRequestedProtocols);
                        return httpContext.WebSockets.AcceptWebSocketAsync("p2");
                    });
                });
            }).Start();

        var testServer = host.GetTestServer();

        var result = await testServer.SendAsync(httpContext =>
        {
            httpContext.Request.Method = HttpMethods.Connect;
            httpContext.Features.Set<IHttpExtendedConnectFeature>(new ConnectFeature()
            {
                IsExtendedConnect = true,
                Protocol = "WebSocket",
            });
            httpContext.Request.Headers.SecWebSocketVersion = Constants.Headers.SupportedVersion;
            httpContext.Request.Headers.SecWebSocketProtocol = "p1, p2";
        });

        Assert.Equal(StatusCodes.Status200OK, result.Response.StatusCode);
        var headers = result.Response.Headers;
        Assert.Equal("p2", headers.SecWebSocketProtocol);
        Assert.False(headers.TryGetValue(HeaderNames.Connection, out var _));
        Assert.False(headers.TryGetValue(HeaderNames.Upgrade, out var _));
        Assert.False(headers.TryGetValue(HeaderNames.SecWebSocketAccept, out var _));
    }

    public sealed class ConnectFeature : IHttpExtendedConnectFeature
    {
        public bool IsExtendedConnect { get; set; }
        public string Protocol { get; set; }
        public Stream Stream { get; set; } = Stream.Null;

        /// <inheritdoc/>
        public ValueTask<Stream> AcceptAsync()
        {
            if (!IsExtendedConnect)
            {
                throw new InvalidOperationException("This is not an Extended CONNECT request.");
            }

            return new ValueTask<Stream>(Stream);
        }
    }
}