New: List Status Checks/Backoffs

2026-04-19 21:46:50 -04:00 · 2020-04-07 22:37:48 -04:00
parent d10e60587b
commit b1fd924188
25 changed files with 396 additions and 34 deletions
@@ -0,0 +1,87 @@
+using System;
+using System.Collections.Generic;
+using Moq;
+using NUnit.Framework;
+using NzbDrone.Core.HealthCheck.Checks;
+using NzbDrone.Core.NetImport;
+using NzbDrone.Core.Test.Framework;
+
+namespace NzbDrone.Core.Test.HealthCheck.Checks
+{
+    [TestFixture]
+    public class NetImportStatusCheckFixture : CoreTest<NetImportStatusCheck>
+    {
+        private List<INetImport> _lists = new List<INetImport>();
+        private List<NetImportStatus> _blockedLists = new List<NetImportStatus>();
+
+        [SetUp]
+        public void SetUp()
+        {
+            Mocker.GetMock<INetImportFactory>()
+                  .Setup(v => v.GetAvailableProviders())
+                  .Returns(_lists);
+
+            Mocker.GetMock<INetImportStatusService>()
+                   .Setup(v => v.GetBlockedProviders())
+                   .Returns(_blockedLists);
+        }
+
+        private Mock<INetImport> GivenList(int i, double backoffHours, double failureHours)
+        {
+            var id = i;
+
+            var mockList = new Mock<INetImport>();
+            mockList.SetupGet(s => s.Definition).Returns(new NetImportDefinition { Id = id });
+            mockList.SetupGet(s => s.EnableAuto).Returns(true);
+
+            _lists.Add(mockList.Object);
+
+            if (backoffHours != 0.0)
+            {
+                _blockedLists.Add(new NetImportStatus
+                {
+                    ProviderId = id,
+                    InitialFailure = DateTime.UtcNow.AddHours(-failureHours),
+                    MostRecentFailure = DateTime.UtcNow.AddHours(-0.1),
+                    EscalationLevel = 5,
+                    DisabledTill = DateTime.UtcNow.AddHours(backoffHours)
+                });
+            }
+
+            return mockList;
+        }
+
+        [Test]
+        public void should_not_return_error_when_no_indexers()
+        {
+            Subject.Check().ShouldBeOk();
+        }
+
+        [Test]
+        public void should_return_warning_if_indexer_unavailable()
+        {
+            GivenList(1, 10.0, 24.0);
+            GivenList(2, 0.0, 0.0);
+
+            Subject.Check().ShouldBeWarning();
+        }
+
+        [Test]
+        public void should_return_error_if_all_indexers_unavailable()
+        {
+            GivenList(1, 10.0, 24.0);
+
+            Subject.Check().ShouldBeError();
+        }
+
+        [Test]
+        public void should_return_warning_if_few_indexers_unavailable()
+        {
+            GivenList(1, 10.0, 24.0);
+            GivenList(2, 10.0, 24.0);
+            GivenList(3, 0.0, 0.0);
+
+            Subject.Check().ShouldBeWarning();
+        }
+    }
+}