101 lines
3.0 KiB
C#
101 lines
3.0 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using AutoMoq;
|
|
using FizzWare.NBuilder;
|
|
using Gallio.Framework;
|
|
using MbUnit.Framework;
|
|
using MbUnit.Framework.ContractVerifiers;
|
|
using Moq;
|
|
using Ninject;
|
|
using Ninject.Moq;
|
|
using NzbDrone.Core.Providers;
|
|
using NzbDrone.Core.Repository;
|
|
using NzbDrone.Core.Repository.Quality;
|
|
using SubSonic.Repository;
|
|
using TvdbLib.Data;
|
|
using SubSonic.Extensions;
|
|
|
|
namespace NzbDrone.Core.Test
|
|
{
|
|
[TestFixture]
|
|
// ReSharper disable InconsistentNaming
|
|
public class EpisodeProviderTest
|
|
{
|
|
[Test]
|
|
public void RefreshEpisodeInfo()
|
|
{
|
|
//Arrange
|
|
const int seriesId = 71663;
|
|
const int episodeCount = 10;
|
|
|
|
var fakeEpisodes = Builder<TvdbSeries>.CreateNew().With(
|
|
c => c.Episodes =
|
|
new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(episodeCount).
|
|
WhereAll()
|
|
.Have(l => l.Language = new TvdbLanguage(0, "eng", "a"))
|
|
.Build())
|
|
).With(c => c.Id = seriesId).Build();
|
|
|
|
var mocker = new AutoMoqer();
|
|
|
|
mocker.SetConstant(MockLib.GetEmptyRepository());
|
|
|
|
mocker.GetMock<TvDbProvider>()
|
|
.Setup(c => c.GetSeries(seriesId, true))
|
|
.Returns(fakeEpisodes).Verifiable();
|
|
|
|
//mocker.GetMock<IRepository>().SetReturnsDefault();
|
|
|
|
|
|
|
|
//Act
|
|
var sw = Stopwatch.StartNew();
|
|
mocker.Resolve<EpisodeProvider>().RefreshEpisodeInfo(seriesId);
|
|
var actualCount = mocker.Resolve<EpisodeProvider>().GetEpisodeBySeries(seriesId);
|
|
//Assert
|
|
mocker.GetMock<TvDbProvider>().VerifyAll();
|
|
Assert.Count(episodeCount, actualCount);
|
|
Console.WriteLine("Duration: " + sw.Elapsed);
|
|
}
|
|
|
|
[Test]
|
|
public void IsNeededTrue()
|
|
{
|
|
//Setup
|
|
var season = new Mock<ISeasonProvider>();
|
|
var series = new Mock<ISeriesProvider>();
|
|
//var history = new Mock<IHistoryProvider>();
|
|
//var quality = new Mock<IQualityProvider>();
|
|
var repo = new Mock<IRepository>();
|
|
|
|
var epInDb = new Episode
|
|
{
|
|
AirDate = DateTime.Today,
|
|
EpisodeId = 55555,
|
|
EpisodeNumber = 5,
|
|
Language = "en",
|
|
SeasonId = 4444,
|
|
SeasonNumber = 1
|
|
};
|
|
|
|
season.Setup(s => s.IsIgnored(12345, 1)).Returns(false);
|
|
series.Setup(s => s.QualityWanted(12345, QualityTypes.TV)).Returns(true);
|
|
repo.Setup(s => s.Single<Episode>(c => c.SeriesId == 12345 && c.SeasonNumber == 1 && c.EpisodeNumber == 5)).
|
|
Returns(epInDb);
|
|
|
|
//repo.Setup(s => s.All<EpisodeFile>()).Returns();
|
|
//repo.All<EpisodeFile>().Where(c => c.EpisodeId == episode.EpisodeId);
|
|
|
|
|
|
|
|
//Act
|
|
|
|
|
|
//Assert
|
|
}
|
|
}
|
|
}
|