src/Shared/SignalR/VerifyNoErrorScope.cs - main - githubmirror/aspnetcore

src/Shared/SignalR/VerifyNoErrorScope.cs
67 строк2 KB
Keith Dahlby
Resolve CA2201: No Reserved Exceptions (#47873)
03 май 2023, 20:03
03 май 2023, 20:030c8a569
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using System;
using System.Linq;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Logging.Testing;
using Xunit;

namespace Microsoft.AspNetCore.SignalR.Tests;

public class VerifyNoErrorsScope : IDisposable
{
    private readonly IDisposable _wrappedDisposable;
    private readonly Func<WriteContext, bool> _expectedErrorsFilter;
    private readonly LogSinkProvider _sink;

    public ILoggerFactory LoggerFactory { get; }

    public IList<LogRecord> GetLogs() => _sink.GetLogs();

    public VerifyNoErrorsScope(ILoggerFactory loggerFactory = null, IDisposable wrappedDisposable = null, Func<WriteContext, bool> expectedErrorsFilter = null)
    {
        _wrappedDisposable = wrappedDisposable;
        _expectedErrorsFilter = expectedErrorsFilter;
        _sink = new LogSinkProvider();

        LoggerFactory = loggerFactory ?? new LoggerFactory();
        LoggerFactory.AddProvider(_sink);
    }

    public void Dispose()
    {
        _wrappedDisposable?.Dispose();

        var results = _sink.GetLogs().Where(w => w.Write.LogLevel >= LogLevel.Error).ToList();

        if (_expectedErrorsFilter != null)
        {
            results = results.Where(w => !_expectedErrorsFilter(w.Write)).ToList();
        }

        if (results.Count > 0)
        {
            string errorMessage = $"{results.Count} error(s) logged.";
            errorMessage += Environment.NewLine;
            errorMessage += string.Join(Environment.NewLine, results.Select(record =>
            {
                var r = record.Write;

                string lineMessage = r.LoggerName + " - " + r.EventId.ToString() + " - " + r.Formatter(r.State, r.Exception);
                if (r.Exception != null)
                {
                    lineMessage += Environment.NewLine;
                    lineMessage += "===================";
                    lineMessage += Environment.NewLine;
                    lineMessage += r.Exception;
                    lineMessage += Environment.NewLine;
                    lineMessage += "===================";
                }
                return lineMessage;
            }));

            Assert.Fail(errorMessage);
        }
    }
}