aspnetcore/src/SignalR/common/testassets/Tests.Utils/FunctionalTestBase.cs

46 lines
1.5 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.Threading.Tasks;
using Microsoft.Extensions.Logging.Testing;
namespace Microsoft.AspNetCore.SignalR.Tests
{
public class FunctionalTestBase : VerifiableLoggedTest
{
private readonly Func<WriteContext, bool> _globalExpectedErrorsFilter;
public FunctionalTestBase()
{
// Suppress errors globally here
_globalExpectedErrorsFilter = (writeContext) => false;
}
private Func<WriteContext, bool> ResolveExpectedErrorsFilter(Func<WriteContext, bool> expectedErrorsFilter)
{
if (expectedErrorsFilter == null)
{
return _globalExpectedErrorsFilter;
}
return (writeContext) =>
{
if (expectedErrorsFilter(writeContext))
{
return true;
}
return _globalExpectedErrorsFilter(writeContext);
};
}
public Task<InProcessTestServer<T>> StartServer<T>(Func<WriteContext, bool> expectedErrorsFilter = null) where T : class
{
var disposable = base.StartVerifiableLog(ResolveExpectedErrorsFilter(expectedErrorsFilter));
return InProcessTestServer<T>.StartServer(LoggerFactory, disposable);
}
}
}