2013-03-02 18:25:39 +00:00
|
|
|
|
/*
|
2013-03-28 22:07:09 +00:00
|
|
|
|
|
2011-10-20 23:42:17 +00:00
|
|
|
|
|
2011-11-24 08:28:20 +00:00
|
|
|
|
using System;
|
2011-06-22 01:12:20 +00:00
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using FizzWare.NBuilder;
|
|
|
|
|
using FluentAssertions;
|
|
|
|
|
using NUnit.Framework;
|
2013-02-19 06:01:03 +00:00
|
|
|
|
using NzbDrone.Core.Tv;
|
2011-06-22 01:12:20 +00:00
|
|
|
|
using NzbDrone.Core.Model;
|
|
|
|
|
using NzbDrone.Core.Test.Framework;
|
2011-12-20 00:45:47 +00:00
|
|
|
|
using NzbDrone.Test.Common;
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2013-02-19 06:01:03 +00:00
|
|
|
|
namespace NzbDrone.Core.Test.TvTests.EpisodeProviderTests
|
2011-06-22 01:12:20 +00:00
|
|
|
|
{
|
|
|
|
|
[TestFixture]
|
2013-03-28 22:07:09 +00:00
|
|
|
|
|
2013-03-02 18:25:39 +00:00
|
|
|
|
public class EpisodeProviderTest_GetEpisodesByParseResult : ObjectDbTest
|
2011-06-22 01:12:20 +00:00
|
|
|
|
{
|
2013-02-22 00:47:09 +00:00
|
|
|
|
private IEpisodeService episodeService;
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
private Series fakeSeries;
|
|
|
|
|
private Series fakeDailySeries;
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
private Episode fakeEpisode;
|
|
|
|
|
private Episode fakeDailyEpisode;
|
|
|
|
|
private Episode fakeEpisode2;
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
[SetUp]
|
|
|
|
|
public void Setup()
|
2011-06-22 01:12:20 +00:00
|
|
|
|
{
|
2012-01-06 21:25:27 +00:00
|
|
|
|
fakeSeries = Builder<Series>.CreateNew().Build();
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
fakeDailySeries = Builder<Series>.CreateNew()
|
2013-02-20 02:05:15 +00:00
|
|
|
|
.With(c => c.SeriesType = SeriesType.Daily)
|
2012-01-06 21:25:27 +00:00
|
|
|
|
.Build();
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
fakeEpisode = Builder<Episode>.CreateNew()
|
2013-02-26 03:58:57 +00:00
|
|
|
|
.With(e => e.SeriesId = fakeSeries.Id)
|
2012-02-03 06:49:34 +00:00
|
|
|
|
.With(e => e.Title = "Episode (1)")
|
2012-01-06 21:25:27 +00:00
|
|
|
|
.Build();
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
fakeEpisode2 = Builder<Episode>.CreateNew()
|
2013-02-26 03:58:57 +00:00
|
|
|
|
.With(e => e.SeriesId = fakeSeries.Id)
|
2012-01-06 21:25:27 +00:00
|
|
|
|
.With(e => e.SeasonNumber = fakeEpisode.SeasonNumber)
|
|
|
|
|
.With(e => e.EpisodeNumber = fakeEpisode.EpisodeNumber + 1)
|
2012-02-03 06:49:34 +00:00
|
|
|
|
.With(e => e.Title = "Episode (2)")
|
2012-01-06 21:25:27 +00:00
|
|
|
|
.Build();
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
fakeDailyEpisode = Builder<Episode>.CreateNew()
|
2013-02-26 03:58:57 +00:00
|
|
|
|
.With(e => e.SeriesId = fakeSeries.Id)
|
2012-01-06 21:25:27 +00:00
|
|
|
|
.With(e => e.AirDate = DateTime.Now.Date)
|
|
|
|
|
.With(e => e.Title = "Daily Episode 1")
|
|
|
|
|
.Build();
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2013-03-02 18:25:39 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
|
2013-02-20 02:05:15 +00:00
|
|
|
|
episodeService = Mocker.Resolve<EpisodeService>();
|
2011-06-22 01:12:20 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Test]
|
2012-01-06 21:25:27 +00:00
|
|
|
|
public void existing_single_episode_should_return_single_existing_episode()
|
2011-06-22 01:12:20 +00:00
|
|
|
|
{
|
2012-01-06 21:25:27 +00:00
|
|
|
|
Db.Insert(fakeEpisode);
|
|
|
|
|
Db.Insert(fakeSeries);
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
|
|
|
|
var parseResult = new EpisodeParseResult
|
2011-12-01 19:05:17 +00:00
|
|
|
|
{
|
|
|
|
|
Series = fakeSeries,
|
2012-01-06 21:25:27 +00:00
|
|
|
|
SeasonNumber = fakeEpisode.SeasonNumber,
|
|
|
|
|
EpisodeNumbers = new List<int> { fakeEpisode.EpisodeNumber }
|
2011-12-01 19:05:17 +00:00
|
|
|
|
};
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2013-02-20 02:05:15 +00:00
|
|
|
|
var ep = episodeService.GetEpisodesByParseResult(parseResult);
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
|
|
|
|
ep.Should().HaveCount(1);
|
2012-01-06 21:25:27 +00:00
|
|
|
|
parseResult.EpisodeTitle.Should().Be(fakeEpisode.Title);
|
|
|
|
|
VerifyEpisode(ep[0], fakeEpisode);
|
2013-03-02 18:25:39 +00:00
|
|
|
|
Db<Episode>().Should().HaveCount(1);
|
2011-06-22 01:12:20 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Test]
|
2012-01-06 21:25:27 +00:00
|
|
|
|
public void single_none_existing_episode_should_return_nothing_and_add_nothing()
|
2011-06-22 01:12:20 +00:00
|
|
|
|
{
|
|
|
|
|
var parseResult = new EpisodeParseResult
|
2011-12-01 19:05:17 +00:00
|
|
|
|
{
|
|
|
|
|
Series = fakeSeries,
|
2012-01-06 21:25:27 +00:00
|
|
|
|
SeasonNumber = fakeEpisode.SeasonNumber,
|
|
|
|
|
EpisodeNumbers = new List<int> { 10 }
|
2011-12-01 19:05:17 +00:00
|
|
|
|
};
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2013-02-20 02:05:15 +00:00
|
|
|
|
var episode = episodeService.GetEpisodesByParseResult(parseResult);
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
episode.Should().BeEmpty();
|
|
|
|
|
Db.Fetch<Episode>().Should().HaveCount(0);
|
2011-06-22 01:12:20 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Test]
|
2012-01-06 21:25:27 +00:00
|
|
|
|
public void single_none_existing_series_should_return_nothing_and_add_nothing()
|
2011-06-22 01:12:20 +00:00
|
|
|
|
{
|
|
|
|
|
var parseResult = new EpisodeParseResult
|
2012-01-06 21:25:27 +00:00
|
|
|
|
{
|
|
|
|
|
Series = fakeSeries,
|
|
|
|
|
SeasonNumber = 10,
|
|
|
|
|
EpisodeNumbers = new List<int> { 10 }
|
|
|
|
|
};
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2013-02-20 02:05:15 +00:00
|
|
|
|
var episode = episodeService.GetEpisodesByParseResult(parseResult);
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
episode.Should().BeEmpty();
|
2011-12-01 19:05:17 +00:00
|
|
|
|
Db.Fetch<Episode>().Should().HaveCount(0);
|
2011-06-22 01:12:20 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Test]
|
2012-01-06 21:25:27 +00:00
|
|
|
|
public void existing_multi_episode_should_return_all_episodes()
|
2011-06-22 01:12:20 +00:00
|
|
|
|
{
|
2012-01-06 21:25:27 +00:00
|
|
|
|
Db.Insert(fakeSeries);
|
|
|
|
|
Db.Insert(fakeEpisode);
|
|
|
|
|
Db.Insert(fakeEpisode2);
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var parseResult = new EpisodeParseResult
|
2011-12-01 19:05:17 +00:00
|
|
|
|
{
|
|
|
|
|
Series = fakeSeries,
|
2012-01-06 21:25:27 +00:00
|
|
|
|
SeasonNumber = fakeEpisode.SeasonNumber,
|
|
|
|
|
EpisodeNumbers = new List<int> { fakeEpisode.EpisodeNumber, fakeEpisode2.EpisodeNumber }
|
2011-12-01 19:05:17 +00:00
|
|
|
|
};
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
2013-02-20 02:05:15 +00:00
|
|
|
|
var ep = episodeService.GetEpisodesByParseResult(parseResult);
|
2011-06-22 01:12:20 +00:00
|
|
|
|
|
|
|
|
|
ep.Should().HaveCount(2);
|
2011-12-01 19:05:17 +00:00
|
|
|
|
Db.Fetch<Episode>().Should().HaveCount(2);
|
2011-10-04 02:01:39 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
VerifyEpisode(ep[0], fakeEpisode);
|
|
|
|
|
VerifyEpisode(ep[1], fakeEpisode2);
|
2012-02-03 06:49:34 +00:00
|
|
|
|
parseResult.EpisodeTitle.Should().Be("Episode");
|
2012-01-06 21:25:27 +00:00
|
|
|
|
}
|
2011-10-04 02:01:39 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
2012-01-06 21:25:27 +00:00
|
|
|
|
public void none_existing_multi_episode_should_not_return_or_add_anything()
|
2011-10-04 02:01:39 +00:00
|
|
|
|
{
|
|
|
|
|
var parseResult = new EpisodeParseResult
|
2012-01-06 21:25:27 +00:00
|
|
|
|
{
|
|
|
|
|
Series = fakeSeries,
|
|
|
|
|
SeasonNumber = fakeEpisode.SeasonNumber,
|
|
|
|
|
EpisodeNumbers = new List<int> { fakeEpisode.EpisodeNumber, fakeEpisode2.EpisodeNumber }
|
|
|
|
|
};
|
2011-10-04 02:01:39 +00:00
|
|
|
|
|
2013-02-20 02:05:15 +00:00
|
|
|
|
var ep = episodeService.GetEpisodesByParseResult(parseResult);
|
2011-10-04 02:01:39 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
ep.Should().BeEmpty();
|
2012-02-03 06:49:34 +00:00
|
|
|
|
Db.Fetch<Episode>().Should().BeEmpty();
|
2011-10-04 02:01:39 +00:00
|
|
|
|
}
|
2011-10-20 23:07:47 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
|
2011-10-20 23:07:47 +00:00
|
|
|
|
[Test]
|
|
|
|
|
public void GetEpisodeParseResult_should_return_empty_list_if_episode_list_is_null()
|
|
|
|
|
{
|
2013-03-28 22:07:09 +00:00
|
|
|
|
|
2013-02-20 02:05:15 +00:00
|
|
|
|
var episodes = episodeService.GetEpisodesByParseResult(new EpisodeParseResult());
|
2013-03-28 22:07:09 +00:00
|
|
|
|
|
2011-10-20 23:07:47 +00:00
|
|
|
|
episodes.Should().NotBeNull();
|
|
|
|
|
episodes.Should().BeEmpty();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
public void GetEpisodeParseResult_should_return_empty_list_if_episode_list_is_empty()
|
|
|
|
|
{
|
2013-03-28 22:07:09 +00:00
|
|
|
|
|
2013-02-20 02:05:15 +00:00
|
|
|
|
var episodes = episodeService.GetEpisodesByParseResult(new EpisodeParseResult { EpisodeNumbers = new List<int>() });
|
2013-03-28 22:07:09 +00:00
|
|
|
|
|
2011-10-20 23:07:47 +00:00
|
|
|
|
episodes.Should().NotBeNull();
|
|
|
|
|
episodes.Should().BeEmpty();
|
|
|
|
|
}
|
|
|
|
|
|
2011-11-24 08:28:20 +00:00
|
|
|
|
[Test]
|
2012-01-06 21:25:27 +00:00
|
|
|
|
public void should_return_single_episode_when_air_date_is_provided()
|
2011-11-24 08:28:20 +00:00
|
|
|
|
{
|
2011-12-01 01:10:36 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
Db.Insert(fakeSeries);
|
|
|
|
|
Db.Insert(fakeDailyEpisode);
|
2011-11-24 08:28:20 +00:00
|
|
|
|
|
2013-03-28 22:07:09 +00:00
|
|
|
|
|
2013-02-20 02:05:15 +00:00
|
|
|
|
var episodes = episodeService.GetEpisodesByParseResult(new EpisodeParseResult { AirDate = DateTime.Today, Series = fakeDailySeries });
|
2011-11-24 08:28:20 +00:00
|
|
|
|
|
2013-03-28 22:07:09 +00:00
|
|
|
|
|
2011-11-24 08:28:20 +00:00
|
|
|
|
episodes.Should().HaveCount(1);
|
2012-01-06 21:25:27 +00:00
|
|
|
|
VerifyEpisode(episodes[0], fakeDailyEpisode);
|
2011-12-01 01:10:36 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
Db.Fetch<Episode>().Should().HaveCount(1);
|
2011-11-24 08:28:20 +00:00
|
|
|
|
}
|
2011-12-01 19:05:17 +00:00
|
|
|
|
|
|
|
|
|
[Test]
|
2012-01-06 21:25:27 +00:00
|
|
|
|
public void should_not_add_episode_when_episode_doesnt_exist()
|
2011-12-01 19:05:17 +00:00
|
|
|
|
{
|
2013-02-20 02:05:15 +00:00
|
|
|
|
var episodes = episodeService.GetEpisodesByParseResult(new EpisodeParseResult { AirDate = DateTime.Today, Series = fakeDailySeries });
|
2011-12-01 19:05:17 +00:00
|
|
|
|
|
2013-03-28 22:07:09 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
episodes.Should().HaveCount(0);
|
|
|
|
|
Db.Fetch<Episode>().Should().HaveCount(0);
|
2011-12-01 19:05:17 +00:00
|
|
|
|
}
|
2011-12-02 06:26:24 +00:00
|
|
|
|
|
2012-01-06 21:25:27 +00:00
|
|
|
|
|
2011-12-02 06:26:24 +00:00
|
|
|
|
[Test]
|
2012-02-03 06:49:34 +00:00
|
|
|
|
public void GetEpisodeParseResult_should_return_single_title_for_multiple_episodes()
|
2011-12-02 06:26:24 +00:00
|
|
|
|
{
|
2012-01-06 21:25:27 +00:00
|
|
|
|
Db.Insert(fakeSeries);
|
2011-12-02 06:26:24 +00:00
|
|
|
|
Db.Insert(fakeEpisode);
|
|
|
|
|
Db.Insert(fakeEpisode2);
|
2012-01-06 21:25:27 +00:00
|
|
|
|
|
2011-12-02 06:26:24 +00:00
|
|
|
|
var parseResult = new EpisodeParseResult
|
|
|
|
|
{
|
|
|
|
|
Series = fakeSeries,
|
2012-01-06 21:25:27 +00:00
|
|
|
|
SeasonNumber = fakeEpisode.SeasonNumber,
|
2011-12-03 02:29:05 +00:00
|
|
|
|
EpisodeNumbers = new List<int> { fakeEpisode.EpisodeNumber, fakeEpisode2.EpisodeNumber }
|
2011-12-02 06:26:24 +00:00
|
|
|
|
};
|
|
|
|
|
|
2013-02-20 02:05:15 +00:00
|
|
|
|
var ep = episodeService.GetEpisodesByParseResult(parseResult);
|
2011-12-02 06:26:24 +00:00
|
|
|
|
|
|
|
|
|
ep.Should().HaveCount(2);
|
|
|
|
|
Db.Fetch<Episode>().Should().HaveCount(2);
|
2012-01-06 21:25:27 +00:00
|
|
|
|
|
|
|
|
|
VerifyEpisode(ep[0], fakeEpisode);
|
|
|
|
|
VerifyEpisode(ep[1], fakeEpisode2);
|
|
|
|
|
|
2012-02-03 06:49:34 +00:00
|
|
|
|
parseResult.EpisodeTitle.Should().Be("Episode");
|
2011-12-02 06:26:24 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
public void GetEpisodeParseResult_should_return_single_title_for_single_episode()
|
|
|
|
|
{
|
|
|
|
|
Db.Insert(fakeEpisode);
|
|
|
|
|
Db.Insert(fakeSeries);
|
|
|
|
|
|
|
|
|
|
var parseResult = new EpisodeParseResult
|
|
|
|
|
{
|
|
|
|
|
Series = fakeSeries,
|
2012-01-06 21:25:27 +00:00
|
|
|
|
SeasonNumber = fakeEpisode.SeasonNumber,
|
2011-12-03 02:29:05 +00:00
|
|
|
|
EpisodeNumbers = new List<int> { fakeEpisode.EpisodeNumber }
|
2011-12-02 06:26:24 +00:00
|
|
|
|
};
|
|
|
|
|
|
2013-02-20 02:05:15 +00:00
|
|
|
|
var ep = episodeService.GetEpisodesByParseResult(parseResult);
|
2011-12-02 06:26:24 +00:00
|
|
|
|
|
|
|
|
|
ep.Should().HaveCount(1);
|
|
|
|
|
Db.Fetch<Episode>().Should().HaveCount(1);
|
|
|
|
|
ep.First().ShouldHave().AllPropertiesBut(e => e.Series);
|
|
|
|
|
parseResult.EpisodeTitle.Should().Be(fakeEpisode.Title);
|
|
|
|
|
}
|
2011-12-20 00:45:47 +00:00
|
|
|
|
|
|
|
|
|
[Test]
|
2012-01-06 21:25:27 +00:00
|
|
|
|
public void GetEpisodeParseResult_should_return_nothing_when_series_is_not_daily_but_parsed_daily()
|
2011-12-20 00:45:47 +00:00
|
|
|
|
{
|
|
|
|
|
Db.Insert(fakeSeries);
|
|
|
|
|
|
|
|
|
|
var parseResult = new EpisodeParseResult
|
|
|
|
|
{
|
|
|
|
|
Series = fakeSeries,
|
|
|
|
|
AirDate = DateTime.Today
|
|
|
|
|
};
|
|
|
|
|
|
2013-02-20 02:05:15 +00:00
|
|
|
|
var ep = episodeService.GetEpisodesByParseResult(parseResult);
|
2011-12-20 00:45:47 +00:00
|
|
|
|
|
|
|
|
|
ep.Should().BeEmpty();
|
2011-12-20 00:58:26 +00:00
|
|
|
|
ExceptionVerification.ExpectedWarns(1);
|
2011-12-20 00:45:47 +00:00
|
|
|
|
}
|
2012-01-06 21:25:27 +00:00
|
|
|
|
|
|
|
|
|
private void VerifyEpisode(Episode actual, Episode excpected)
|
|
|
|
|
{
|
|
|
|
|
actual.ShouldHave().AllProperties().But(e => e.Series).But(e => e.EpisodeFile).EqualTo(excpected);
|
|
|
|
|
}
|
2011-06-22 01:12:20 +00:00
|
|
|
|
}
|
2011-12-01 19:05:17 +00:00
|
|
|
|
}
|
2013-03-02 18:25:39 +00:00
|
|
|
|
*/
|