Searching refactored

#ND-135 fixed
2026-04-19 21:44:30 -04:00 · 2013-01-13 00:24:48 -08:00
parent 829ba1cf92
commit 5e7c0951b7
24 changed files with 1848 additions and 1536 deletions
@@ -0,0 +1,100 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using FizzWare.NBuilder;
+using Moq;
+using NUnit.Framework;
+using NzbDrone.Core.Model;
+using NzbDrone.Core.Model.Notification;
+using NzbDrone.Core.Providers;
+using NzbDrone.Core.Providers.Indexer;
+using NzbDrone.Core.Repository;
+using NzbDrone.Test.Common;
+
+namespace NzbDrone.Core.Test.ProviderTests.SearchTests
+{
+    public class PerformSearchTestBase : TestBase
+    {
+        protected Series _series;
+        protected Episode _episode;
+        protected List<Episode> _episodes;
+        protected ProgressNotification notification = new ProgressNotification("Testing");
+
+        protected Mock<IndexerBase> _indexer1;
+        protected Mock<IndexerBase> _indexer2;
+        protected List<IndexerBase> _indexers;
+        protected IList<EpisodeParseResult> _parseResults;
+
+        [SetUp]
+        public void Setup()
+        {
+            _series = Builder<Series>
+                    .CreateNew()
+                    .Build();
+
+            _episode = Builder<Episode>
+                    .CreateNew()
+                    .With(e => e.SeriesId = _series.SeriesId)
+                    .With(e => e.Series = _series)
+                    .Build();
+
+            _episodes = Builder<Episode>
+                    .CreateListOfSize(10)
+                    .All()
+                    .With(e => e.SeriesId = _series.SeriesId)
+                    .With(e => e.Series = _series)
+                    .Build()
+                    .ToList();
+
+            _parseResults = Builder<EpisodeParseResult>
+                    .CreateListOfSize(10)
+                    .Build();
+
+            _indexer1 = new Mock<IndexerBase>();
+            _indexer2 = new Mock<IndexerBase>();
+            _indexers = new List<IndexerBase> { _indexer1.Object, _indexer2.Object };
+
+            Mocker.GetMock<IndexerProvider>()
+                  .Setup(c => c.GetEnabledIndexers())
+                  .Returns(_indexers);
+        }
+
+        protected void WithValidIndexers()
+        {
+            _indexer1.Setup(c => c.FetchEpisode(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
+                .Returns(_parseResults);
+            _indexer1.Setup(c => c.FetchDailyEpisode(It.IsAny<string>(), It.IsAny<DateTime>()))
+                .Returns(_parseResults);
+            _indexer1.Setup(c => c.FetchPartialSeason(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
+                .Returns(_parseResults);
+
+            _indexer2.Setup(c => c.FetchEpisode(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
+                .Returns(_parseResults);
+            _indexer2.Setup(c => c.FetchDailyEpisode(It.IsAny<string>(), It.IsAny<DateTime>()))
+                .Returns(_parseResults);
+            _indexer2.Setup(c => c.FetchPartialSeason(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
+                .Returns(_parseResults);
+        }
+
+        protected void WithInvalidIndexers()
+        {
+            _indexer1.Setup(c => c.FetchEpisode(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
+                .Throws(new Exception());
+            _indexer1.Setup(c => c.FetchDailyEpisode(It.IsAny<string>(), It.IsAny<DateTime>()))
+                .Throws(new Exception());
+            _indexer1.Setup(c => c.FetchPartialSeason(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
+                .Throws(new Exception());
+
+            _indexer2.Setup(c => c.FetchEpisode(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
+                .Throws(new Exception());
+            _indexer2.Setup(c => c.FetchDailyEpisode(It.IsAny<string>(), It.IsAny<DateTime>()))
+                .Throws(new Exception());
+            _indexer2.Setup(c => c.FetchPartialSeason(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
+                .Throws(new Exception());
+
+            _indexer1.SetupGet(c => c.Name).Returns("Indexer1");
+            _indexer1.SetupGet(c => c.Name).Returns("Indexer2");
+        }
+    }
+}