sonarr-repo-only/NzbDrone.Core.Test/TvTests/EpisodeProviderTests/EpisodeProviderTest.cs

1518 lines
56 KiB
C#
Raw Normal View History

2013-03-02 18:25:39 +00:00
/*
2013-03-28 22:07:09 +00:00
using System;
using System.Collections.Generic;
using System.Linq;
using FizzWare.NBuilder;
using FluentAssertions;
using Moq;
using NUnit.Framework;
2013-02-24 06:48:52 +00:00
using NzbDrone.Core.Configuration;
using NzbDrone.Core.Download;
2013-03-01 07:03:41 +00:00
using NzbDrone.Core.MediaFiles;
2013-02-27 03:19:22 +00:00
using NzbDrone.Core.Qualities;
using NzbDrone.Core.Tv;
using NzbDrone.Core.Model;
using NzbDrone.Core.Providers;
using NzbDrone.Core.Providers.Core;
using NzbDrone.Core.Test.Framework;
2013-03-02 18:25:39 +00:00
using TvdbLib.Data;
namespace NzbDrone.Core.Test.TvTests.EpisodeProviderTests
{
[TestFixture]
2013-03-28 22:07:09 +00:00
2013-03-02 18:25:39 +00:00
public class EpisodeProviderTest : ObjectDbTest
{
[Test]
public void GetEpisodes_exists()
{
2013-03-02 18:25:39 +00:00
var fakeSeries = Builder<Series>.CreateNew().Build();
var fakeEpisodes = Builder<Episode>.CreateListOfSize(5)
.All().With(e => e.SeriesId = 1).Build();
Db.Insert(fakeSeries);
Db.InsertMany(fakeEpisodes);
2013-03-28 22:07:09 +00:00
var episode = Mocker.Resolve<EpisodeService>().GetEpisode(1);
2013-03-28 22:07:09 +00:00
episode.ShouldHave().AllPropertiesBut(e => e.Series, e => e.EpisodeFile).EqualTo(fakeEpisodes.First());
episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries);
}
[Test]
public void GetEpisodes_by_season_episode_exists()
{
2013-03-02 18:25:39 +00:00
var fakeSeries = Builder<Series>.CreateNew()
.With(s => s.Id = 1)
.Build();
var fakeEpisodes = Builder<Episode>.CreateNew()
.With(e => e.SeriesId = 1)
.With(e => e.EpisodeNumber = 1)
.And(e => e.SeasonNumber = 2).Build();
Db.Insert(fakeSeries);
Db.Insert(fakeEpisodes);
2013-03-28 22:07:09 +00:00
var episode = Mocker.Resolve<EpisodeService>().GetEpisode(fakeSeries.Id, 2, 1);
2013-03-28 22:07:09 +00:00
episode.ShouldHave().AllPropertiesBut(e => e.Series).EqualTo(fakeEpisodes);
episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries);
}
[Test]
public void GetEpisodes_by_season_episode_doesnt_exists()
{
2013-03-02 18:25:39 +00:00
2013-03-28 22:07:09 +00:00
var episode = Mocker.Resolve<EpisodeService>().GetEpisode(1, 1, 1);
2013-03-28 22:07:09 +00:00
episode.Should().BeNull();
}
[Test]
public void GetEpisode_with_EpisodeFile()
{
2013-03-02 18:25:39 +00:00
var fakeSeries = Builder<Series>.CreateNew().Build();
2013-03-01 07:03:41 +00:00
var fakeFile = Builder<EpisodeFile>.CreateNew().With(f => f.Id).With(c => c.Quality = Quality.SDTV).Build();
var fakeEpisodes = Builder<Episode>.CreateListOfSize(5)
2013-03-01 07:03:41 +00:00
.All().With(e => e.SeriesId = 1).TheFirst(1).With(e => e.EpisodeFile = new EpisodeFile { Id = 1 }).With(e => e.EpisodeFile = fakeFile).Build();
Db.Insert(fakeSeries);
Db.InsertMany(fakeEpisodes);
Db.Insert(fakeFile);
2013-03-28 22:07:09 +00:00
var episode = Mocker.Resolve<EpisodeService>().GetEpisode(1);
2013-03-28 22:07:09 +00:00
episode.ShouldHave().AllPropertiesBut(e => e.Series, e => e.EpisodeFile).EqualTo(fakeEpisodes.First());
episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries);
episode.EpisodeFile.Should().NotBeNull();
}
[Test]
[ExpectedException(typeof(InvalidOperationException), ExpectedMessage = "Sequence contains no elements")]
public void GetEpisodes_invalid_series()
{
2013-03-02 18:25:39 +00:00
Mocker.Resolve<SeriesService>();
var fakeEpisodes = Builder<Episode>.CreateListOfSize(5)
.All().With(e => e.SeriesId = 1).Build();
Db.InsertMany(fakeEpisodes);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().GetEpisode(1);
}
[Test]
public void GetEpisodesBySeason_success()
{
2013-03-02 18:25:39 +00:00
var fakeSeries = Builder<Series>.CreateNew()
.With(s => s.Id = 12)
.Build();
var episodes = Builder<Episode>.CreateListOfSize(10)
.All().With(c => c.SeriesId = 12).And(c => c.SeasonNumber = 2)
.TheFirst(5).With(c => c.SeasonNumber = 1)
.Build();
Db.Insert(fakeSeries);
Db.InsertMany(episodes);
2013-03-28 22:07:09 +00:00
var seasonEposodes = Mocker.Resolve<EpisodeService>().GetEpisodesBySeason(12, 2);
2013-03-28 22:07:09 +00:00
Db.Fetch<Episode>().Should().HaveCount(10);
seasonEposodes.Should().HaveCount(5);
}
[Test]
public void RefreshEpisodeInfo_emptyRepo()
{
//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).
All()
.With(l => l.Language = new TvdbLanguage(0, "eng", "a"))
.Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
2013-03-02 18:25:39 +00:00
Db.Insert(fakeSeries);
Mocker.GetMock<TvDbProvider>()
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(fakeEpisodes);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
var actualCount = Mocker.Resolve<EpisodeService>().GetEpisodeBySeries(seriesId).ToList().Count;
Mocker.GetMock<TvDbProvider>().VerifyAll();
actualCount.Should().Be(episodeCount);
}
[Test]
public void RefreshEpisodeInfo_should_set_older_than_1900_to_null()
{
//Arrange
const int seriesId = 71663;
var fakeEpisodes = Builder<TvdbSeries>.CreateNew().With(
c => c.Episodes =
new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(10).
All()
.With(l => l.Language = new TvdbLanguage(0, "eng", "a")).And(e => e.FirstAired = DateTime.Now)
.TheFirst(7).With(e => e.FirstAired = new DateTime(1800, 1, 1))
.Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
2013-03-02 18:25:39 +00:00
Db.Insert(fakeSeries);
Mocker.GetMock<TvDbProvider>()
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(fakeEpisodes);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
var storedEpisodes = Mocker.Resolve<EpisodeService>().GetEpisodeBySeries(seriesId).ToList();
storedEpisodes.Should().HaveCount(10);
storedEpisodes.Where(e => e.AirDate == null).Should().HaveCount(7);
storedEpisodes.Where(e => e.AirDate != null).Should().HaveCount(3);
}
[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.Id = seriesId).Build();
2013-03-02 18:25:39 +00:00
Db.Insert(fakeSeries);
Db.Insert(fakeEpisode);
Mocker.GetMock<TvDbProvider>()
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(fakeTvDbEpisodes);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
var storedEpisodes = Mocker.Resolve<EpisodeService>().GetEpisodeBySeries(seriesId).ToList();
storedEpisodes.Should().HaveCount(1);
storedEpisodes.Where(e => e.AirDate == null).Should().HaveCount(1);
}
[Test]
public void RefreshEpisodeInfo_ignore_episode_zero()
{
//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).
All()
.With(l => l.Language = new TvdbLanguage(0, "eng", "a"))
.TheFirst(1)
2011-10-23 05:39:14 +00:00
.With(e => e.EpisodeNumber = 0)
.With(e => e.SeasonNumber = 15)
.Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
2013-03-02 18:25:39 +00:00
Db.Insert(fakeSeries);
Mocker.GetMock<TvDbProvider>()
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(fakeEpisodes);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
var result = Mocker.Resolve<EpisodeService>().GetEpisodeBySeries(seriesId).ToList();
Mocker.GetMock<TvDbProvider>().VerifyAll();
result.Should().HaveCount(episodeCount);
result.Where(e => e.EpisodeNumber == 0 && e.SeasonNumber == 15).Single().Ignored.Should().BeTrue();
}
[Test]
public void RefreshEpisodeInfo_should_skip_future_episodes_with_no_title()
{
//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).
All()
.With(a => c.FirstAired = DateTime.Now.AddDays(-2))
.With(e => e.EpisodeName = "Something")
.TheFirst(3)
.With(e => e.EpisodeName = "")
.With(e => e.FirstAired = DateTime.Now.AddDays(10))
.Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
2013-03-02 18:25:39 +00:00
Db.Insert(fakeSeries);
Mocker.GetMock<TvDbProvider>()
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(fakeEpisodes);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
var result = Mocker.Resolve<EpisodeService>().GetEpisodeBySeries(seriesId).ToList();
result.Should().HaveCount(episodeCount - 3);
result.Should().OnlyContain(c => !string.IsNullOrWhiteSpace(c.Title) || c.AirDate < DateTime.Now);
}
[Test]
public void RefreshEpisodeInfo_should_skip_older_than_1900_year_episodes_with_no_title()
{
//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).
All()
.With(a => c.FirstAired = DateTime.Now.AddDays(-2))
.With(e => e.EpisodeName = "Something")
.TheFirst(3)
.With(e => e.EpisodeName = "")
.With(e => e.FirstAired = new DateTime(1889, 1, 1))
.Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
2013-03-02 18:25:39 +00:00
Db.Insert(fakeSeries);
Mocker.GetMock<TvDbProvider>()
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(fakeEpisodes);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
var result = Mocker.Resolve<EpisodeService>().GetEpisodeBySeries(seriesId).ToList();
result.Should().HaveCount(episodeCount - 3);
result.Should().OnlyContain(c => !string.IsNullOrWhiteSpace(c.Title) || c.AirDate < DateTime.Now);
}
[Test]
public void RefreshEpisodeInfo_should_add_future_episodes_with_title()
{
const int seriesId = 71663;
var fakeEpisodes = Builder<TvdbSeries>.CreateNew().With(
c => c.Episodes = new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(10).
All()
.With(a => a.FirstAired = DateTime.Now.AddDays(10))
.With(e => e.EpisodeName = "Something")
.Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
2013-03-02 18:25:39 +00:00
Db.Insert(fakeSeries);
Mocker.GetMock<TvDbProvider>()
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(fakeEpisodes);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
var result = Mocker.Resolve<EpisodeService>().GetEpisodeBySeries(seriesId).ToList();
result.Should().HaveSameCount(fakeEpisodes.Episodes);
}
[Test]
public void RefreshEpisodeInfo_should_add_old_episodes_with_no_title()
{
const int seriesId = 71663;
var fakeEpisodes = Builder<TvdbSeries>.CreateNew().With(
c => c.Episodes = new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(10).
All()
.With(a => a.FirstAired = DateTime.Now.AddDays(-10))
.With(e => e.EpisodeName = string.Empty)
.Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
2013-03-02 18:25:39 +00:00
Db.Insert(fakeSeries);
Mocker.GetMock<TvDbProvider>()
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(fakeEpisodes);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
var result = Mocker.Resolve<EpisodeService>().GetEpisodeBySeries(seriesId).ToList();
result.Should().HaveSameCount(fakeEpisodes.Episodes);
}
[Test]
public void RefreshEpisodeInfo_ignore_season_zero()
{
//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).
All()
.With(l => l.Language = new TvdbLanguage(0, "eng", "a"))
.With(e => e.SeasonNumber = 0)
.Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
2013-03-02 18:25:39 +00:00
Db.Insert(fakeSeries);
Mocker.GetMock<TvDbProvider>()
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(fakeEpisodes);
2013-02-24 20:24:31 +00:00
Mocker.GetMock<ISeasonRepository>()
.Setup(s => s.IsIgnored(seriesId, 0))
.Returns(true);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
var result = Mocker.Resolve<EpisodeService>().GetEpisodeBySeries(seriesId).ToList();
Mocker.GetMock<TvDbProvider>().VerifyAll();
result.Should().HaveCount(episodeCount);
result.Where(e => e.Ignored).Should().HaveCount(episodeCount);
}
[Test]
public void new_episodes_only_calls_Insert()
{
const int seriesId = 71663;
var tvdbSeries = Builder<TvdbSeries>.CreateNew()
.With(c => c.Episodes = new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(5).Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
var currentEpisodes = new List<Episode>();
Mocker.GetMock<TvDbProvider>(MockBehavior.Strict)
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(tvdbSeries);
Mocker.GetMock<IDatabase>()
.Setup(d => d.Fetch<Episode, Series, EpisodeFile>(It.IsAny<String>(), It.IsAny<Object[]>()))
.Returns(currentEpisodes);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
Mocker.GetMock<IDatabase>().Verify(c => c.InsertMany(It.Is<IEnumerable<Episode>>(l => l.Count() == 5)), Times.Once());
Mocker.GetMock<IDatabase>().Verify(c => c.Update(It.IsAny<IEnumerable<Episode>>()), Times.Never());
Mocker.VerifyAllMocks();
}
[Test]
public void existing_episodes_only_calls_Update()
{
const int seriesId = 71663;
var tvdbSeries = Builder<TvdbSeries>.CreateNew()
.With(c => c.Episodes = new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(5).Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
var currentEpisodes = new List<Episode>();
foreach (var tvDbEpisode in tvdbSeries.Episodes)
{
currentEpisodes.Add(new Episode { TvDbEpisodeId = tvDbEpisode.Id, Series = fakeSeries });
}
Mocker.GetMock<TvDbProvider>(MockBehavior.Strict)
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(tvdbSeries);
Mocker.GetMock<IDatabase>()
.Setup(d => d.Fetch<Episode, Series, EpisodeFile>(It.IsAny<String>(), It.IsAny<Object[]>()))
.Returns(currentEpisodes);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
Mocker.GetMock<IDatabase>().Verify(c => c.InsertMany(It.Is<IEnumerable<Episode>>(l => l.Count() == 0)), Times.Once());
Mocker.GetMock<IDatabase>().Verify(c => c.UpdateMany(It.Is<IEnumerable<Episode>>(l => l.Count() == 5)), Times.Once());
Mocker.VerifyAllMocks();
}
[Test]
public void should_try_to_get_existing_episode_using_tvdbid_first()
{
const int seriesId = 71663;
var fakeTvDbResult = Builder<TvdbSeries>.CreateNew()
.With(c => c.Id = seriesId)
.With(c => c.Episodes = new List<TvdbEpisode>(
Builder<TvdbEpisode>.CreateListOfSize(1)
.All().With(g => g.Id = 99)
.Build())
)
.Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
var fakeEpisodeList = new List<Episode> { new Episode { TvDbEpisodeId = 99, SeasonNumber = 10, EpisodeNumber = 10, Series = fakeSeries } };
Mocker.GetMock<IDatabase>()
.Setup(d => d.Fetch<Episode, Series, EpisodeFile>(It.IsAny<String>(), It.IsAny<Object[]>()))
.Returns(fakeEpisodeList);
Mocker.GetMock<TvDbProvider>()
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(fakeTvDbResult);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
Mocker.VerifyAllMocks();
Mocker.GetMock<IDatabase>().Verify(c => c.UpdateMany(fakeEpisodeList), Times.Once());
}
[Test]
public void should_try_to_get_existing_episode_using_tvdbid_first_then_season_episode()
{
const int seriesId = 71663;
var tvdbSeries = Builder<TvdbSeries>.CreateNew()
.With(c => c.Id = seriesId)
.With(c => c.Episodes = new List<TvdbEpisode>{
Builder<TvdbEpisode>.CreateNew()
.With(g => g.Id = 99)
.With(g => g.SeasonNumber = 4)
.With(g => g.EpisodeNumber = 15)
.With(g=>g.SeriesId = seriesId)
.Build()
})
.Build();
var localEpisode = Builder<Episode>.CreateNew()
.With(c => c.SeriesId = seriesId)
.With(c => c.SeasonNumber = 4)
.With(c => c.EpisodeNumber = 15)
.Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
Mocker.GetMock<TvDbProvider>(MockBehavior.Strict)
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(tvdbSeries);
Mocker.GetMock<IDatabase>()
.Setup(d => d.Fetch<Episode, Series, EpisodeFile>(It.IsAny<String>(), It.IsAny<Object[]>()))
.Returns(new List<Episode> { localEpisode });
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
Mocker.VerifyAllMocks();
Mocker.GetMock<IDatabase>().Verify(c => c.UpdateMany(new List<Episode> { localEpisode }), Times.Once());
}
[Test]
public void existing_episodes_keep_their_episodeId_file_id()
{
const int seriesId = 71663;
var tvdbSeries = Builder<TvdbSeries>.CreateNew()
.With(c => c.Episodes = new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(5).Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
var currentEpisodes = new List<Episode>();
foreach (var tvDbEpisode in tvdbSeries.Episodes)
{
currentEpisodes.Add(new Episode
{
TvDbEpisodeId = tvDbEpisode.Id,
Id = 99,
2013-03-01 07:03:41 +00:00
EpisodeFile = new EpisodeFile { Id = 69 },
Ignored = true,
Series = fakeSeries,
EpisodeNumber = tvDbEpisode.EpisodeNumber,
SeasonNumber = tvDbEpisode.SeasonNumber
});
}
Mocker.GetMock<TvDbProvider>(MockBehavior.Strict)
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(tvdbSeries);
var updatedEpisodes = new List<Episode>();
Mocker.GetMock<IDatabase>()
.Setup(d => d.Fetch<Episode, Series, EpisodeFile>(It.IsAny<String>(), It.IsAny<Object[]>()))
.Returns(currentEpisodes);
Mocker.GetMock<IDatabase>()
.Setup(c => c.UpdateMany(It.IsAny<IEnumerable<Episode>>()))
.Callback<IEnumerable<Episode>>(ep => updatedEpisodes = ep.ToList());
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
updatedEpisodes.Should().HaveSameCount(tvdbSeries.Episodes);
updatedEpisodes.Should().OnlyContain(c => c.Id == 99);
updatedEpisodes.Should().OnlyContain(c => c.EpisodeFileId == 69);
updatedEpisodes.Should().OnlyContain(c => c.Ignored == true);
}
[Test]
public void existing_episodes_remote_their_episodeId_file_id_when_episode_number_doesnt_match_tvdbid()
{
const int seriesId = 71663;
var tvdbSeries = Builder<TvdbSeries>.CreateNew()
.With(c => c.Episodes = new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(5).Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
var currentEpisodes = new List<Episode>();
foreach (var tvDbEpisode in tvdbSeries.Episodes)
{
currentEpisodes.Add(new Episode
{
TvDbEpisodeId = tvDbEpisode.Id,
Id = 99,
2013-03-01 07:03:41 +00:00
EpisodeFile = new EpisodeFile { Id = 69 },
Ignored = true,
Series = fakeSeries,
EpisodeNumber = tvDbEpisode.EpisodeNumber + 1,
SeasonNumber = tvDbEpisode.SeasonNumber
});
}
Mocker.GetMock<TvDbProvider>(MockBehavior.Strict)
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(tvdbSeries);
var updatedEpisodes = new List<Episode>();
Mocker.GetMock<IDatabase>()
.Setup(d => d.Fetch<Episode, Series, EpisodeFile>(It.IsAny<String>(), It.IsAny<Object[]>()))
.Returns(currentEpisodes);
Mocker.GetMock<IDatabase>()
.Setup(c => c.UpdateMany(It.IsAny<IEnumerable<Episode>>()))
.Callback<IEnumerable<Episode>>(ep => updatedEpisodes = ep.ToList());
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
updatedEpisodes.Should().OnlyContain(c => c.EpisodeFileId == 0);
}
[Test]
public void existing_episodes_remote_their_episodeId_file_id_when_season_number_doesnt_match_tvdbid()
{
const int seriesId = 71663;
var tvdbSeries = Builder<TvdbSeries>.CreateNew()
.With(c => c.Episodes = new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(5).Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
var currentEpisodes = new List<Episode>();
foreach (var tvDbEpisode in tvdbSeries.Episodes)
{
currentEpisodes.Add(new Episode
{
TvDbEpisodeId = tvDbEpisode.Id,
Id = 99,
2013-03-01 07:03:41 +00:00
EpisodeFile = new EpisodeFile { Id = 69 },
Ignored = true,
Series = fakeSeries,
EpisodeNumber = tvDbEpisode.EpisodeNumber,
SeasonNumber = tvDbEpisode.SeasonNumber + 1
});
}
Mocker.GetMock<TvDbProvider>(MockBehavior.Strict)
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(tvdbSeries);
var updatedEpisodes = new List<Episode>();
Mocker.GetMock<IDatabase>()
.Setup(d => d.Fetch<Episode, Series, EpisodeFile>(It.IsAny<String>(), It.IsAny<Object[]>()))
.Returns(currentEpisodes);
Mocker.GetMock<IDatabase>()
.Setup(c => c.UpdateMany(It.IsAny<IEnumerable<Episode>>()))
.Callback<IEnumerable<Episode>>(ep => updatedEpisodes = ep.ToList());
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
updatedEpisodes.Should().OnlyContain(c => c.EpisodeFileId == 0);
}
[Test]
public void RefreshEpisodeInfo_should_ignore_new_episode_for_ignored_season()
{
//Arrange
const int seriesId = 71663;
const int episodeCount = 2;
var fakeEpisode = Builder<Episode>.CreateNew()
.With(e => e.SeasonNumber = 5)
.With(e => e.EpisodeNumber = 1)
.With(e => e.TvDbEpisodeId = 11)
.With(e => e.SeriesId = seriesId)
.With(e => e.Ignored = true)
.Build();
var tvdbSeries = Builder<TvdbSeries>.CreateNew().With(
c => c.Episodes =
new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(episodeCount).
All()
.With(l => l.Language = new TvdbLanguage(0, "eng", "a"))
.With(e => e.SeasonNumber = 5)
.TheFirst(1)
.With(e => e.EpisodeNumber = 1)
.With(e => e.Id = 11)
.TheNext(1)
.With(e => e.EpisodeNumber = 2)
.With(e => e.Id = 22)
.Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
2013-03-02 18:25:39 +00:00
Db.Insert(fakeSeries);
Db.Insert(fakeEpisode);
Mocker.GetMock<TvDbProvider>()
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(tvdbSeries);
2013-02-24 20:24:31 +00:00
Mocker.GetMock<ISeasonRepository>()
.Setup(s => s.IsIgnored(seriesId, It.IsAny<int>()))
.Returns(true);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
var result = Mocker.Resolve<EpisodeService>().GetEpisodeBySeries(seriesId).ToList();
Mocker.GetMock<TvDbProvider>().VerifyAll();
result.Should().HaveCount(episodeCount);
result.Where(e => e.Ignored).Should().HaveCount(episodeCount);
}
[Test]
[Explicit]
public void Add_daily_show_episodes()
{
2013-03-02 18:25:39 +00:00
Mocker.Resolve<TvDbProvider>();
2013-02-24 19:39:31 +00:00
Mocker.GetMock<IConfigService>()
.Setup(e => e.DefaultQualityProfile).Returns(1);
Db.Insert(Builder<QualityProfile>.CreateNew().Build());
var seriesRepo = Mocker.Resolve<ISeriesRepository>();
const int tvDbSeriesId = 71256;
2013-03-28 22:07:09 +00:00
var seriesProvider = Mocker.Resolve<SeriesService>();
seriesProvider.AddSeries("Test Series", "c:\\test\\", tvDbSeriesId, 1, null);
var episodeProvider = Mocker.Resolve<EpisodeService>();
episodeProvider.RefreshEpisodeInfo(seriesRepo.Get(tvDbSeriesId));
2013-03-28 22:07:09 +00:00
var episodes = episodeProvider.GetEpisodeBySeries(tvDbSeriesId);
episodes.Should().NotBeEmpty();
}
[Test]
public void GetEpisode_by_Season_Episode_none_existing()
{
2013-03-02 18:25:39 +00:00
2013-03-28 22:07:09 +00:00
var episode = Mocker.Resolve<EpisodeService>().GetEpisode(1, 1, 1);
2013-03-28 22:07:09 +00:00
episode.Should().BeNull();
}
[Test]
public void GetEpisode_by_Season_Episode_with_EpisodeFile()
{
2013-03-02 18:25:39 +00:00
var fakeSeries = Builder<Series>.CreateNew().Build();
2013-03-01 07:03:41 +00:00
var fakeFile = Builder<EpisodeFile>.CreateNew().With(f => f.Id).With(c => c.Quality = Quality.SDTV).Build();
var fakeEpisodes = Builder<Episode>.CreateListOfSize(5)
2013-03-01 07:03:41 +00:00
.All().With(e => e.SeriesId = 1).TheFirst(1).With(c => c.EpisodeFile = new EpisodeFile { Id = 1 }).With(e => e.EpisodeFile = fakeFile).Build();
Db.Insert(fakeSeries);
Db.InsertMany(fakeEpisodes);
Db.Insert(fakeFile);
2013-03-28 22:07:09 +00:00
var episode = Mocker.Resolve<EpisodeService>().GetEpisode(1, 1, 1);
2013-03-28 22:07:09 +00:00
episode.ShouldHave().AllPropertiesBut(e => e.Series, e => e.EpisodeFile).EqualTo(fakeEpisodes.First());
episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries);
episode.EpisodeFile.Should().NotBeNull();
}
[Test]
public void GetEpisode_by_Season_Episode_without_EpisodeFile()
{
2013-03-02 18:25:39 +00:00
var fakeSeries = Builder<Series>.CreateNew().Build();
var fakeEpisodes = Builder<Episode>.CreateListOfSize(5)
.All().With(e => e.SeriesId = 1).TheFirst(1).Build();
Db.Insert(fakeSeries);
Db.InsertMany(fakeEpisodes);
2013-03-28 22:07:09 +00:00
var episode = Mocker.Resolve<EpisodeService>().GetEpisode(1, 1, 1);
2013-03-28 22:07:09 +00:00
episode.ShouldHave().AllPropertiesBut(e => e.Series).EqualTo(fakeEpisodes.First());
episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries);
episode.EpisodeFile.Should().BeNull();
}
[Test]
public void GetEpisode_by_AirDate_with_EpisodeFile()
{
2013-03-02 18:25:39 +00:00
var fakeSeries = Builder<Series>.CreateNew().Build();
2013-03-01 07:03:41 +00:00
var fakeFile = Builder<EpisodeFile>.CreateNew().With(f => f.Id).With(c => c.Quality = Quality.SDTV).Build();
var fakeEpisodes = Builder<Episode>.CreateListOfSize(5)
2013-03-01 07:03:41 +00:00
.All().With(e => e.SeriesId = 1).TheFirst(1).With(e => e.EpisodeFile = new EpisodeFile { Id = 1 }).With(e => e.EpisodeFile = fakeFile).Build();
Db.Insert(fakeSeries);
Db.InsertMany(fakeEpisodes);
Db.Insert(fakeFile);
2013-03-28 22:07:09 +00:00
var episode = Mocker.Resolve<EpisodeService>().GetEpisode(1, fakeEpisodes[0].AirDate.Value);
2013-03-28 22:07:09 +00:00
episode.ShouldHave().AllPropertiesBut(e => e.Series, e => e.EpisodeFile).EqualTo(fakeEpisodes.First());
episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries);
episode.EpisodeFile.Should().NotBeNull();
}
[Test]
public void GetEpisode_by_AirDate_without_EpisodeFile()
{
2013-03-02 18:25:39 +00:00
var fakeSeries = Builder<Series>.CreateNew().Build();
var fakeEpisodes = Builder<Episode>.CreateListOfSize(5)
2013-03-01 07:03:41 +00:00
.All().With(e => e.SeriesId = 1).TheFirst(1).With(e => e.EpisodeFile = new EpisodeFile { Id = 1 }).Build();
Db.InsertMany(fakeEpisodes);
Db.Insert(fakeSeries);
2013-03-28 22:07:09 +00:00
var episode = Mocker.Resolve<EpisodeService>().GetEpisode(1, fakeEpisodes[0].AirDate.Value);
2013-03-28 22:07:09 +00:00
episode.ShouldHave().AllPropertiesBut(e => e.Series).EqualTo(fakeEpisodes.First());
episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries);
episode.EpisodeFile.Should().BeNull();
}
[Test]
public void MarkEpisodeAsFetched()
{
var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
.All().With(e => e.GrabDate = null)
.Build();
var parseResult = new EpisodeParseResult() { Episodes = fakeEpisodes };
Mocker.Resolve<EpisodeService>().Handle(new EpisodeGrabbedEvent(parseResult));
Mocker.GetMock<IEpisodeRepository>().Verify(c=>c.Update(fakeEpisodes[0]),Times.Once());
Mocker.GetMock<IEpisodeRepository>().Verify(c=>c.Update(fakeEpisodes[1]),Times.Once());
}
[Test]
public void AddEpisode_episode_is_ignored_when_full_season_is_ignored()
{
2013-03-02 18:25:39 +00:00
var episodes = Builder<Episode>.CreateListOfSize(4)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
.With(c => c.Ignored = true)
.Build().ToList();
episodes.ForEach(c => Db.Insert(c));
var newEpisode = Builder<Episode>.CreateNew()
.With(e => e.SeriesId = 10)
.With(e => e.SeasonNumber = 1)
.With(e => e.EpisodeNumber = 8)
.With(e => e.SeasonNumber = 1)
.With(e => e.Ignored = false)
.Build();
2013-02-24 20:24:31 +00:00
Mocker.GetMock<ISeasonRepository>()
.Setup(s => s.IsIgnored(newEpisode.SeriesId, newEpisode.SeasonNumber))
.Returns(true);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().AddEpisode(newEpisode);
2013-03-28 22:07:09 +00:00
var episodesInDb = Db.Fetch<Episode>(@"SELECT * FROM Episodes");
episodesInDb.Should().HaveCount(5);
episodesInDb.Should().OnlyContain(e => e.Ignored);
Mocker.VerifyAllMocks();
}
[Test]
public void AddEpisode_episode_is_not_ignored_when_full_season_is_not_ignored()
{
2013-03-02 18:25:39 +00:00
var episodes = Builder<Episode>.CreateListOfSize(4)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
.With(c => c.Ignored = false)
.Build().ToList();
episodes.ForEach(c => Db.Insert(c));
var newEpisode = Builder<Episode>.CreateNew()
.With(e => e.SeriesId = 10)
.With(e => e.SeasonNumber = 1)
.With(e => e.EpisodeNumber = 8)
.With(e => e.SeasonNumber = 1)
.With(e => e.Ignored = false)
.Build();
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().AddEpisode(newEpisode);
2013-03-28 22:07:09 +00:00
var episodesInDb = Db.Fetch<Episode>(@"SELECT * FROM Episodes");
episodesInDb.Should().HaveCount(5);
episodesInDb.Should().OnlyContain(e => e.Ignored == false);
Mocker.VerifyAllMocks();
}
[Test]
public void AddEpisode_episode_is_not_ignored_when_not_full_season_is_not_ignored()
{
2013-03-02 18:25:39 +00:00
var episodes = Builder<Episode>.CreateListOfSize(4)
.All()
.With(c => c.SeriesId = 10)
.And(c => c.SeasonNumber = 1)
.And(c => c.Ignored = true)
.TheFirst(2)
.With(c => c.Ignored = false)
.Build().ToList();
episodes.ForEach(c => Db.Insert(c));
var newEpisode = Builder<Episode>.CreateNew()
.With(e => e.SeriesId = 10)
.With(e => e.SeasonNumber = 1)
.With(e => e.EpisodeNumber = 8)
.With(e => e.SeasonNumber = 1)
.With(e => e.Ignored = false)
.Build();
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().AddEpisode(newEpisode);
2013-03-28 22:07:09 +00:00
var episodesInDb = Db.Fetch<Episode>(@"SELECT * FROM Episodes");
episodesInDb.Should().HaveCount(5);
episodesInDb.Where(e => e.EpisodeNumber == 8 && !e.Ignored).Should().HaveCount(1);
Mocker.VerifyAllMocks();
}
[Test]
public void IgnoreEpisode_Ignore()
{
2013-03-02 18:25:39 +00:00
var episodes = Builder<Episode>.CreateListOfSize(4)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
.With(c => c.Ignored = false)
.Build().ToList();
episodes.ForEach(c => Db.Insert(c));
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().SetEpisodeIgnore(1, true);
2013-03-28 22:07:09 +00:00
var episodesInDb = Db.Fetch<Episode>(@"SELECT * FROM Episodes");
episodesInDb.Should().HaveCount(4);
episodesInDb.Where(e => e.Ignored).Should().HaveCount(1);
Mocker.VerifyAllMocks();
}
[Test]
public void IgnoreEpisode_RemoveIgnore()
{
2013-03-02 18:25:39 +00:00
var episodes = Builder<Episode>.CreateListOfSize(4)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
.With(c => c.Ignored = true)
.Build().ToList();
episodes.ForEach(c => Db.Insert(c));
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().SetEpisodeIgnore(1, false);
2013-03-28 22:07:09 +00:00
var episodesInDb = Db.Fetch<Episode>(@"SELECT * FROM Episodes");
episodesInDb.Should().HaveCount(4);
episodesInDb.Where(e => !e.Ignored).Should().HaveCount(1);
Mocker.VerifyAllMocks();
}
[Test]
public void EpisodesWithoutFiles_no_specials()
{
2013-03-02 18:25:39 +00:00
var series = Builder<Series>.CreateNew()
.With(s => s.Id = 10)
.Build();
var episodes = Builder<Episode>.CreateListOfSize(4)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
.With(c => c.AirDate = DateTime.Today.AddDays(-4))
.With(c => c.Ignored = true)
.TheFirst(2)
2011-10-23 05:39:14 +00:00
.Section(1, 2)
.With(c => c.Ignored = false)
.Build().ToList();
var specials = Builder<Episode>.CreateListOfSize(2)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 0)
.With(c => c.AirDate = DateTime.Today.AddDays(-4))
.With(c => c.Ignored = false)
.TheFirst(1).With(c => c.Ignored = true)
.Build().ToList();
Db.Insert(series);
Db.InsertMany(episodes);
Db.InsertMany(specials);
2013-03-28 22:07:09 +00:00
var missingFiles = Mocker.Resolve<EpisodeService>().EpisodesWithoutFiles(false);
2013-03-28 22:07:09 +00:00
missingFiles.Should().HaveCount(1);
missingFiles.Where(e => e.EpisodeFileId == 0).Should().HaveCount(1);
Mocker.VerifyAllMocks();
}
[Test]
public void EpisodesWithoutFiles_with_specials()
{
2013-03-02 18:25:39 +00:00
var series = Builder<Series>.CreateNew()
.With(s => s.Id = 10)
.Build();
var episodes = Builder<Episode>.CreateListOfSize(4)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
.With(c => c.AirDate = DateTime.Today.AddDays(-4))
.With(c => c.Ignored = true)
.TheFirst(2)
2011-10-23 05:39:14 +00:00
.Section(1, 2)
.With(c => c.Ignored = false)
.Build().ToList();
var specials = Builder<Episode>.CreateListOfSize(2)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 0)
.With(c => c.AirDate = DateTime.Today.AddDays(-4))
.With(c => c.Ignored = false)
.TheFirst(1)
.With(c => c.Ignored = true)
.Build().ToList();
Db.Insert(series);
Db.InsertMany(episodes);
Db.InsertMany(specials);
2013-03-28 22:07:09 +00:00
var missingFiles = Mocker.Resolve<EpisodeService>().EpisodesWithoutFiles(true);
2013-03-28 22:07:09 +00:00
missingFiles.Should().HaveCount(2);
missingFiles.Where(e => e.EpisodeFileId == 0).Should().HaveCount(2);
Mocker.VerifyAllMocks();
}
[Test]
public void EpisodesWithFiles_success()
{
2013-03-02 18:25:39 +00:00
var series = Builder<Series>.CreateNew()
.With(s => s.Id = 10)
.Build();
var episodeFile = Builder<EpisodeFile>.CreateNew()
2013-03-01 07:03:41 +00:00
.With(c => c.Id = 1)
2013-02-27 03:19:22 +00:00
.With(c => c.Quality = Quality.SDTV)
.Build();
var episodes = Builder<Episode>.CreateListOfSize(2)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
.With(c => c.AirDate = DateTime.Today.AddDays(-4))
.With(c => c.Ignored = true)
.With(c => c.EpisodeFile = episodeFile)
.Build().ToList();
Db.Insert(series);
Db.Insert(episodeFile);
Db.InsertMany(episodes);
2013-03-28 22:07:09 +00:00
var withFiles = Mocker.Resolve<EpisodeService>().EpisodesWithFiles();
2013-03-28 22:07:09 +00:00
withFiles.Should().HaveCount(2);
withFiles.Where(e => e.EpisodeFileId == 0).Should().HaveCount(0);
withFiles.Where(e => e.EpisodeFile == null).Should().HaveCount(0);
foreach (var withFile in withFiles)
{
withFile.EpisodeFile.Should().NotBeNull();
withFile.Series.Title.Should().NotBeNullOrEmpty();
}
Mocker.VerifyAllMocks();
}
[Test]
public void EpisodesWithFiles_no_files()
{
2013-03-02 18:25:39 +00:00
var series = Builder<Series>.CreateNew()
.With(s => s.Id = 10)
.Build();
var episodes = Builder<Episode>.CreateListOfSize(2)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
.With(c => c.AirDate = DateTime.Today.AddDays(-4))
.With(c => c.Ignored = true)
.Build().ToList();
Db.Insert(series);
Db.InsertMany(episodes);
2013-03-28 22:07:09 +00:00
var withFiles = Mocker.Resolve<EpisodeService>().EpisodesWithFiles();
2013-03-28 22:07:09 +00:00
withFiles.Should().HaveCount(0);
Mocker.VerifyAllMocks();
}
[Test]
public void GetEpisodesByFileId_multi_episodes()
{
2013-03-02 18:25:39 +00:00
var series = Builder<Series>.CreateNew()
.With(s => s.Id = 10)
.Build();
var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
2013-03-01 07:03:41 +00:00
.With(c => c.EpisodeFile = new EpisodeFile { Id = 12345 })
.Build();
Db.Insert(series);
Db.InsertMany(fakeEpisodes);
2013-03-28 22:07:09 +00:00
var episodes = Mocker.Resolve<EpisodeService>().GetEpisodesByFileId(12345);
2013-03-28 22:07:09 +00:00
episodes.Should().HaveCount(2);
Mocker.VerifyAllMocks();
}
[Test]
public void GetEpisodesByFileId_single_episode()
{
2013-03-02 18:25:39 +00:00
var series = Builder<Series>.CreateNew()
.With(s => s.Id = 10)
.Build();
var fakeEpisode = Builder<Episode>.CreateNew()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
2013-03-01 07:03:41 +00:00
.With(c => c.EpisodeFile = new EpisodeFile { Id = 12345 })
.Build();
Db.Insert(series);
Db.Insert(fakeEpisode);
2013-03-28 22:07:09 +00:00
var episodes = Mocker.Resolve<EpisodeService>().GetEpisodesByFileId(12345);
2013-03-28 22:07:09 +00:00
episodes.Should().HaveCount(1);
episodes.First().ShouldHave().AllPropertiesBut(e => e.Series).EqualTo(fakeEpisode);
Mocker.VerifyAllMocks();
}
[Test]
public void IsFirstOrLastEpisodeInSeason_false()
{
2013-03-02 18:25:39 +00:00
var fakeEpisodes = Builder<Episode>.CreateListOfSize(10)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
.Build();
Db.InsertMany(fakeEpisodes);
2013-03-28 22:07:09 +00:00
var result = Mocker.Resolve<EpisodeService>().IsFirstOrLastEpisodeOfSeason(10, 1, 5);
2013-03-28 22:07:09 +00:00
result.Should().BeFalse();
}
[Test]
public void IsFirstOrLastEpisodeInSeason_true_first()
{
2013-03-02 18:25:39 +00:00
var fakeEpisodes = Builder<Episode>.CreateListOfSize(10)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
.Build();
Db.InsertMany(fakeEpisodes);
2013-03-28 22:07:09 +00:00
var result = Mocker.Resolve<EpisodeService>().IsFirstOrLastEpisodeOfSeason(10, 1, 1);
2013-03-28 22:07:09 +00:00
result.Should().BeFalse();
}
[Test]
public void IsFirstOrLastEpisodeInSeason_true_last()
{
2013-03-02 18:25:39 +00:00
var fakeEpisodes = Builder<Episode>.CreateListOfSize(10)
.All()
.With(c => c.SeriesId = 10)
.With(c => c.SeasonNumber = 1)
.Build();
Db.InsertMany(fakeEpisodes);
2013-03-28 22:07:09 +00:00
var result = Mocker.Resolve<EpisodeService>().IsFirstOrLastEpisodeOfSeason(10, 1, 10);
2013-03-28 22:07:09 +00:00
result.Should().BeFalse();
}
[TestCase("The Office (US) - S01E01 - Episode Title", PostDownloadStatusType.Unpacking, 1)]
[TestCase("The Office (US) - S01E01 - Episode Title", PostDownloadStatusType.Failed, 1)]
[TestCase("The Office (US) - S01E01E02 - Episode Title", PostDownloadStatusType.Unpacking, 2)]
[TestCase("The Office (US) - S01E01E02 - Episode Title", PostDownloadStatusType.Failed, 2)]
[TestCase("The Office (US) - Season 01 - Episode Title", PostDownloadStatusType.Unpacking, 10)]
[TestCase("The Office (US) - Season 01 - Episode Title", PostDownloadStatusType.Failed, 10)]
public void SetPostDownloadStatus(string folderName, PostDownloadStatusType postDownloadStatus, int episodeCount)
{
2013-03-02 18:25:39 +00:00
var fakeSeries = Builder<Series>.CreateNew()
.With(s => s.Id = 12345)
.With(s => s.CleanTitle = "officeus")
.Build();
var fakeEpisodes = Builder<Episode>.CreateListOfSize(episodeCount)
.All()
.With(c => c.SeriesId = 12345)
.With(c => c.SeasonNumber = 1)
.With(c => c.PostDownloadStatus = PostDownloadStatusType.Unknown)
.Build();
Db.Insert(fakeSeries);
Db.InsertMany(fakeEpisodes);
Mocker.GetMock<ISeriesRepository>().Setup(s => s.GetByTitle("officeus")).Returns(fakeSeries);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().SetPostDownloadStatus(fakeEpisodes.Select(e => e.Id).ToList(), postDownloadStatus);
2013-03-28 22:07:09 +00:00
var result = Db.Fetch<Episode>();
result.Where(e => e.PostDownloadStatus == postDownloadStatus).Count().Should().Be(episodeCount);
}
[Test]
public void SetPostDownloadStatus_Invalid_EpisodeId()
{
2013-03-02 18:25:39 +00:00
var postDownloadStatus = PostDownloadStatusType.Failed;
var fakeSeries = Builder<Series>.CreateNew()
.With(s => s.Id = 12345)
.With(s => s.CleanTitle = "officeus")
.Build();
var fakeEpisodes = Builder<Episode>.CreateListOfSize(1)
.All()
.With(c => c.SeriesId = 12345)
.With(c => c.SeasonNumber = 1)
.With(c => c.PostDownloadStatus = PostDownloadStatusType.Unknown)
.Build();
Db.Insert(fakeSeries);
Db.InsertMany(fakeEpisodes);
Mocker.GetMock<ISeriesRepository>().Setup(s => s.GetByTitle("officeus")).Returns(fakeSeries);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().SetPostDownloadStatus(new List<int> { 300 }, postDownloadStatus);
2013-03-28 22:07:09 +00:00
var result = Db.Fetch<Episode>();
result.Where(e => e.PostDownloadStatus == postDownloadStatus).Count().Should().Be(0);
}
[Test]
[ExpectedException(typeof(ArgumentException))]
public void SetPostDownloadStatus_should_throw_if_episode_list_is_empty()
{
Mocker.Resolve<EpisodeService>().SetPostDownloadStatus(new List<int>(), PostDownloadStatusType.Failed);
}
[Test]
public void RefreshEpisodeInfo_should_ignore_episode_zero_except_if_season_one()
{
//Arrange
const int seriesId = 71663;
const int episodeCount = 5;
var tvdbSeries = Builder<TvdbSeries>.CreateNew().With(
c => c.Episodes =
new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(episodeCount).
All()
.With(l => l.Language = new TvdbLanguage(0, "eng", "a"))
.With(e => e.EpisodeNumber = 0)
.TheFirst(1)
.With(e => e.SeasonNumber = 1)
.TheNext(1)
.With(e => e.SeasonNumber = 2)
.TheNext(1)
.With(e => e.SeasonNumber = 3)
.TheNext(1)
.With(e => e.SeasonNumber = 4)
.TheNext(1)
.With(e => e.SeasonNumber = 5)
.Build())
).With(c => c.Id = seriesId).Build();
var fakeSeries = Builder<Series>.CreateNew().With(c => c.Id = seriesId).Build();
2013-03-02 18:25:39 +00:00
Db.Insert(fakeSeries);
Mocker.GetMock<TvDbProvider>()
.Setup(c => c.GetSeries(seriesId, true, false))
.Returns(tvdbSeries);
2013-03-28 22:07:09 +00:00
Mocker.Resolve<EpisodeService>().RefreshEpisodeInfo(fakeSeries);
2013-03-28 22:07:09 +00:00
var result = Mocker.Resolve<EpisodeService>().GetEpisodeBySeries(seriesId).ToList();
result.Should().HaveCount(episodeCount);
result.Where(e => e.Ignored).Should().HaveCount(episodeCount - 1);
result.Single(e => e.SeasonNumber == 1).Ignored.Should().BeFalse();
}
}
}
2013-03-02 18:25:39 +00:00
*/