2011-05-22 18:53:21 +02:00
|
|
|
|
// ReSharper disable RedundantUsingDirective
|
|
|
|
|
using System;
|
2011-05-19 05:55:35 +02:00
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using NLog;
|
|
|
|
|
using NLog.Targets;
|
|
|
|
|
using MbUnit.Framework;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Test.Framework
|
|
|
|
|
{
|
|
|
|
|
public class ExceptionVerification : Target
|
|
|
|
|
{
|
|
|
|
|
private static List<LogEventInfo> _logs = new List<LogEventInfo>();
|
|
|
|
|
|
|
|
|
|
protected override void Write(LogEventInfo logEvent)
|
|
|
|
|
{
|
|
|
|
|
if (logEvent.Level >= LogLevel.Warn)
|
|
|
|
|
{
|
|
|
|
|
_logs.Add(logEvent);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
internal static void Reset()
|
|
|
|
|
{
|
|
|
|
|
_logs = new List<LogEventInfo>();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
internal static void AssertNoError()
|
|
|
|
|
{
|
2011-05-27 05:04:36 +02:00
|
|
|
|
ExcpectedFatals(0);
|
|
|
|
|
ExcpectedErrors(0);
|
|
|
|
|
ExcpectedWarns(0);
|
2011-05-19 05:55:35 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static string GetLogsString(IEnumerable<LogEventInfo> logs)
|
|
|
|
|
{
|
|
|
|
|
string errors = "";
|
|
|
|
|
foreach (var log in logs)
|
|
|
|
|
{
|
|
|
|
|
string exception = "";
|
|
|
|
|
if (log.Exception != null)
|
|
|
|
|
{
|
2011-05-27 05:04:36 +02:00
|
|
|
|
exception = log.Exception.Message;
|
2011-05-19 05:55:35 +02:00
|
|
|
|
}
|
2011-05-27 05:04:36 +02:00
|
|
|
|
errors += Environment.NewLine + String.Format("[{0}] {1}: {2} [{3}]", log.Level, log.LoggerName, log.FormattedMessage, exception);
|
2011-05-19 05:55:35 +02:00
|
|
|
|
}
|
|
|
|
|
return errors;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
internal static void ExcpectedErrors(int count)
|
|
|
|
|
{
|
|
|
|
|
Excpected(LogLevel.Error, count);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
internal static void ExcpectedFatals(int count)
|
|
|
|
|
{
|
|
|
|
|
Excpected(LogLevel.Fatal, count);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
internal static void ExcpectedWarns(int count)
|
|
|
|
|
{
|
|
|
|
|
Excpected(LogLevel.Warn, count);
|
|
|
|
|
}
|
|
|
|
|
|
2011-05-20 06:18:51 +02:00
|
|
|
|
internal static void IgnoreWarns()
|
|
|
|
|
{
|
|
|
|
|
Ignore(LogLevel.Warn);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
internal static void IgnoreErrors()
|
|
|
|
|
{
|
|
|
|
|
Ignore(LogLevel.Error);
|
|
|
|
|
}
|
|
|
|
|
|
2011-05-19 05:55:35 +02:00
|
|
|
|
private static void Excpected(LogLevel level, int count)
|
|
|
|
|
{
|
|
|
|
|
var levelLogs = _logs.Where(l => l.Level == level).ToList();
|
|
|
|
|
|
|
|
|
|
if (levelLogs.Count != count)
|
|
|
|
|
{
|
2011-05-27 05:04:36 +02:00
|
|
|
|
|
2011-05-19 05:55:35 +02:00
|
|
|
|
var message = String.Format("{0} {1}(s) were expected but {2} were logged.\n\r{3}",
|
2011-05-27 05:04:36 +02:00
|
|
|
|
count, level, levelLogs.Count, GetLogsString(levelLogs));
|
|
|
|
|
|
|
|
|
|
message =
|
|
|
|
|
"********************************************************************************************************************************\n\r"
|
|
|
|
|
+ message +
|
|
|
|
|
"\n\r********************************************************************************************************************************";
|
2011-05-19 05:55:35 +02:00
|
|
|
|
|
|
|
|
|
Assert.Fail(message);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
levelLogs.ForEach(c => _logs.Remove(c));
|
|
|
|
|
}
|
2011-05-20 06:18:51 +02:00
|
|
|
|
|
|
|
|
|
private static void Ignore(LogLevel level)
|
|
|
|
|
{
|
|
|
|
|
var levelLogs = _logs.Where(l => l.Level == level).ToList();
|
|
|
|
|
levelLogs.ForEach(c => _logs.Remove(c));
|
|
|
|
|
}
|
2011-05-19 05:55:35 +02:00
|
|
|
|
}
|
|
|
|
|
}
|