New: Renamed Blacklist to Blocklist

2026-04-27 22:57:09 -04:00 · 2021-08-19 20:13:14 +01:00
parent 8024f4658c
commit 13224f03cc
52 changed files with 614 additions and 589 deletions
@@ -0,0 +1,91 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using FizzWare.NBuilder;
+using FluentAssertions;
+using NUnit.Framework;
+using NzbDrone.Core.Blocklisting;
+using NzbDrone.Core.Languages;
+using NzbDrone.Core.Movies;
+using NzbDrone.Core.Qualities;
+using NzbDrone.Core.Test.Framework;
+
+namespace NzbDrone.Core.Test.Blocklisting
+{
+    [TestFixture]
+    public class BlocklistRepositoryFixture : DbTest<BlocklistRepository, Blocklist>
+    {
+        private Blocklist _blocklist;
+        private Movie _movie1;
+        private Movie _movie2;
+
+        [SetUp]
+        public void Setup()
+        {
+            _blocklist = new Blocklist
+            {
+                MovieId = 1234,
+                Quality = new QualityModel(),
+                Languages = new List<Language>(),
+                SourceTitle = "movie.title.1998",
+                Date = DateTime.UtcNow
+            };
+
+            _movie1 = Builder<Movie>.CreateNew()
+                         .With(s => s.Id = 7)
+                         .Build();
+
+            _movie2 = Builder<Movie>.CreateNew()
+                                     .With(s => s.Id = 8)
+                                     .Build();
+        }
+
+        [Test]
+        public void should_be_able_to_write_to_database()
+        {
+            Subject.Insert(_blocklist);
+            Subject.All().Should().HaveCount(1);
+        }
+
+        [Test]
+        public void should_should_have_movie_id()
+        {
+            Subject.Insert(_blocklist);
+
+            Subject.All().First().MovieId.Should().Be(_blocklist.MovieId);
+        }
+
+        [Test]
+        public void should_check_for_blocklisted_title_case_insensative()
+        {
+            Subject.Insert(_blocklist);
+
+            Subject.BlocklistedByTitle(_blocklist.MovieId, _blocklist.SourceTitle.ToUpperInvariant()).Should().HaveCount(1);
+        }
+
+        [Test]
+        public void should_delete_blocklists_by_movieId()
+        {
+            var blocklistItems = Builder<Blocklist>.CreateListOfSize(5)
+                .TheFirst(1)
+                .With(c => c.MovieId = _movie2.Id)
+                .TheRest()
+                .With(c => c.MovieId = _movie1.Id)
+                .All()
+                .With(c => c.Quality = new QualityModel())
+                .With(c => c.Languages = new List<Language>())
+                .With(c => c.Id = 0)
+                .BuildListOfNew();
+
+            Db.InsertMany(blocklistItems);
+
+            Subject.DeleteForMovies(new List<int> { _movie1.Id });
+
+            var removedMovieBlocklists = Subject.BlocklistedByMovie(_movie1.Id);
+            var nonRemovedMovieBlocklists = Subject.BlocklistedByMovie(_movie2.Id);
+
+            removedMovieBlocklists.Should().HaveCount(0);
+            nonRemovedMovieBlocklists.Should().HaveCount(1);
+        }
+    }
+}
@@ -0,0 +1,56 @@
+using System;
+using Moq;
+using NUnit.Framework;
+using NzbDrone.Core.Blocklisting;
+using NzbDrone.Core.Download;
+using NzbDrone.Core.Qualities;
+using NzbDrone.Core.Test.Framework;
+
+namespace NzbDrone.Core.Test.Blocklisting
+{
+    [TestFixture]
+    public class BlocklistServiceFixture : CoreTest<BlocklistService>
+    {
+        private DownloadFailedEvent _event;
+
+        [SetUp]
+        public void Setup()
+        {
+            _event = new DownloadFailedEvent
+            {
+                MovieId = 69,
+                Quality = new QualityModel(),
+                SourceTitle = "series.title.s01e01",
+                DownloadClient = "SabnzbdClient",
+                DownloadId = "Sabnzbd_nzo_2dfh73k"
+            };
+
+            _event.Data.Add("publishedDate", DateTime.UtcNow.ToString("s") + "Z");
+            _event.Data.Add("size", "1000");
+            _event.Data.Add("indexer", "nzbs.org");
+            _event.Data.Add("protocol", "1");
+            _event.Data.Add("message", "Marked as failed");
+        }
+
+        [Test]
+        public void should_add_to_repository()
+        {
+            Subject.Handle(_event);
+
+            Mocker.GetMock<IBlocklistRepository>()
+                .Verify(v => v.Insert(It.Is<Blocklist>(b => b.MovieId == _event.MovieId)), Times.Once());
+        }
+
+        [Test]
+        public void should_add_to_repository_missing_size_and_protocol()
+        {
+            Subject.Handle(_event);
+
+            _event.Data.Remove("size");
+            _event.Data.Remove("protocol");
+
+            Mocker.GetMock<IBlocklistRepository>()
+                .Verify(v => v.Insert(It.Is<Blocklist>(b => b.MovieId == _event.MovieId)), Times.Once());
+        }
+    }
+}