From 3d05913534e40e1b9ff217798d806d0b7c170d2d Mon Sep 17 00:00:00 2001
From: Bogdan <mynameisbogdan@users.noreply.github.com>
Date: Fri, 10 Nov 2023 04:55:01 +0200
Subject: [PATCH] Fixed: Record status for notifications on tests

---
 .../Notifications/NotificationFactory.cs      | 22 +++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/src/NzbDrone.Core/Notifications/NotificationFactory.cs b/src/NzbDrone.Core/Notifications/NotificationFactory.cs
index ca7fa6215..35823feca 100644
--- a/src/NzbDrone.Core/Notifications/NotificationFactory.cs
+++ b/src/NzbDrone.Core/Notifications/NotificationFactory.cs
@@ -1,6 +1,7 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
+using FluentValidation.Results;
 using NLog;
 using NzbDrone.Core.Messaging.Events;
 using NzbDrone.Core.ThingiProvider;
@@ -193,5 +194,26 @@ namespace NzbDrone.Core.Notifications
             definition.SupportsOnApplicationUpdate = provider.SupportsOnApplicationUpdate;
             definition.SupportsOnManualInteractionRequired = provider.SupportsOnManualInteractionRequired;
         }
+
+        public override ValidationResult Test(NotificationDefinition definition)
+        {
+            var result = base.Test(definition);
+
+            if (definition.Id == 0)
+            {
+                return result;
+            }
+
+            if (result == null || result.IsValid)
+            {
+                _notificationStatusService.RecordSuccess(definition.Id);
+            }
+            else
+            {
+                _notificationStatusService.RecordFailure(definition.Id);
+            }
+
+            return result;
+        }
     }
 }