using System; using System.Collections.Generic; using FluentAssertions; using FluentValidation.Results; using NUnit.Framework; using NzbDrone.Core.MediaFiles; using NzbDrone.Core.Movies; using NzbDrone.Core.Notifications; using NzbDrone.Core.Validation; using NzbDrone.Test.Common; namespace NzbDrone.Core.Test.NotificationTests { [TestFixture] public class NotificationBaseFixture : TestBase { private class TestSetting : NotificationSettingsBase { public override NzbDroneValidationResult Validate() { return new NzbDroneValidationResult(); } } private class TestNotificationWithOnDownload : NotificationBase { public override string Name => "TestNotification"; public override string Link => ""; public override ValidationResult Test() { throw new NotImplementedException(); } public override void OnDownload(DownloadMessage downloadMessage) { TestLogger.Info("OnDownload was called"); } } private class TestNotificationWithAllEvents : NotificationBase { public override string Name => "TestNotification"; public override string Link => ""; public override ValidationResult Test() { throw new NotImplementedException(); } public override void OnGrab(GrabMessage grabMessage) { TestLogger.Info("OnGrab was called"); } public override void OnDownload(DownloadMessage message) { TestLogger.Info("OnDownload was called"); } public override void OnMovieRename(Movie movie, List renamedFiles) { TestLogger.Info("OnRename was called"); } public override void OnMovieAdded(Movie movie) { TestLogger.Info("OnMovieAdded was called"); } public override void OnMovieFileDelete(MovieFileDeleteMessage message) { TestLogger.Info("OnMovieFileDelete was called"); } public override void OnMovieDelete(MovieDeleteMessage deleteMessage) { TestLogger.Info("OnMovieDelete was called"); } public override void OnHealthIssue(NzbDrone.Core.HealthCheck.HealthCheck artist) { TestLogger.Info("OnHealthIssue was called"); } public override void OnHealthRestored(Core.HealthCheck.HealthCheck healthCheck) { TestLogger.Info("OnHealthRestored was called"); } public override void OnApplicationUpdate(ApplicationUpdateMessage updateMessage) { TestLogger.Info("OnApplicationUpdate was called"); } public override void OnManualInteractionRequired(ManualInteractionRequiredMessage message) { TestLogger.Info("OnManualInteractionRequired was called"); } } private class TestNotificationWithNoEvents : NotificationBase { public override string Name => "TestNotification"; public override string Link => ""; public override ValidationResult Test() { throw new NotImplementedException(); } } [Test] public void should_support_OnUpgrade_should_link_to_OnDownload() { var notification = new TestNotificationWithOnDownload(); notification.SupportsOnDownload.Should().BeTrue(); notification.SupportsOnUpgrade.Should().BeTrue(); notification.SupportsOnGrab.Should().BeFalse(); notification.SupportsOnRename.Should().BeFalse(); } [Test] public void should_support_all_if_implemented() { var notification = new TestNotificationWithAllEvents(); notification.SupportsOnGrab.Should().BeTrue(); notification.SupportsOnDownload.Should().BeTrue(); notification.SupportsOnUpgrade.Should().BeTrue(); notification.SupportsOnRename.Should().BeTrue(); notification.SupportsOnMovieAdded.Should().BeTrue(); notification.SupportsOnMovieDelete.Should().BeTrue(); notification.SupportsOnMovieFileDelete.Should().BeTrue(); notification.SupportsOnMovieFileDeleteForUpgrade.Should().BeTrue(); notification.SupportsOnHealthIssue.Should().BeTrue(); notification.SupportsOnHealthRestored.Should().BeTrue(); notification.SupportsOnApplicationUpdate.Should().BeTrue(); notification.SupportsOnManualInteractionRequired.Should().BeTrue(); } [Test] public void should_support_none_if_none_are_implemented() { var notification = new TestNotificationWithNoEvents(); notification.SupportsOnGrab.Should().BeFalse(); notification.SupportsOnDownload.Should().BeFalse(); notification.SupportsOnUpgrade.Should().BeFalse(); notification.SupportsOnRename.Should().BeFalse(); notification.SupportsOnMovieAdded.Should().BeFalse(); notification.SupportsOnMovieDelete.Should().BeFalse(); notification.SupportsOnMovieFileDelete.Should().BeFalse(); notification.SupportsOnMovieFileDeleteForUpgrade.Should().BeFalse(); notification.SupportsOnHealthIssue.Should().BeFalse(); notification.SupportsOnHealthRestored.Should().BeFalse(); notification.SupportsOnApplicationUpdate.Should().BeFalse(); notification.SupportsOnManualInteractionRequired.Should().BeFalse(); } } }