From 4b53cc7e5296ab49ea12a098193ba8401d61dd89 Mon Sep 17 00:00:00 2001
From: Taloth Saldono <Taloth@users.noreply.github.com>
Date: Mon, 21 Jul 2014 22:56:13 +0200
Subject: [PATCH] Fixed: Manually marking a grabbed release failed will now
 mark all episodes in that release failed instead of only one.

---
 .../Download/FailedDownloadServiceFixture.cs  | 25 +++++++++++++++++++
 .../Download/FailedDownloadService.cs         | 14 +++++++++--
 2 files changed, 37 insertions(+), 2 deletions(-)

diff --git a/src/NzbDrone.Core.Test/Download/FailedDownloadServiceFixture.cs b/src/NzbDrone.Core.Test/Download/FailedDownloadServiceFixture.cs
index 44e563718..f293bcdbd 100644
--- a/src/NzbDrone.Core.Test/Download/FailedDownloadServiceFixture.cs
+++ b/src/NzbDrone.Core.Test/Download/FailedDownloadServiceFixture.cs
@@ -9,6 +9,7 @@ using NzbDrone.Core.Download;
 using NzbDrone.Core.History;
 using NzbDrone.Core.Messaging.Events;
 using NzbDrone.Core.Test.Framework;
+using NzbDrone.Test.Common;
 
 namespace NzbDrone.Core.Test.Download
 {
@@ -379,6 +380,30 @@ namespace NzbDrone.Core.Test.Download
             Subject.Execute(new CheckForFinishedDownloadCommand());
 
             VerifyNoFailedDownloads();
+
+            ExceptionVerification.IgnoreWarns();
+        }
+
+        [Test]
+        public void should_manual_mark_all_episodes_of_release_as_failed()
+        {
+            var historyFailed = Builder<History.History>.CreateListOfSize(2)
+                .All()
+                .With(v => v.EventType == HistoryEventType.Grabbed)
+                .Do(v => v.Data.Add("downloadClient", "SabnzbdClient"))
+                .Do(v => v.Data.Add("downloadClientId", "test"))
+                .Build()
+                .ToList();
+
+            GivenGrabbedHistory(historyFailed);
+
+            Mocker.GetMock<IHistoryService>()
+                .Setup(s => s.Get(It.IsAny<Int32>()))
+                .Returns<Int32>(i => historyFailed.FirstOrDefault(v => v.Id == i));
+
+            Subject.MarkAsFailed(1);
+
+            VerifyFailedDownloads(2);
         }
     }
 }
diff --git a/src/NzbDrone.Core/Download/FailedDownloadService.cs b/src/NzbDrone.Core/Download/FailedDownloadService.cs
index b0137c663..c104897bd 100644
--- a/src/NzbDrone.Core/Download/FailedDownloadService.cs
+++ b/src/NzbDrone.Core/Download/FailedDownloadService.cs
@@ -35,14 +35,24 @@ namespace NzbDrone.Core.Download
             _logger = logger;
         }
 
-        public void MarkAsFailed(TrackedDownload trackedDownload, History.History grabbedHistory)
+        public void MarkAsFailed(TrackedDownload trackedDownload, History.History history)
         {
             if (trackedDownload != null && trackedDownload.State == TrackedDownloadState.Downloading)
             {
                 trackedDownload.State = TrackedDownloadState.DownloadFailed;
             }
 
-            PublishDownloadFailedEvent(new List<History.History> { grabbedHistory }, "Manually marked as failed");
+            var downloadClientId = history.Data.GetValueOrDefault(DownloadTrackingService.DOWNLOAD_CLIENT_ID);
+            if (downloadClientId.IsNullOrWhiteSpace())
+            {
+                PublishDownloadFailedEvent(new List<History.History> { history }, "Manually marked as failed");
+            }
+            else
+            {
+                var grabbedHistory = GetHistoryItems(_historyService.Grabbed(), downloadClientId);
+
+                PublishDownloadFailedEvent(grabbedHistory, "Manually marked as failed");
+            }
         }
 
         public void CheckForFailedItem(IDownloadClient downloadClient, TrackedDownload trackedDownload, List<History.History> grabbedHistory, List<History.History> failedHistory)