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.
Radarr/src/NzbDrone.Test.Common/ExceptionVerification.cs

175 lines
4.8 KiB

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using NLog;
using NLog.Targets;
using NUnit.Framework;
namespace NzbDrone.Test.Common
{
public class ExceptionVerification : Target
{
private static List<LogEventInfo> _logs = new List<LogEventInfo>();
private static ManualResetEventSlim _waitEvent = new ManualResetEventSlim();
protected override void Write(LogEventInfo logEvent)
{
lock (_logs)
{
if (logEvent.Level >= LogLevel.Warn)
{
_logs.Add(logEvent);
_waitEvent.Set();
}
}
}
public static void Reset()
{
lock (_logs)
{
_logs.Clear();
_waitEvent.Reset();
}
}
public static void AssertNoUnexpectedLogs()
{
ExpectedFatals(0);
ExpectedErrors(0);
ExpectedWarns(0);
}
private static string GetLogsString(IEnumerable<LogEventInfo> logs)
{
var errors = "";
foreach (var log in logs)
{
var exception = "";
if (log.Exception != null)
{
exception = string.Format("[{0}: {1}]", log.Exception.GetType(), log.Exception.Message);
}
errors += Environment.NewLine + string.Format("[{0}] {1}: {2} {3}", log.Level, log.LoggerName, log.FormattedMessage, exception);
}
return errors;
}
public static void WaitForErrors(int count, int msec)
{
while (true)
{
lock (_logs)
{
var levelLogs = _logs.Where(l => l.Level == LogLevel.Error).ToList();
if (levelLogs.Count >= count)
{
break;
}
_waitEvent.Reset();
}
if (!_waitEvent.Wait(msec))
{
break;
}
}
Expected(LogLevel.Error, count);
}
public static void ExpectedErrors(int count)
{
Expected(LogLevel.Error, count);
}
public static void ExpectedFatals(int count)
{
Expected(LogLevel.Fatal, count);
}
public static void ExpectedWarns(int count)
{
Expected(LogLevel.Warn, count);
}
public static void IgnoreWarns()
{
Ignore(LogLevel.Warn);
}
public static void IgnoreErrors()
{
Ignore(LogLevel.Error);
}
public static void MarkInconclusive(Type exception)
{
lock (_logs)
{
var inconclusiveLogs = _logs.Where(l => l.Exception != null && l.Exception.GetType() == exception).ToList();
if (inconclusiveLogs.Any())
{
inconclusiveLogs.ForEach(c => _logs.Remove(c));
Assert.Inconclusive(GetLogsString(inconclusiveLogs));
}
}
}
public static void MarkInconclusive(string text)
{
lock (_logs)
{
var inconclusiveLogs = _logs.Where(l => l.FormattedMessage.ToLower().Contains(text.ToLower())).ToList();
if (inconclusiveLogs.Any())
{
inconclusiveLogs.ForEach(c => _logs.Remove(c));
Assert.Inconclusive(GetLogsString(inconclusiveLogs));
}
}
}
private static void Expected(LogLevel level, int count)
{
lock (_logs)
{
var levelLogs = _logs.Where(l => l.Level == level).ToList();
if (levelLogs.Count != count)
{
var message = string.Format("{0} {1}(s) were expected but {2} were logged.\n\r{3}",
count,
level,
levelLogs.Count,
GetLogsString(levelLogs));
message = "\n\r****************************************************************************************\n\r"
+ message +
"\n\r****************************************************************************************";
Assert.Fail(message);
}
levelLogs.ForEach(c => _logs.Remove(c));
}
}
private static void Ignore(LogLevel level)
{
lock (_logs)
{
var levelLogs = _logs.Where(l => l.Level == level).ToList();
levelLogs.ForEach(c => _logs.Remove(c));
}
}
}
}