2014-02-26 05:40:47 +00:00
|
|
|
|
using FluentAssertions;
|
2017-02-28 19:59:22 +00:00
|
|
|
|
using NzbDrone.Common.Extensions;
|
2014-02-26 05:40:47 +00:00
|
|
|
|
using NzbDrone.Core.HealthCheck;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Test.HealthCheck.Checks
|
|
|
|
|
{
|
|
|
|
|
public static class HealthCheckFixtureExtensions
|
|
|
|
|
{
|
2014-04-10 00:15:13 +00:00
|
|
|
|
public static void ShouldBeOk(this Core.HealthCheck.HealthCheck result)
|
|
|
|
|
{
|
|
|
|
|
result.Type.Should().Be(HealthCheckResult.Ok);
|
|
|
|
|
}
|
|
|
|
|
|
2019-08-24 10:17:28 +00:00
|
|
|
|
public static void ShouldBeNotice(this Core.HealthCheck.HealthCheck result, string message = null)
|
|
|
|
|
{
|
|
|
|
|
result.Type.Should().Be(HealthCheckResult.Notice);
|
|
|
|
|
|
|
|
|
|
if (message.IsNotNullOrWhiteSpace())
|
|
|
|
|
{
|
|
|
|
|
result.Message.Should().Contain(message);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2021-05-15 23:19:56 +00:00
|
|
|
|
public static void ShouldBeWarning(this Core.HealthCheck.HealthCheck result, string message = null, string wikiFragment = null)
|
2014-02-26 05:40:47 +00:00
|
|
|
|
{
|
2014-04-10 00:15:13 +00:00
|
|
|
|
result.Type.Should().Be(HealthCheckResult.Warning);
|
2017-02-28 19:59:22 +00:00
|
|
|
|
|
|
|
|
|
if (message.IsNotNullOrWhiteSpace())
|
|
|
|
|
{
|
|
|
|
|
result.Message.Should().Contain(message);
|
|
|
|
|
}
|
2021-05-15 23:19:56 +00:00
|
|
|
|
|
|
|
|
|
if (wikiFragment.IsNotNullOrWhiteSpace())
|
|
|
|
|
{
|
|
|
|
|
result.WikiUrl.Fragment.Should().Be(wikiFragment);
|
|
|
|
|
}
|
2014-02-26 05:40:47 +00:00
|
|
|
|
}
|
|
|
|
|
|
2019-04-04 15:15:43 +00:00
|
|
|
|
public static void ShouldBeError(this Core.HealthCheck.HealthCheck result, string message = null, string wikiFragment = null)
|
2014-02-26 05:40:47 +00:00
|
|
|
|
{
|
2014-04-10 00:15:13 +00:00
|
|
|
|
result.Type.Should().Be(HealthCheckResult.Error);
|
2017-02-28 19:59:22 +00:00
|
|
|
|
|
|
|
|
|
if (message.IsNotNullOrWhiteSpace())
|
|
|
|
|
{
|
|
|
|
|
result.Message.Should().Contain(message);
|
|
|
|
|
}
|
2019-04-04 15:15:43 +00:00
|
|
|
|
|
|
|
|
|
if (wikiFragment.IsNotNullOrWhiteSpace())
|
|
|
|
|
{
|
|
|
|
|
result.WikiUrl.Fragment.Should().Be(wikiFragment);
|
|
|
|
|
}
|
2014-02-26 05:40:47 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|