New: Add tag support to indexers

(cherry picked from commit c3d54b312ef18b837d54605ea78f1a263fd6900b)
2026-04-25 22:37:27 -04:00 · 2021-03-22 00:00:06 +00:00
parent 4e1aa5b946
commit 64045b0810
17 changed files with 277 additions and 25 deletions
@@ -0,0 +1,102 @@
+using System.Collections.Generic;
+using FizzWare.NBuilder;
+using FluentAssertions;
+using Moq;
+using NUnit.Framework;
+using NzbDrone.Core.DecisionEngine.Specifications.RssSync;
+using NzbDrone.Core.Indexers;
+using NzbDrone.Core.IndexerSearch.Definitions;
+using NzbDrone.Core.Movies;
+using NzbDrone.Core.Parser.Model;
+using NzbDrone.Core.Test.Framework;
+
+namespace NzbDrone.Core.Test.DecisionEngineTests.RssSync
+{
+    [TestFixture]
+    public class IndexerTagSpecificationFixture : CoreTest<IndexerTagSpecification>
+    {
+        private IndexerTagSpecification _specification;
+
+        private RemoteMovie _parseResultMulti;
+        private IndexerDefinition _fakeIndexerDefinition;
+        private Movie _fakeMovie;
+        private ReleaseInfo _fakeRelease;
+
+        [SetUp]
+        public void Setup()
+        {
+            _fakeIndexerDefinition = new IndexerDefinition
+            {
+                Tags = new HashSet<int>()
+            };
+
+            Mocker
+                .GetMock<IIndexerRepository>()
+                .Setup(m => m.Get(It.IsAny<int>()))
+                .Returns(_fakeIndexerDefinition);
+
+            _specification = Mocker.Resolve<IndexerTagSpecification>();
+
+            _fakeMovie = Builder<Movie>.CreateNew()
+                .With(c => c.Monitored = true)
+                .With(c => c.Tags = new HashSet<int>())
+                .Build();
+
+            _fakeRelease = new ReleaseInfo
+            {
+                IndexerId = 1
+            };
+
+            _parseResultMulti = new RemoteMovie
+            {
+                Movie = _fakeMovie,
+                Release = _fakeRelease
+            };
+        }
+
+        [Test]
+        public void indexer_and_movie_without_tags_should_return_true()
+        {
+            _fakeIndexerDefinition.Tags = new HashSet<int>();
+            _fakeMovie.Tags = new HashSet<int>();
+
+            _specification.IsSatisfiedBy(_parseResultMulti, new MovieSearchCriteria()).Accepted.Should().BeTrue();
+        }
+
+        [Test]
+        public void indexer_with_tags_movie_without_tags_should_return_false()
+        {
+            _fakeIndexerDefinition.Tags = new HashSet<int> { 123 };
+            _fakeMovie.Tags = new HashSet<int>();
+
+            _specification.IsSatisfiedBy(_parseResultMulti, new MovieSearchCriteria()).Accepted.Should().BeFalse();
+        }
+
+        [Test]
+        public void indexer_without_tags_movie_with_tags_should_return_true()
+        {
+            _fakeIndexerDefinition.Tags = new HashSet<int>();
+            _fakeMovie.Tags = new HashSet<int> { 123 };
+
+            _specification.IsSatisfiedBy(_parseResultMulti, new MovieSearchCriteria()).Accepted.Should().BeTrue();
+        }
+
+        [Test]
+        public void indexer_with_tags_movie_with_matching_tags_should_return_true()
+        {
+            _fakeIndexerDefinition.Tags = new HashSet<int> { 123, 456 };
+            _fakeMovie.Tags = new HashSet<int> { 123, 789 };
+
+            _specification.IsSatisfiedBy(_parseResultMulti, new MovieSearchCriteria()).Accepted.Should().BeTrue();
+        }
+
+        [Test]
+        public void indexer_with_tags_movie_with_different_tags_should_return_false()
+        {
+            _fakeIndexerDefinition.Tags = new HashSet<int> { 456 };
+            _fakeMovie.Tags = new HashSet<int> { 123, 789 };
+
+            _specification.IsSatisfiedBy(_parseResultMulti, new MovieSearchCriteria()).Accepted.Should().BeFalse();
+        }
+    }
+}