From 2aaa23b11d0410c10bdb345d41c8ba740701f12d Mon Sep 17 00:00:00 2001
From: Mark McDowall <markus.mcd5@gmail.com>
Date: Wed, 23 Nov 2011 22:28:52 -0800
Subject: [PATCH] Refreshing EpisodeInfo will now set new or existing episodes
 that have air times prior to 1900 to null.

---
 .../ProviderTests/EpisodeProviderTest.cs      | 45 +++++++++++++++++++
 NzbDrone.Core/Providers/EpisodeProvider.cs    |  5 ++-
 2 files changed, 48 insertions(+), 2 deletions(-)

diff --git a/NzbDrone.Core.Test/ProviderTests/EpisodeProviderTest.cs b/NzbDrone.Core.Test/ProviderTests/EpisodeProviderTest.cs
index 036fb8a52..0b75cb2a0 100644
--- a/NzbDrone.Core.Test/ProviderTests/EpisodeProviderTest.cs
+++ b/NzbDrone.Core.Test/ProviderTests/EpisodeProviderTest.cs
@@ -326,6 +326,51 @@ public void RefreshEpisodeInfo_should_set_older_than_1900_to_null()
             mocker.VerifyAllMocks();
         }
 
+        [Test]
+        public void RefreshEpisodeInfo_should_set_older_than_1900_to_null_for_existing_episodes()
+        {
+            //Arrange
+            const int seriesId = 71663;
+
+            var fakeEpisode = Builder<Episode>.CreateNew()
+                    .With(e => e.TvDbEpisodeId = 12345)
+                    .With(e => e.AirDate = DateTime.Today)
+                    .Build();
+
+            var fakeTvDbEpisodes = Builder<TvdbSeries>.CreateNew().With(
+                c => c.Episodes =
+                     new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(1)
+                         .All()
+                         .With(l => l.Language = new TvdbLanguage(0, "eng", "a")).And(e => e.FirstAired = DateTime.Now)
+                         .TheFirst(1).With(e => e.FirstAired = new DateTime(1800, 1, 1))
+                         .Build())
+                ).With(c => c.Id = seriesId).Build();
+
+            var fakeSeries = Builder<Series>.CreateNew().With(c => c.SeriesId = seriesId).Build();
+
+            var mocker = new AutoMoqer();
+
+            var db = TestDbHelper.GetEmptyDatabase();
+            mocker.SetConstant(db);
+
+            db.Insert(fakeSeries);
+            db.Insert(fakeEpisode);
+
+            mocker.GetMock<TvDbProvider>()
+                .Setup(c => c.GetSeries(seriesId, true))
+                .Returns(fakeTvDbEpisodes);
+
+            //Act
+            mocker.Resolve<EpisodeProvider>().RefreshEpisodeInfo(fakeSeries);
+
+            //Assert
+            var storedEpisodes = mocker.Resolve<EpisodeProvider>().GetEpisodeBySeries(seriesId).ToList();
+            storedEpisodes.Should().HaveCount(1);
+            storedEpisodes.Where(e => e.AirDate == null).Should().HaveCount(1);
+
+            mocker.VerifyAllMocks();
+        }
+
         [Test]
         public void RefreshEpisodeInfo_ignore_episode_zero()
         {
diff --git a/NzbDrone.Core/Providers/EpisodeProvider.cs b/NzbDrone.Core/Providers/EpisodeProvider.cs
index 508abcf1c..a740800d5 100644
--- a/NzbDrone.Core/Providers/EpisodeProvider.cs
+++ b/NzbDrone.Core/Providers/EpisodeProvider.cs
@@ -258,9 +258,10 @@ public virtual void RefreshEpisodeInfo(Series series)
                     episodeToUpdate.Overview = episode.Overview;
 
                     if (episode.FirstAired.Year > 1900)
-                    {
                         episodeToUpdate.AirDate = episode.FirstAired.Date;
-                    }
+
+                    else
+                        episodeToUpdate.AirDate = null;
 
                     successCount++;
                 }