Archive-Team
/
sonarr-repo-only
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update episode info now uses UpdateMany, InsertMany

This commit is contained in:
kay.one 2011-08-28 20:03:40 -07:00
parent 2980c91f19
commit 9244a4e6ef
2 changed files with 12 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
NzbDrone.Core.Test/EpisodeProviderTest.cs
View File

@ -388,8 +388,8 @@ namespace NzbDrone.Core.Test
mocker.Resolve<EpisodeProvider>().RefreshEpisodeInfo(fakeSeries); mocker.Resolve<EpisodeProvider>().RefreshEpisodeInfo(fakeSeries);
//Assert //Assert
mocker.GetMock<IDatabase>().Verify(c => c.Insert(It.IsAny<Object>()), Times.Never()); mocker.GetMock<IDatabase>().Verify(c => c.InsertMany(It.Is<IEnumerable<Episode>>(l => l.Count() == 0)), Times.Once());
mocker.GetMock<IDatabase>().Verify(c => c.Update(It.IsAny<Object>()), Times.Exactly(tvdbSeries.Episodes.Count)); mocker.GetMock<IDatabase>().Verify(c => c.UpdateMany(It.Is<IEnumerable<Episode>>(l => l.Count() == 5)), Times.Once());
mocker.VerifyAllMocks(); mocker.VerifyAllMocks();
} }
@ -499,9 +499,8 @@ namespace NzbDrone.Core.Test
.Returns(currentEpisodes); .Returns(currentEpisodes);
mocker.GetMock<IDatabase>() mocker.GetMock<IDatabase>()
.Setup(c => c.Update(It.IsAny<Episode>())) .Setup(c => c.UpdateMany(It.IsAny<IEnumerable<Episode>>()))
.Returns(1) .Callback<IEnumerable<Episode>>(ep => updatedEpisodes =ep.ToList());
.Callback<Episode>(ep => updatedEpisodes.Add(ep));
//Act //Act
mocker.Resolve<EpisodeProvider>().RefreshEpisodeInfo(fakeSeries); mocker.Resolve<EpisodeProvider>().RefreshEpisodeInfo(fakeSeries);

13
NzbDrone.Core/Providers/EpisodeProvider.cs
View File

@ -241,17 +241,8 @@ namespace NzbDrone.Core.Providers
} }
} }
using (var tran = _database.GetTransaction()) _database.InsertMany(newList);
{ _database.UpdateMany(updateList);
newList.ForEach(AddEpisode);
updateList.ForEach(episode => _database.Update(episode));
//Shouldn't run if Database is a mock since transaction will be null
if (_database.GetType().Namespace != "Castle.Proxies" && tran != null)
{
tran.Complete();
}
}
Logger.Info("Finished episode refresh for series: {0}. Successful: {1} - Failed: {2} ", Logger.Info("Finished episode refresh for series: {0}. Successful: {1} - Failed: {2} ",
tvDbSeriesInfo.SeriesName, successCount, failCount); tvDbSeriesInfo.SeriesName, successCount, failCount);