From ae92b2272735ad31a4bd4d4763a5f9a03ea4254e 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

(cherry picked from commit 3d05913534e40e1b9ff217798d806d0b7c170d2d)
---
 .../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 d40808346..9f41a8e81 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;
@@ -217,5 +218,26 @@ namespace NzbDrone.Core.Notifications
             definition.SupportsOnBookRetag = provider.SupportsOnBookRetag;
             definition.SupportsOnApplicationUpdate = provider.SupportsOnApplicationUpdate;
         }
+
+        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;
+        }
     }
 }