New: Banner will be deleted when series is deleted.

2026-04-19 21:44:30 -04:00 · 2012-03-06 18:53:49 -08:00
parent 0552fd9ad3
commit 5764059f1d
8 changed files with 246 additions and 196 deletions
@@ -0,0 +1,102 @@
+using System;
+using System.IO;
+using System.Linq;
+using System.Net;
+using FizzWare.NBuilder;
+using FluentAssertions;
+using Moq;
+using NUnit.Framework;
+using NzbDrone.Common;
+using NzbDrone.Core.Model.Notification;
+using NzbDrone.Core.Providers;
+using NzbDrone.Core.Repository;
+using NzbDrone.Core.Repository.Quality;
+using NzbDrone.Core.Test.Framework;
+using NzbDrone.Test.Common.AutoMoq;
+
+namespace NzbDrone.Core.Test.ProviderTests
+{
+    [TestFixture]
+    // ReSharper disable InconsistentNaming
+    public class BannerProviderTest : CoreTest
+    {
+        private Series _series;
+        private ProgressNotification _notification;
+
+        [SetUp]
+        public void Setup()
+        {
+            _series = Builder<Series>.CreateNew()
+                .With(s => s.SeriesId = 12345)
+                    .Build();
+
+            _notification = new ProgressNotification("Test");
+
+            var path = @"C:\Windows\Temp";
+
+            Mocker.GetMock<DiskProvider>().Setup(s => s.CreateDirectory(path));
+        }
+
+        private void WithSuccessfulDownload()
+        {
+            Mocker.GetMock<HttpProvider>().Setup(s => s.DownloadFile(It.IsAny<string>(), It.IsAny<string>()));
+        }
+
+        private void WithFailedDownload()
+        {
+            Mocker.GetMock<HttpProvider>().Setup(s => s.DownloadFile(It.IsAny<string>(), It.IsAny<string>()))
+                    .Throws(new WebException("Failed to download file (Mocked)"));
+        }
+
+        [Test]
+        public void Download_should_return_true_when_banner_is_downloaded_successfully()
+        {
+            WithSuccessfulDownload();
+            var result = Mocker.Resolve<BannerProvider>().Download(_notification, _series);
+            result.Should().BeTrue();
+        }
+
+        [Test]
+        public void Download_should_return_false_when_banner_download_fails()
+        {
+            WithFailedDownload();
+            var result = Mocker.Resolve<BannerProvider>().Download(_notification, _series);
+            result.Should().BeFalse();
+        }
+
+        [Test]
+        public void Delete_should_delete_banner_file_when_it_exists()
+        {
+            Mocker.GetMock<DiskProvider>().Setup(s => s.FileExists(It.IsAny<string>()))
+                    .Returns(true);
+
+            Mocker.GetMock<DiskProvider>().Setup(s => s.DeleteFile(It.IsAny<string>()));
+
+            var result = Mocker.Resolve<BannerProvider>().Delete(1);
+            result.Should().BeTrue();
+        }
+
+        [Test]
+        public void Delete_should_return_true_even_when_file_sint_deleted()
+        {
+            Mocker.GetMock<DiskProvider>().Setup(s => s.FileExists(It.IsAny<string>()))
+                    .Returns(false);
+
+            var result = Mocker.Resolve<BannerProvider>().Delete(1);
+            result.Should().BeTrue();
+        }
+
+        [Test]
+        public void Delete_should_return_false_when_file_fails_to_delete()
+        {
+            Mocker.GetMock<DiskProvider>().Setup(s => s.FileExists(It.IsAny<string>()))
+                    .Returns(true);
+
+            Mocker.GetMock<DiskProvider>().Setup(s => s.DeleteFile(It.IsAny<string>()))
+                .Throws(new SystemException("File not found."));
+
+            var result = Mocker.Resolve<BannerProvider>().Delete(1);
+            result.Should().BeFalse();
+        }
+    }
+}