From 3fdfa69b94e50a413b8c539478cf50f23dcef6c3 Mon Sep 17 00:00:00 2001
From: Mark McDowall <markus.mcd5@gmail.com>
Date: Thu, 17 May 2012 18:23:32 -0700
Subject: [PATCH] Updated tests for UpcomingProvider. #ND-45 fixed

---
 .../UpcomingEpisodesProviderTest.cs           | 84 ++-----------------
 .../Providers/UpcomingEpisodesProvider.cs     | 22 +----
 2 files changed, 8 insertions(+), 98 deletions(-)

diff --git a/NzbDrone.Core.Test/ProviderTests/UpcomingEpisodesProviderTest.cs b/NzbDrone.Core.Test/ProviderTests/UpcomingEpisodesProviderTest.cs
index 627dd2ca1..317c5b0fb 100644
--- a/NzbDrone.Core.Test/ProviderTests/UpcomingEpisodesProviderTest.cs
+++ b/NzbDrone.Core.Test/ProviderTests/UpcomingEpisodesProviderTest.cs
@@ -64,97 +64,27 @@ namespace NzbDrone.Core.Test.ProviderTests
         }
 
         [Test]
-        public void Get_Yesterday()
+        public void Get_UpcomingEpisodes()
         {
-            var result = Mocker.Resolve<UpcomingEpisodesProvider>().Yesterday();
+            var result = Mocker.Resolve<UpcomingEpisodesProvider>().UpcomingEpisodes();
 
             //Assert
-            result.Should().NotBeEmpty();
-            result.Should().OnlyContain(c => c.AirDate.Value.Date == DateTime.Today.AddDays(-1).Date);
-            result.First().Series.Should().NotBeNull();
-            result.First().Series.SeriesId.Should().Be(series.SeriesId);
-        }
-
-        [Test]
-        public void Get_Today()
-        {
-            //Act
-            var result = Mocker.Resolve<UpcomingEpisodesProvider>().Today();
-
-            result.Should().NotBeEmpty();
-            result.Should().OnlyContain(c => c.AirDate.Value.Date == DateTime.Today.Date);
-            result.First().Series.Should().NotBeNull();
-            result.First().Series.SeriesId.Should().Be(series.SeriesId);
-        }
-
-        [Test]
-        public void Get_Tomorrow()
-        {
-            var result = Mocker.Resolve<UpcomingEpisodesProvider>().Tomorrow();
-
-            result.Should().NotBeEmpty();
-            result.Should().OnlyContain(c => c.AirDate.Value.Date == DateTime.Today.AddDays(1).Date);
-            result.First().Series.Should().NotBeNull();
-            result.First().Series.SeriesId.Should().Be(series.SeriesId);
-        }
-
-        [Test]
-        public void Get_Week()
-        {
-            var result = Mocker.Resolve<UpcomingEpisodesProvider>().Week();
-
-            //Assert
-            result.Should().HaveCount(2);
+            result.Should().HaveCount(5);
             result.Should().OnlyContain(c => c.Series != null && c.SeriesId == series.SeriesId);
         }
 
         [Test]
-        public void Get_Yesterday_should_skip_ingored()
-        {
-            WithIgnoredEpisodes();
-            Mocker.Resolve<UpcomingEpisodesProvider>().Yesterday().Should().BeEmpty();
-        }
-
-        [Test]
-        public void Get_Today_should_skip_ingored()
+        public void Get_UpcomingEpisodes_should_skip_ingored()
         {
             WithIgnoredEpisodes();
-            Mocker.Resolve<UpcomingEpisodesProvider>().Today().Should().BeEmpty();
-        }
-
-        [Test]
-        public void Get_Tomorrow_should_skip_ingored()
-        {
-            WithIgnoredEpisodes();
-            Mocker.Resolve<UpcomingEpisodesProvider>().Tomorrow().Should().BeEmpty();
-        }
-
-        [Test]
-        public void Get_Week_should_skip_ingored()
-        {
-            WithIgnoredEpisodes();
-            Mocker.Resolve<UpcomingEpisodesProvider>().Week().Should().BeEmpty();
-        }
-
-        [Test]
-        public void Get_Today_should_skip_unmonitored_series()
-        {
-            WithIgnoredSeries();
-            Mocker.Resolve<UpcomingEpisodesProvider>().Today().Should().BeEmpty();
-        }
-
-        [Test]
-        public void Get_Tomoroww_should_skip_unmonitored_series()
-        {
-            WithIgnoredSeries();
-            Mocker.Resolve<UpcomingEpisodesProvider>().Tomorrow().Should().BeEmpty();
+            Mocker.Resolve<UpcomingEpisodesProvider>().UpcomingEpisodes().Should().BeEmpty();
         }
 
         [Test]
-        public void Get_Week_should_skip_unmonitored_series()
+        public void Get_UpcomingEpisodes_should_skip_unmonitored_series()
         {
             WithIgnoredSeries();
-            Mocker.Resolve<UpcomingEpisodesProvider>().Week().Should().BeEmpty();
+            Mocker.Resolve<UpcomingEpisodesProvider>().UpcomingEpisodes().Should().BeEmpty();
         }
     }
 }
diff --git a/NzbDrone.Core/Providers/UpcomingEpisodesProvider.cs b/NzbDrone.Core/Providers/UpcomingEpisodesProvider.cs
index 00c924d84..7cb4df06f 100644
--- a/NzbDrone.Core/Providers/UpcomingEpisodesProvider.cs
+++ b/NzbDrone.Core/Providers/UpcomingEpisodesProvider.cs
@@ -18,27 +18,7 @@ namespace NzbDrone.Core.Providers
             _database = database;
         }
 
-        public virtual List<Episode> Yesterday()
-        {
-            return RecentEpisodes().Where(c => c.AirDate.Value.Date == DateTime.Now.Date.AddDays(-1)).ToList();
-        }
-
-        public virtual List<Episode> Today()
-        {
-            return RecentEpisodes().Where(c => c.AirDate.Value.Date == DateTime.Now.Date).ToList();
-        }
-
-        public virtual List<Episode> Tomorrow()
-        {
-            return RecentEpisodes().Where(c => c.AirDate.Value.Date == DateTime.Now.Date.AddDays(1)).ToList();
-        }
-
-        public virtual List<Episode> Week()
-        {
-            return RecentEpisodes().Where(c => c.AirDate >= DateTime.Today.AddDays(2).Date).ToList();
-        }
-
-        public virtual List<Episode> RecentEpisodes()
+        public virtual List<Episode> UpcomingEpisodes()
         {
             return _database.Fetch<Episode, Series>(@"SELECT * FROM Episodes 
                                                         INNER JOIN Series ON Episodes.SeriesId = Series.SeriesId