You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Prowlarr/src/NzbDrone.Test.Common/LoggingTest.cs

109 lines
4.0 KiB

using System;
using System.IO;
using NLog;
using NLog.Config;
using NLog.Targets;
using NUnit.Framework;
using NUnit.Framework.Interfaces;
using NzbDrone.Common.EnvironmentInfo;
using NzbDrone.Common.Extensions;
using NzbDrone.Common.Instrumentation;
namespace NzbDrone.Test.Common
{
public abstract class LoggingTest
{
protected static readonly Logger TestLogger = NzbDroneLogger.GetLogger("TestLogger");
protected static void InitLogging()
{
new StartupContext();
if (LogManager.Configuration == null || LogManager.Configuration.AllTargets.None(c => c is ExceptionVerification))
{
LogManager.Configuration = new LoggingConfiguration();
var logOutput = TestLogOutput.Console;
4 years ago
Enum.TryParse<TestLogOutput>(Environment.GetEnvironmentVariable("PROWLARR_TESTS_LOG_OUTPUT"), out logOutput);
RegisterSentryLogger();
switch (logOutput)
{
case TestLogOutput.Console:
RegisterConsoleLogger();
break;
case TestLogOutput.File:
RegisterFileLogger();
break;
}
RegisterExceptionVerification();
LogManager.ReconfigExistingLoggers();
}
}
private static void RegisterConsoleLogger()
{
var consoleTarget = new ConsoleTarget { Layout = "${level}: ${message} ${exception}" };
LogManager.Configuration.AddTarget(consoleTarget.GetType().Name, consoleTarget);
LogManager.Configuration.LoggingRules.Add(new LoggingRule("*", LogLevel.Trace, consoleTarget));
}
private static void RegisterFileLogger()
{
const string layout = @"${level}|${message}${onexception:inner=${newline}${newline}${exception:format=ToString}${newline}}";
var fileTarget = new FileTarget();
fileTarget.Name = "Test File Logger";
fileTarget.FileName = Path.Combine(TestContext.CurrentContext.WorkDirectory, "TestLog.txt");
fileTarget.AutoFlush = false;
fileTarget.KeepFileOpen = true;
fileTarget.ConcurrentWrites = true;
fileTarget.ConcurrentWriteAttemptDelay = 50;
fileTarget.ConcurrentWriteAttempts = 10;
fileTarget.Layout = layout;
LogManager.Configuration.AddTarget(fileTarget.GetType().Name, fileTarget);
LogManager.Configuration.LoggingRules.Add(new LoggingRule("*", LogLevel.Trace, fileTarget));
}
private static void RegisterSentryLogger()
{
// Register a null target for sentry logs, so they aren't caught by other loggers.
var loggingRuleSentry = new LoggingRule("Sentry", LogLevel.Debug, new NullTarget()) { Final = true };
LogManager.Configuration.LoggingRules.Insert(0, loggingRuleSentry);
}
private static void RegisterExceptionVerification()
{
var exceptionVerification = new ExceptionVerification();
LogManager.Configuration.AddTarget("ExceptionVerification", exceptionVerification);
LogManager.Configuration.LoggingRules.Add(new LoggingRule("*", LogLevel.Warn, exceptionVerification));
}
[SetUp]
public void LoggingTestSetup()
{
InitLogging();
ExceptionVerification.Reset();
TestLogger.Info("--- Start: {0} ---", TestContext.CurrentContext.Test.FullName);
}
[TearDown]
public void LoggingDownBase()
{
//can't use because of a bug in mono with 2.6.2,
//https://bugs.launchpad.net/nunitv2/+bug/1076932
if (BuildInfo.IsDebug && TestContext.CurrentContext.Result.Outcome == ResultState.Success)
{
ExceptionVerification.AssertNoUnexpectedLogs();
}
TestLogger.Info("--- End: {0} ---", TestContext.CurrentContext.Test.FullName);
}
}
}