73 lines
3.0 KiB
C#
73 lines
3.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.Collections.Generic;
|
|
using System.IO.Pipelines;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.AspNetCore.Server.Kestrel.Core.Internal;
|
|
using Microsoft.AspNetCore.Server.Kestrel.Transport.Abstractions.Internal;
|
|
using Microsoft.AspNetCore.Testing;
|
|
using Moq;
|
|
using Xunit;
|
|
|
|
namespace Microsoft.AspNetCore.Server.Kestrel.Core.Tests
|
|
{
|
|
public class ConnectionDispatcherTests
|
|
{
|
|
[Fact]
|
|
public void OnConnectionCreatesLogScopeWithConnectionId()
|
|
{
|
|
var serviceContext = new TestServiceContext();
|
|
var tcs = new TaskCompletionSource<object>();
|
|
var dispatcher = new ConnectionDispatcher(serviceContext, _ => tcs.Task);
|
|
|
|
var connection = new Mock<TransportConnection> { CallBase = true }.Object;
|
|
connection.ConnectionClosed = new CancellationToken(canceled: true);
|
|
|
|
dispatcher.OnConnection(connection);
|
|
|
|
// The scope should be created
|
|
var scopeObjects = ((TestKestrelTrace)serviceContext.Log)
|
|
.Logger
|
|
.Scopes
|
|
.OfType<IReadOnlyList<KeyValuePair<string, object>>>()
|
|
.ToList();
|
|
|
|
Assert.Single(scopeObjects);
|
|
var pairs = scopeObjects[0].ToDictionary(p => p.Key, p => p.Value);
|
|
Assert.True(pairs.ContainsKey("ConnectionId"));
|
|
Assert.Equal(connection.ConnectionId, pairs["ConnectionId"]);
|
|
|
|
tcs.TrySetResult(null);
|
|
|
|
// Verify the scope was disposed after request processing completed
|
|
Assert.True(((TestKestrelTrace)serviceContext.Log).Logger.Scopes.IsEmpty);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task OnConnectionCompletesTransportPipesAfterReturning()
|
|
{
|
|
var serviceContext = new TestServiceContext();
|
|
var dispatcher = new ConnectionDispatcher(serviceContext, _ => Task.CompletedTask);
|
|
|
|
var mockConnection = new Mock<TransportConnection> { CallBase = true };
|
|
mockConnection.Object.ConnectionClosed = new CancellationToken(canceled: true);
|
|
var mockPipeReader = new Mock<PipeReader>();
|
|
var mockPipeWriter = new Mock<PipeWriter>();
|
|
var mockPipe = new Mock<IDuplexPipe>();
|
|
mockPipe.Setup(m => m.Input).Returns(mockPipeReader.Object);
|
|
mockPipe.Setup(m => m.Output).Returns(mockPipeWriter.Object);
|
|
mockConnection.Setup(m => m.Transport).Returns(mockPipe.Object);
|
|
var connection = mockConnection.Object;
|
|
|
|
await dispatcher.OnConnection(connection);
|
|
|
|
mockPipeWriter.Verify(m => m.Complete(It.IsAny<Exception>()), Times.Once());
|
|
mockPipeReader.Verify(m => m.Complete(It.IsAny<Exception>()), Times.Once());
|
|
}
|
|
}
|
|
}
|