HistoryProvider now uses PetaPoco

NzbDrone.Core.Test/HistoryProviderTest.cs

@@ -23,38 +23,65 @@ namespace NzbDrone.Core.Test
         public void AllItems()
         {
             //Setup
-            var episode = new Episode
-                              {
-                                  AirDate = DateTime.Today.AddDays(-1),
-                                  EpisodeId = 1234,
-                                  EpisodeNumber = 5,
-                                  Overview = "This is an Overview",
-                                  SeasonNumber = 1,
-                                  SeriesId = 5656
-                              };
+            var historyItem = Builder<History>.CreateListOfSize(10).Build();
 
-            var list = new List<History>
-                           {
-                               new History
-                                   {
-                                       HistoryId = new int(),
-                                       Date = DateTime.Now,
-                                       IsProper = false,
-                                       Quality = QualityTypes.SDTV,
-                                       EpisodeId = episode.EpisodeId
-                                   }
-                           };
+            var mocker = new AutoMoqer();
+            var db = MockLib.GetEmptyDatabase();
+            mocker.SetConstant(db);
 
-            var repo = new Mock<IRepository>();
-            repo.Setup(r => r.All<History>()).Returns(list.AsQueryable());
+            db.InsertMany(historyItem);
 
-            var target = new HistoryProvider(repo.Object);
 
             //Act
-            var result = target.AllItems();
+            var result = mocker.Resolve<HistoryProvider>().AllItems();
 
             //Assert
-            Assert.AreEqual(result.Count(), 1);
+            result.Should().HaveSameCount(historyItem);
+        }
+
+        [Test]
+        public void PurgeItem()
+        {
+            //Setup
+            var historyItem = Builder<History>.CreateListOfSize(10).Build();
+
+            var mocker = new AutoMoqer();
+            var db = MockLib.GetEmptyDatabase();
+            mocker.SetConstant(db);
+
+            db.InsertMany(historyItem);
+
+
+            //Act
+            db.Fetch<History>().Should().HaveCount(10);
+            mocker.Resolve<HistoryProvider>().Purge();
+
+            //Assert
+            db.Fetch<History>().Should().HaveCount(0);
+        }
+
+        [Test]
+        public void Trim_Items()
+        {
+            //Setup
+            var historyItem = Builder<History>.CreateListOfSize(20)
+                .WhereTheFirst(10).Have(c => c.Date = DateTime.Now)
+                .AndTheNext(10).Have(c => c.Date = DateTime.Now.AddDays(-31))
+                .Build();
+
+            var mocker = new AutoMoqer();
+            var db = MockLib.GetEmptyDatabase();
+            mocker.SetConstant(db);
+
+            db.InsertMany(historyItem);
+
+
+            //Act
+            db.Fetch<History>().Should().HaveCount(20);
+            mocker.Resolve<HistoryProvider>().Trim();
+
+            //Assert
+            db.Fetch<History>().Should().HaveCount(10);
         }
 
 
@@ -63,7 +90,7 @@ namespace NzbDrone.Core.Test
         {
             var mocker = new AutoMoqer(MockBehavior.Strict);
 
-            mocker.SetConstant(MockLib.GetEmptyRepository());
+            mocker.SetConstant(MockLib.GetEmptyDatabase());
 
             //Act
             var result = mocker.Resolve<HistoryProvider>().GetBestQualityInHistory(12);
@@ -77,12 +104,12 @@ namespace NzbDrone.Core.Test
         {
             var mocker = new AutoMoqer(MockBehavior.Strict);
 
-
-            var repo = MockLib.GetEmptyRepository();
+            var db = MockLib.GetEmptyDatabase();
             var history = Builder<History>.CreateNew()
                 .With(h => h.Quality = QualityTypes.Bluray720p).Build();
-            repo.Add(history);
-            mocker.SetConstant(repo);
+
+            db.Insert(history);
+            mocker.SetConstant(db);
 
             //Act
             var result = mocker.Resolve<HistoryProvider>().GetBestQualityInHistory(history.EpisodeId);
@@ -97,35 +124,40 @@ namespace NzbDrone.Core.Test
         {
             //Arange
             var mocker = new AutoMoqer();
-            var repo = MockLib.GetEmptyRepository();
+            var db = MockLib.GetEmptyDatabase();
 
-            mocker.SetConstant(repo);
+            mocker.SetConstant(db);
 
-            var episodes = MockLib.GetFakeEpisodes(1);
-            repo.AddMany(episodes);
+            var episode = Builder<Episode>.CreateNew().Build();
 
-            var episode = episodes[5];
+            const QualityTypes quality = QualityTypes.HDTV;
+            const bool proper = true;
 
             var history = new History
                               {
                                   Date = DateTime.Now,
                                   EpisodeId = episode.EpisodeId,
+                                  SeriesId = episode.SeriesId,
                                   NzbTitle = "my title",
-                                  Indexer = "Fake Indexer"
+                                  Indexer = "Fake Indexer",
+                                  Quality = quality,
+                                  IsProper = proper
                               };
 
             //Act
             mocker.Resolve<HistoryProvider>().Add(history);
 
             //Assert
-            var storedHistory = repo.All<History>();
-            var newHistiory = repo.All<History>().First();
+            var storedHistory = db.Fetch<History>();
 
             storedHistory.Should().HaveCount(1);
-            Assert.AreEqual(history.Date, newHistiory.Date);
-            Assert.AreEqual(history.EpisodeId, newHistiory.EpisodeId);
-            Assert.AreEqual(history.NzbTitle, newHistiory.NzbTitle);
-            Assert.AreEqual(history.Indexer, newHistiory.Indexer);
+            Assert.AreEqual(history.Date, storedHistory.First().Date);
+            Assert.AreEqual(history.EpisodeId, storedHistory.First().EpisodeId);
+            Assert.AreEqual(history.SeriesId, storedHistory.First().SeriesId);
+            Assert.AreEqual(history.NzbTitle, storedHistory.First().NzbTitle);
+            Assert.AreEqual(history.Indexer, storedHistory.First().Indexer);
+            Assert.AreEqual(history.Quality, storedHistory.First().Quality);
+            Assert.AreEqual(history.IsProper, storedHistory.First().IsProper);
         }
 
 

NzbDrone.Core/Providers/HistoryProvider.cs

@@ -5,47 +5,46 @@ using Ninject;
 using NLog;
 using NzbDrone.Core.Model;
 using NzbDrone.Core.Repository;
-using NzbDrone.Core.Repository.Quality;
-using SubSonic.Repository;
+using PetaPoco;
 
 namespace NzbDrone.Core.Providers
 {
     public class HistoryProvider
     {
+        private readonly IDatabase _database;
         private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
-        private readonly IRepository _repository;
+
 
         [Inject]
-        public HistoryProvider(IRepository repository)
+        public HistoryProvider(IDatabase database)
         {
-            _repository = repository;
+            _database = database;
         }
 
         public HistoryProvider()
         {
         }
 
-        public virtual IQueryable<History> AllItems()
+        public virtual IEnumerable<History> AllItems()
         {
-            return _repository.All<History>();
+            return _database.Query<History>("");
         }
 
         public virtual void Purge()
         {
-            _repository.DeleteMany(AllItems());
+            _database.Delete<History>("");
             Logger.Info("History has been Purged");
         }
 
         public virtual void Trim()
         {
-            var old = AllItems().Where(h => h.Date < DateTime.Now.AddDays(-30));
-            _repository.DeleteMany(old);
+            _database.Delete<History>("WHERE Date < @0", DateTime.Now.AddDays(-30).Date);
             Logger.Info("History has been trimmed, items older than 30 days have been removed");
         }
 
         public virtual void Add(History item)
         {
-            _repository.Add(item);
+            _database.Insert(item);
             Logger.Debug("Item added to history: {0}", item.NzbTitle);
         }
 
@@ -55,15 +54,5 @@ namespace NzbDrone.Core.Providers
 
             return history.FirstOrDefault();
         }
-
-        public virtual void Delete(int historyId)
-        {
-            _repository.Delete<History>(historyId);
-        }
-
-        public virtual void DeleteForEpisode(int episodeId)
-        {
-            _repository.DeleteMany<History>(h => h.EpisodeId == episodeId);
-        }
     }
 }

NzbDrone.Core/Repository/History.cs

@@ -1,24 +1,22 @@
 using System;
 using NzbDrone.Core.Model;
 using NzbDrone.Core.Repository.Quality;
+using PetaPoco;
 using SubSonic.SqlGeneration.Schema;
 
 namespace NzbDrone.Core.Repository
 {
+    [PrimaryKey("HistoryId")]
     public class History
     {
-        [SubSonicPrimaryKey]
-        public virtual int HistoryId { get; set; }
+        public int HistoryId { get; set; }
 
-        public virtual int EpisodeId { get; set; }
+        public int EpisodeId { get; set; }
+        public int SeriesId { get; set; }
         public string NzbTitle { get; set; }
         public QualityTypes Quality { get; set; }
         public DateTime Date { get; set; }
         public bool IsProper { get; set; }
         public string Indexer { get; set; }
-        
-        [SubSonicToOneRelation(ThisClassContainsJoinKey = true)]
-        public virtual Episode Episode { get; protected set; }
-
     }
 }

NzbDrone.Web/Controllers/HistoryController.cs

@@ -12,10 +12,12 @@ namespace NzbDrone.Web.Controllers
     public class HistoryController : Controller
     {
         private readonly HistoryProvider _historyProvider;
+        private readonly EpisodeProvider _episodeProvider;
 
-        public HistoryController(HistoryProvider historyProvider)
+        public HistoryController(HistoryProvider historyProvider, EpisodeProvider episodeProvider)
         {
             _historyProvider = historyProvider;
+            _episodeProvider = episodeProvider;
         }
 
         //
@@ -48,12 +50,11 @@ namespace NzbDrone.Web.Controllers
             var historyDb = _historyProvider.AllItems().ToList();
             
             var history = new List<HistoryModel>();
-            
+
             foreach (var item in historyDb)
             {
-                var episode = item.Episode;
-                var series = episode.Series;
-
+                var episode = _episodeProvider.GetEpisode(item.EpisodeId);
+       
                 history.Add(new HistoryModel
                                             {
                                                 HistoryId = item.HistoryId,
@@ -61,7 +62,7 @@ namespace NzbDrone.Web.Controllers
                                                 EpisodeNumber = episode.EpisodeNumber,
                                                 EpisodeTitle = episode.Title,
                                                 EpisodeOverview = episode.Overview,
-                                                SeriesTitle = series.Title,
+                                                SeriesTitle = episode.Series.Title,
                                                 NzbTitle = item.NzbTitle,
                                                 Quality = item.Quality.ToString(),
                                                 IsProper = item.IsProper,